Hexor/tracker_watcher
1
0
Fork 0
mirror of https://github.com/house-of-vanity/tracker_watcher.git synced 2025-07-07 09:54:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

It works

Browse Source
This commit is contained in:
UltraDesu
2018-01-17 17:25:36 +03:00
parent 1b5ceb3c1b
commit 0ab1b49a14
2 changed files with 36 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
updater.py
View File

@ -5,6 +5,8 @@ import urllib.request, json
import datetime as dt import datetime as dt
from configparser import ConfigParser from configparser import ConfigParser
import pytz import pytz
from urllib.parse import urlencode
parser = ConfigParser() parser = ConfigParser()
parser.read('settings.ini') parser.read('settings.ini')
@ -24,15 +26,26 @@ connection = pymysql.connect(host=mysql_host,
# If u_date which already stored older than fresh u_date # If u_date which already stored older than fresh u_date
# so going to notify # so going to notify
# Any way update last_check # Any way update last_check
def check_updates(id, u_date): def check_updates(id, u_date, cursor):
with urllib.request.urlopen( link = "http://api.rutracker.org/v1/get_tor_topic_data?by=topic_id&val="+id
"http://api.rutracker.org/v1/get_tor_topic_data?by=topic_id&val="+id) as url: with urllib.request.urlopen(link) as url:
data = json.loads(url.read().decode()) data = json.loads(url.read().decode())
last_check = dt.datetime.now(tz=pytz.utc) last_check = dt.datetime.now(tz=pytz.utc)
new_u_date = dt.datetime.fromtimestamp(int(data['result'][id]['reg_time']), tz=pytz.utc) new_u_date = dt.datetime.fromtimestamp(int(data['result'][id]['reg_time']), tz=pytz.utc)
u_date = pytz.utc.localize(u_date) u_date = pytz.utc.localize(u_date)
if new_u_date > u_date: if new_u_date > u_date:
print("There is an update. Going to notify.") sql = "SELECT c.username, c.user_id " \
"FROM notification n LEFT JOIN contact c " \
"ON n.user_id = c.user_id WHERE n.topic_id = '%s'" % id
cursor.execute(sql)
result = cursor.fetchall()
for contact in result:
print(contact)
msg = "%s has been updated. %s" % (
data['result'][id]['topic_title'],
'https://rutracker.org/forum/viewtopic.php?t='+id)
send(contact['user_id'], msg)
with connection.cursor() as cursor: with connection.cursor() as cursor:
# Create a new record # Create a new record
@ -43,6 +56,15 @@ def check_updates(id, u_date):
id)) id))
connection.commit() connection.commit()
def send(id, msg):
url = parser.get('bot', 'telegram_api') + 'bot'+ parser.get('bot', 'telegram_key') + '/sendMessage'
post_fields = {
'text': msg,
'chat_id': id
}
request = urllib.request.Request(url, urlencode(post_fields).encode())
json = urllib.request.urlopen(request).read().decode()
try: try:
with connection.cursor() as cursor: with connection.cursor() as cursor:
# Read a single record # Read a single record
@ -52,6 +74,6 @@ try:
result = cursor.fetchall() result = cursor.fetchall()
for line in result: for line in result:
print('Going to check %s. Last check was at %s' % (line['link'], line['last_check'])) print('Going to check %s. Last check was at %s' % (line['link'], line['last_check']))
check_updates(line['link'], line['u_date']) check_updates(line['link'], line['u_date'], cursor)
finally: finally:
connection.close() connection.close()

14
webhook/index.php
View File

@ -68,7 +68,12 @@ function notify($url){
send('URL is invalid.', $die = TRUE); send('URL is invalid.', $die = TRUE);
die(); die();
} }
$url = $url['t']; $url = $url['t'];
$json = file_get_contents(
'http://api.rutracker.org/v1/get_tor_topic_data?by=topic_id&val='.$url
);
$obj = json_decode($json);
$stmt = $dbh->query( $stmt = $dbh->query(
'SELECT * FROM notification n 'SELECT * FROM notification n
LEFT JOIN contact c ON n.user_id = c.user_id LEFT JOIN contact c ON n.user_id = c.user_id
@ -76,9 +81,9 @@ function notify($url){
); );
if($stmt->rowCount() > 0) if($stmt->rowCount() > 0)
{ {
send('You already subscribed for '.$url." updates.", $die = TRUE); send('You already subscribed to '.$obj->result->{$url}->topic_title, $die = TRUE);
}else{ }else{
send('You will be subscribed for '.$url." updates."); send('You will be subscribed to '.$obj->result->{$url}->topic_title);
$stmt = $dbh->query( $stmt = $dbh->query(
'SELECT * FROM `url` 'SELECT * FROM `url`
WHERE link = "'.$url.'"' WHERE link = "'.$url.'"'
@ -93,10 +98,7 @@ function notify($url){
); );
}else{ }else{
// insert new url and user in db // insert new url and user in db
$json = file_get_contents(
'http://api.rutracker.org/v1/get_tor_topic_data?by=topic_id&val='.$url
);
$obj = json_decode($json);
$stmt = $dbh->query( $stmt = $dbh->query(
'INSERT into url (link,u_date) 'INSERT into url (link,u_date)