mirror of
https://github.com/YGGverse/qBt_SE.git
synced 2025-01-18 02:30:00 +00:00
avoid endless waiting
This commit is contained in:
imDMG
parent
97094d6b4f
commit
c2a0e49937
33
kinozal.py
33
kinozal.py
@ -1,5 +1,5 @@
|
||||
# VERSION: 1.1
|
||||
# AUTHORS: imDMG (imdmgg@gmail.com)
|
||||
# VERSION: 1.2
|
||||
# AUTHORS: imDMG [imdmgg@gmail.com]
|
||||
|
||||
# Kinozal.tv search engine plugin for qBittorrent
|
||||
|
||||
@ -56,6 +56,9 @@ class kinozal(object):
|
||||
cj = CookieJar()
|
||||
self.session = build_opener(HTTPCookieProcessor(cj))
|
||||
|
||||
# avoid endless waiting
|
||||
self.blocked = False
|
||||
|
||||
# add proxy handler if needed
|
||||
if self.config['proxy'] and any(self.config['proxies'].keys()):
|
||||
self.session.add_handler(ProxyHandler(self.config['proxies']))
|
||||
@ -68,16 +71,12 @@ class kinozal(object):
|
||||
# so we first encode keys to cp1251 then do default decode whole string
|
||||
data_encoded = urlencode({k: v.encode('cp1251') for k, v in form_data.items()}).encode()
|
||||
|
||||
response = self._catch_error_request(self.url + '/takelogin.php', data_encoded)
|
||||
# checking that tracker is'nt blocked
|
||||
if self.url not in response.geturl():
|
||||
logging.warning("{} is blocked. Try proxy or another proxy".format(self.url))
|
||||
exit()
|
||||
|
||||
if 'uid' not in [cookie.name for cookie in cj]:
|
||||
logging.warning("we not authorized, please check your credentials")
|
||||
else:
|
||||
logging.info('We successfully authorized')
|
||||
self._catch_error_request(self.url + '/takelogin.php', data_encoded)
|
||||
if not self.blocked:
|
||||
if 'uid' not in [cookie.name for cookie in cj]:
|
||||
logging.warning("we not authorized, please check your credentials")
|
||||
else:
|
||||
logging.info('We successfully authorized')
|
||||
|
||||
class WorstParser(HTMLParser):
|
||||
def __init__(self, url=''):
|
||||
@ -188,6 +187,8 @@ class kinozal(object):
|
||||
pass
|
||||
|
||||
def download_torrent(self, url):
|
||||
if self.blocked:
|
||||
return
|
||||
# Create a torrent file
|
||||
file, path = tempfile.mkstemp('.torrent')
|
||||
file = os.fdopen(file, "wb")
|
||||
@ -204,6 +205,8 @@ class kinozal(object):
|
||||
print(path + " " + url)
|
||||
|
||||
def search(self, what, cat='all'):
|
||||
if self.blocked:
|
||||
return
|
||||
query = '{}/browse.php?s={}&c={}'.format(self.url, what.replace(" ", "+"), self.supported_categories[cat])
|
||||
response = self._catch_error_request(query)
|
||||
parser = self.WorstParser(self.url)
|
||||
@ -235,6 +238,12 @@ class kinozal(object):
|
||||
logging.error(e)
|
||||
raise e
|
||||
|
||||
# checking that tracker is'nt blocked
|
||||
self.blocked = False
|
||||
if self.url not in response.geturl():
|
||||
logging.warning("{} is blocked. Try proxy or another proxy".format(self.url))
|
||||
self.blocked = True
|
||||
|
||||
return response
|
||||
|
||||
|
||||
|
||||
50
nnmclub.py
50
nnmclub.py
@ -1,4 +1,4 @@
|
||||
# VERSION: 1.1
|
||||
# VERSION: 1.2
|
||||
# AUTHORS: imDMG [imdmgg@gmail.com]
|
||||
|
||||
# NoNaMe-Club search engine plugin for qBittorrent
|
||||
@ -60,6 +60,9 @@ class nnmclub(object):
|
||||
cj.set_cookie(c)
|
||||
self.session = build_opener(HTTPCookieProcessor(cj))
|
||||
|
||||
# avoid endless waiting
|
||||
self.blocked = False
|
||||
|
||||
# add proxy handler if needed
|
||||
if self.config['proxy'] and any(self.config['proxies'].keys()):
|
||||
self.session.add_handler(ProxyHandler(self.config['proxies']))
|
||||
@ -69,28 +72,25 @@ class nnmclub(object):
|
||||
self.session.addheaders.append(('User-Agent', self.config['ua']))
|
||||
|
||||
response = self._catch_error_request(self.url + 'login.php')
|
||||
# checking that tracker is'nt blocked
|
||||
if self.url not in response.geturl():
|
||||
logging.warning("{} is blocked. Try proxy or another proxy".format(self.url))
|
||||
exit()
|
||||
parser = self.WorstParser(self.url, True)
|
||||
parser.feed(response.read().decode('cp1251'))
|
||||
parser.close()
|
||||
if not self.blocked:
|
||||
parser = self.WorstParser(self.url, True)
|
||||
parser.feed(response.read().decode('cp1251'))
|
||||
parser.close()
|
||||
|
||||
form_data = {"username": self.config['username'],
|
||||
"password": self.config['password'],
|
||||
"autologin": "on",
|
||||
"code": parser.login_code,
|
||||
"login": "Вход"}
|
||||
# so we first encode keys to cp1251 then do default decode whole string
|
||||
data_encoded = urlencode({k: v.encode('cp1251') for k, v in form_data.items()}).encode()
|
||||
form_data = {"username": self.config['username'],
|
||||
"password": self.config['password'],
|
||||
"autologin": "on",
|
||||
"code": parser.login_code,
|
||||
"login": "Вход"}
|
||||
# so we first encode keys to cp1251 then do default decode whole string
|
||||
data_encoded = urlencode({k: v.encode('cp1251') for k, v in form_data.items()}).encode()
|
||||
|
||||
self._catch_error_request(self.url + 'login.php', data_encoded)
|
||||
self._catch_error_request(self.url + 'login.php', data_encoded)
|
||||
|
||||
if 'phpbb2mysql_4_sid' not in [cookie.name for cookie in cj]:
|
||||
logging.warning("we not authorized, please check your credentials")
|
||||
else:
|
||||
logging.info('We successfully authorized')
|
||||
if 'phpbb2mysql_4_sid' not in [cookie.name for cookie in cj]:
|
||||
logging.warning("we not authorized, please check your credentials")
|
||||
else:
|
||||
logging.info('We successfully authorized')
|
||||
|
||||
class WorstParser(HTMLParser):
|
||||
def __init__(self, url='', login=False):
|
||||
@ -210,6 +210,8 @@ class nnmclub(object):
|
||||
pass
|
||||
|
||||
def download_torrent(self, url):
|
||||
if self.blocked:
|
||||
return
|
||||
# Create a torrent file
|
||||
file, path = tempfile.mkstemp('.torrent')
|
||||
file = os.fdopen(file, "wb")
|
||||
@ -226,6 +228,8 @@ class nnmclub(object):
|
||||
print(path + " " + url)
|
||||
|
||||
def search(self, what, cat='all'):
|
||||
if self.blocked:
|
||||
return
|
||||
c = self.supported_categories[cat]
|
||||
query = '{}tracker.php?nm={}&{}'.format(self.url, what.replace(" ", "+"), "f=-1" if c == '-1' else "c=" + c)
|
||||
response = self._catch_error_request(query)
|
||||
@ -260,6 +264,12 @@ class nnmclub(object):
|
||||
logging.error(e)
|
||||
raise e
|
||||
|
||||
# checking that tracker is'nt blocked
|
||||
self.blocked = False
|
||||
if self.url not in response.geturl():
|
||||
logging.warning("{} is blocked. Try proxy or another proxy".format(self.url))
|
||||
self.blocked = True
|
||||
|
||||
return response
|
||||
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user