mirror of
https://github.com/moparisthebest/SickRage
synced 2024-12-13 03:22:22 -05:00
Merge branch 'feature/feat-webui-improvements' into develop
This commit is contained in:
commit
e2be96542d
@ -18,48 +18,53 @@
|
|||||||
|
|
||||||
import os
|
import os
|
||||||
import traceback
|
import traceback
|
||||||
|
import re
|
||||||
import sickbeard
|
import sickbeard
|
||||||
import six
|
import six
|
||||||
import chardet
|
import chardet
|
||||||
|
import unicodedata
|
||||||
|
|
||||||
|
from string import ascii_letters, digits
|
||||||
from sickbeard import logger
|
from sickbeard import logger
|
||||||
|
|
||||||
# This module tries to deal with the apparently random behavior of python when dealing with unicode <-> utf-8
|
def toSafeString(original):
|
||||||
# encodings. It tries to just use unicode, but if that fails then it tries forcing it to utf-8. Any functions
|
valid_chars = "-_.() %s%s" % (ascii_letters, digits)
|
||||||
# which return something should always return unicode.
|
cleaned_filename = unicodedata.normalize('NFKD', _toUnicode(original)).encode('ASCII', 'ignore')
|
||||||
|
valid_string = ''.join(c for c in cleaned_filename if c in valid_chars)
|
||||||
|
return ' '.join(valid_string.split())
|
||||||
|
|
||||||
|
|
||||||
|
def simplifyString(original):
|
||||||
|
string = stripAccents(original.lower())
|
||||||
|
string = toSafeString(' '.join(re.split('\W+', string)))
|
||||||
|
split = re.split('\W+|_', string.lower())
|
||||||
|
return _toUnicode(' '.join(split))
|
||||||
|
|
||||||
def _toUnicode(x):
|
def _toUnicode(x):
|
||||||
try:
|
if isinstance(x, unicode):
|
||||||
if isinstance(x, unicode):
|
return x
|
||||||
return x
|
else:
|
||||||
else:
|
try:
|
||||||
|
return six.text_type(x)
|
||||||
|
except:
|
||||||
try:
|
try:
|
||||||
return six.text_type(x)
|
if chardet.detect(x).get('encoding') == 'utf-8':
|
||||||
|
return x.decode('utf-8')
|
||||||
|
if isinstance(x, str):
|
||||||
|
try:
|
||||||
|
return x.decode(sickbeard.SYS_ENCODING)
|
||||||
|
except UnicodeDecodeError:
|
||||||
|
raise
|
||||||
|
return x
|
||||||
except:
|
except:
|
||||||
try:
|
return x
|
||||||
if chardet.detect(x).get('encoding') == 'utf-8':
|
|
||||||
return x.decode('utf-8')
|
|
||||||
if isinstance(x, str):
|
|
||||||
try:
|
|
||||||
return x.decode(sickbeard.SYS_ENCODING)
|
|
||||||
except UnicodeDecodeError:
|
|
||||||
raise
|
|
||||||
return x
|
|
||||||
except:
|
|
||||||
raise
|
|
||||||
except:
|
|
||||||
logger.log('Unable to decode value "%s..." : %s ' % (repr(x)[:20], traceback.format_exc()), logger.WARNING)
|
|
||||||
ascii_text = str(x).encode('string_escape')
|
|
||||||
return _toUnicode(ascii_text)
|
|
||||||
|
|
||||||
def ss(x):
|
def ss(x):
|
||||||
u_x = _toUnicode(x)
|
u_x = _toUnicode(x)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
return u_x.encode(sickbeard.SYS_ENCODING)
|
return u_x.encode(sickbeard.SYS_ENCODING)
|
||||||
except Exception as e:
|
except:
|
||||||
logger.log('Failed ss encoding char, force UTF8: %s' % e, logger.WARNING)
|
|
||||||
try:
|
try:
|
||||||
return u_x.encode(sickbeard.SYS_ENCODING, 'replace')
|
return u_x.encode(sickbeard.SYS_ENCODING, 'replace')
|
||||||
except:
|
except:
|
||||||
@ -84,3 +89,6 @@ def ek(func, *args, **kwargs):
|
|||||||
return _toUnicode(result)
|
return _toUnicode(result)
|
||||||
else:
|
else:
|
||||||
return result
|
return result
|
||||||
|
|
||||||
|
def stripAccents(s):
|
||||||
|
return ''.join((c for c in unicodedata.normalize('NFD', _toUnicode(s)) if unicodedata.category(c) != 'Mn'))
|
@ -1431,3 +1431,5 @@ def get_size(start_path='.'):
|
|||||||
total_size += ek.ek(os.path.getsize, fp)
|
total_size += ek.ek(os.path.getsize, fp)
|
||||||
return total_size
|
return total_size
|
||||||
|
|
||||||
|
def md5(text):
|
||||||
|
return hashlib.md5(ek.ss(text)).hexdigest()
|
@ -79,7 +79,7 @@ class Animezb(generic.NZBProvider):
|
|||||||
logger.log(u"Search url: " + search_url, logger.DEBUG)
|
logger.log(u"Search url: " + search_url, logger.DEBUG)
|
||||||
|
|
||||||
results = []
|
results = []
|
||||||
for curItem in self.cache.getRSSFeed(search_url):
|
for curItem in self.cache.getRSSFeed(search_url, items=['entries']) or []:
|
||||||
(title, url) = self._get_title_and_url(curItem)
|
(title, url) = self._get_title_and_url(curItem)
|
||||||
|
|
||||||
if title and url:
|
if title and url:
|
||||||
@ -134,6 +134,6 @@ class AnimezbCache(tvcache.TVCache):
|
|||||||
|
|
||||||
logger.log(self.provider.name + u" cache update URL: " + rss_url, logger.DEBUG)
|
logger.log(self.provider.name + u" cache update URL: " + rss_url, logger.DEBUG)
|
||||||
|
|
||||||
return self.getRSSFeed(rss_url)
|
return self.getRSSFeed(rss_url, items=['entries', 'feed'])
|
||||||
|
|
||||||
provider = Animezb()
|
provider = Animezb()
|
||||||
|
@ -157,7 +157,7 @@ class BitSoupProvider(generic.TorrentProvider):
|
|||||||
items = {'Season': [], 'Episode': [], 'RSS': []}
|
items = {'Season': [], 'Episode': [], 'RSS': []}
|
||||||
|
|
||||||
if not self._doLogin():
|
if not self._doLogin():
|
||||||
return []
|
return results
|
||||||
|
|
||||||
for mode in search_params.keys():
|
for mode in search_params.keys():
|
||||||
for search_string in search_params[mode]:
|
for search_string in search_params[mode]:
|
||||||
@ -273,7 +273,7 @@ class BitSoupCache(tvcache.TVCache):
|
|||||||
|
|
||||||
def _getRSSData(self):
|
def _getRSSData(self):
|
||||||
search_params = {'RSS': ['']}
|
search_params = {'RSS': ['']}
|
||||||
return self.provider._doSearch(search_params)
|
return {'entries': self.provider._doSearch(search_params)}
|
||||||
|
|
||||||
|
|
||||||
provider = BitSoupProvider()
|
provider = BitSoupProvider()
|
||||||
|
@ -92,7 +92,7 @@ class BTNProvider(generic.TorrentProvider):
|
|||||||
parsedJSON = self._api_call(apikey, params)
|
parsedJSON = self._api_call(apikey, params)
|
||||||
if not parsedJSON:
|
if not parsedJSON:
|
||||||
logger.log(u"No data returned from " + self.name, logger.ERROR)
|
logger.log(u"No data returned from " + self.name, logger.ERROR)
|
||||||
return []
|
return results
|
||||||
|
|
||||||
if self._checkAuthFromData(parsedJSON):
|
if self._checkAuthFromData(parsedJSON):
|
||||||
|
|
||||||
@ -311,7 +311,7 @@ class BTNCache(tvcache.TVCache):
|
|||||||
logger.WARNING)
|
logger.WARNING)
|
||||||
seconds_since_last_update = 86400
|
seconds_since_last_update = 86400
|
||||||
|
|
||||||
return self.provider._doSearch(search_params=None, age=seconds_since_last_update)
|
return {'entries': self.provider._doSearch(search_params=None, age=seconds_since_last_update)}
|
||||||
|
|
||||||
|
|
||||||
provider = BTNProvider()
|
provider = BTNProvider()
|
||||||
|
@ -123,7 +123,7 @@ class EZRSSProvider(generic.TorrentProvider):
|
|||||||
logger.log(u"Search string: " + search_url, logger.DEBUG)
|
logger.log(u"Search string: " + search_url, logger.DEBUG)
|
||||||
|
|
||||||
results = []
|
results = []
|
||||||
for curItem in self.cache.getRSSFeed(search_url):
|
for curItem in self.cache.getRSSFeed(search_url, items=['entries']) or []:
|
||||||
|
|
||||||
(title, url) = self._get_title_and_url(curItem)
|
(title, url) = self._get_title_and_url(curItem)
|
||||||
|
|
||||||
@ -172,6 +172,6 @@ class EZRSSCache(tvcache.TVCache):
|
|||||||
rss_url = self.provider.url + 'feed/'
|
rss_url = self.provider.url + 'feed/'
|
||||||
logger.log(self.provider.name + " cache update URL: " + rss_url, logger.DEBUG)
|
logger.log(self.provider.name + " cache update URL: " + rss_url, logger.DEBUG)
|
||||||
|
|
||||||
return self.getRSSFeed(rss_url)
|
return self.getRSSFeed(rss_url, items=['entries', 'feed'])
|
||||||
|
|
||||||
provider = EZRSSProvider()
|
provider = EZRSSProvider()
|
||||||
|
@ -74,7 +74,7 @@ class Fanzub(generic.NZBProvider):
|
|||||||
logger.log(u"Search url: " + search_url, logger.DEBUG)
|
logger.log(u"Search url: " + search_url, logger.DEBUG)
|
||||||
|
|
||||||
results = []
|
results = []
|
||||||
for curItem in self.cache.getRSSFeed(search_url):
|
for curItem in self.cache.getRSSFeed(search_url, items=['entries']) or []:
|
||||||
(title, url) = self._get_title_and_url(curItem)
|
(title, url) = self._get_title_and_url(curItem)
|
||||||
|
|
||||||
if title and url:
|
if title and url:
|
||||||
@ -129,6 +129,6 @@ class FanzubCache(tvcache.TVCache):
|
|||||||
|
|
||||||
logger.log(self.provider.name + u" cache update URL: " + rss_url, logger.DEBUG)
|
logger.log(self.provider.name + u" cache update URL: " + rss_url, logger.DEBUG)
|
||||||
|
|
||||||
return self.getRSSFeed(rss_url)
|
return self.getRSSFeed(rss_url, items=['entries', 'feed'])
|
||||||
|
|
||||||
provider = Fanzub()
|
provider = Fanzub()
|
||||||
|
@ -309,6 +309,6 @@ class FreshOnTVCache(tvcache.TVCache):
|
|||||||
|
|
||||||
def _getRSSData(self):
|
def _getRSSData(self):
|
||||||
search_params = {'RSS': ['']}
|
search_params = {'RSS': ['']}
|
||||||
return self.provider._doSearch(search_params)
|
return {'entries': self.provider._doSearch(search_params)}
|
||||||
|
|
||||||
provider = FreshOnTVProvider()
|
provider = FreshOnTVProvider()
|
@ -210,11 +210,12 @@ class HDBitsCache(tvcache.TVCache):
|
|||||||
try:
|
try:
|
||||||
parsedJSON = self.provider.getURL(self.provider.rss_url, post_data=self.provider._make_post_data_JSON(),
|
parsedJSON = self.provider.getURL(self.provider.rss_url, post_data=self.provider._make_post_data_JSON(),
|
||||||
json=True)
|
json=True)
|
||||||
|
|
||||||
if self.provider._checkAuthFromData(parsedJSON):
|
if self.provider._checkAuthFromData(parsedJSON):
|
||||||
results = parsedJSON['data']
|
results = parsedJSON['data']
|
||||||
except:
|
except:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
return results
|
return {'entries': results}
|
||||||
|
|
||||||
provider = HDBitsProvider()
|
provider = HDBitsProvider()
|
||||||
|
@ -342,7 +342,7 @@ class HDTorrentsCache(tvcache.TVCache):
|
|||||||
|
|
||||||
def _getRSSData(self):
|
def _getRSSData(self):
|
||||||
search_params = {'RSS': []}
|
search_params = {'RSS': []}
|
||||||
return self.provider._doSearch(search_params)
|
return {'entries': self.provider._doSearch(search_params)}
|
||||||
|
|
||||||
|
|
||||||
provider = HDTorrentsProvider()
|
provider = HDTorrentsProvider()
|
||||||
|
@ -279,7 +279,7 @@ class IPTorrentsCache(tvcache.TVCache):
|
|||||||
|
|
||||||
def _getRSSData(self):
|
def _getRSSData(self):
|
||||||
search_params = {'RSS': ['']}
|
search_params = {'RSS': ['']}
|
||||||
return self.provider._doSearch(search_params)
|
return {'entries': self.provider._doSearch(search_params)}
|
||||||
|
|
||||||
|
|
||||||
provider = IPTorrentsProvider()
|
provider = IPTorrentsProvider()
|
||||||
|
@ -357,6 +357,6 @@ class KATCache(tvcache.TVCache):
|
|||||||
|
|
||||||
def _getRSSData(self):
|
def _getRSSData(self):
|
||||||
search_params = {'RSS': ['rss']}
|
search_params = {'RSS': ['rss']}
|
||||||
return self.provider._doSearch(search_params)
|
return {'entries': self.provider._doSearch(search_params)}
|
||||||
|
|
||||||
provider = KATProvider()
|
provider = KATProvider()
|
||||||
|
@ -238,11 +238,10 @@ class NewznabProvider(generic.NZBProvider):
|
|||||||
|
|
||||||
def _checkAuthFromData(self, data):
|
def _checkAuthFromData(self, data):
|
||||||
|
|
||||||
if not data:
|
if not (data.entries and data.feed):
|
||||||
return self._checkAuth()
|
return self._checkAuth()
|
||||||
|
|
||||||
if data.feed.get('error', None):
|
if data.feed.get('error', None):
|
||||||
|
|
||||||
code = data.feed.error.get('code', None)
|
code = data.feed.error.get('code', None)
|
||||||
|
|
||||||
if code == '100':
|
if code == '100':
|
||||||
@ -297,12 +296,12 @@ class NewznabProvider(generic.NZBProvider):
|
|||||||
while (total >= offset) and (offset < 1000):
|
while (total >= offset) and (offset < 1000):
|
||||||
search_url = self.url + 'api?' + urllib.urlencode(params)
|
search_url = self.url + 'api?' + urllib.urlencode(params)
|
||||||
logger.log(u"Search url: " + search_url, logger.DEBUG)
|
logger.log(u"Search url: " + search_url, logger.DEBUG)
|
||||||
data = self.cache.getRSSFeed(search_url)
|
|
||||||
|
|
||||||
if not data or not self._checkAuthFromData(data):
|
data = self.cache.getRSSFeed(search_url, items=['entries', 'feed'])
|
||||||
|
if not self._checkAuthFromData(data):
|
||||||
break
|
break
|
||||||
|
|
||||||
for item in data.entries:
|
for item in data.entries or []:
|
||||||
|
|
||||||
(title, url) = self._get_title_and_url(item)
|
(title, url) = self._get_title_and_url(item)
|
||||||
|
|
||||||
@ -422,56 +421,13 @@ class NewznabCache(tvcache.TVCache):
|
|||||||
|
|
||||||
logger.log(self.provider.name + " cache update URL: " + rss_url, logger.DEBUG)
|
logger.log(self.provider.name + " cache update URL: " + rss_url, logger.DEBUG)
|
||||||
|
|
||||||
return self.getRSSFeed(rss_url)
|
return self.getRSSFeed(rss_url, items=['entries', 'feed'])
|
||||||
|
|
||||||
def _checkAuth(self, data):
|
def _checkAuth(self, data):
|
||||||
return self.provider._checkAuthFromData(data)
|
return self.provider._checkAuthFromData(data)
|
||||||
|
|
||||||
def updateCache(self):
|
|
||||||
if not self.shouldUpdate():
|
|
||||||
return
|
|
||||||
|
|
||||||
try:
|
|
||||||
if self._checkAuth(None):
|
|
||||||
data = self._getRSSData()
|
|
||||||
if not data or not len(data) > 0:
|
|
||||||
return
|
|
||||||
|
|
||||||
# clear cache
|
|
||||||
self._clearCache()
|
|
||||||
|
|
||||||
# set updated
|
|
||||||
self.setLastUpdate()
|
|
||||||
|
|
||||||
try:
|
|
||||||
items = data.get('entries', [])
|
|
||||||
except:
|
|
||||||
items = data
|
|
||||||
|
|
||||||
if self._checkAuth(items):
|
|
||||||
cl = []
|
|
||||||
for item in items:
|
|
||||||
ci = self._parseItem(item)
|
|
||||||
if ci is not None:
|
|
||||||
cl.append(ci)
|
|
||||||
|
|
||||||
if len(cl) > 0:
|
|
||||||
myDB = self._getDB()
|
|
||||||
myDB.mass_action(cl)
|
|
||||||
|
|
||||||
else:
|
|
||||||
raise AuthException(
|
|
||||||
u"Your authentication credentials for " + self.provider.name + " are incorrect, check your config")
|
|
||||||
except AuthException, e:
|
|
||||||
logger.log(u"Authentication error: " + ex(e), logger.ERROR)
|
|
||||||
except Exception, e:
|
|
||||||
logger.log(u"Error while searching " + self.provider.name + ", skipping: " + ex(e), logger.ERROR)
|
|
||||||
logger.log(traceback.format_exc(), logger.DEBUG)
|
|
||||||
|
|
||||||
# overwrite method with that parses the rageid from the newznab feed
|
|
||||||
def _parseItem(self, item):
|
def _parseItem(self, item):
|
||||||
title = item.title
|
(title, url) = self._get_title_and_url(item)
|
||||||
url = item.link
|
|
||||||
|
|
||||||
attrs = item.newznab_attr
|
attrs = item.newznab_attr
|
||||||
if not isinstance(attrs, list):
|
if not isinstance(attrs, list):
|
||||||
|
@ -320,7 +320,7 @@ class NextGenCache(tvcache.TVCache):
|
|||||||
|
|
||||||
def _getRSSData(self):
|
def _getRSSData(self):
|
||||||
search_params = {'RSS': ['']}
|
search_params = {'RSS': ['']}
|
||||||
return self.provider._doSearch(search_params)
|
return {'entries': self.provider._doSearch(search_params)}
|
||||||
|
|
||||||
|
|
||||||
provider = NextGenProvider()
|
provider = NextGenProvider()
|
||||||
|
@ -80,8 +80,7 @@ class NyaaProvider(generic.TorrentProvider):
|
|||||||
logger.log(u"Search string: " + searchURL, logger.DEBUG)
|
logger.log(u"Search string: " + searchURL, logger.DEBUG)
|
||||||
|
|
||||||
results = []
|
results = []
|
||||||
for curItem in self.cache.getRSSFeed(searchURL):
|
for curItem in self.cache.getRSSFeed(searchURL, items=['entries']) or []:
|
||||||
|
|
||||||
(title, url) = self._get_title_and_url(curItem)
|
(title, url) = self._get_title_and_url(curItem)
|
||||||
|
|
||||||
if title and url:
|
if title and url:
|
||||||
@ -126,6 +125,6 @@ class NyaaCache(tvcache.TVCache):
|
|||||||
|
|
||||||
logger.log(u"NyaaTorrents cache update URL: " + url, logger.DEBUG)
|
logger.log(u"NyaaTorrents cache update URL: " + url, logger.DEBUG)
|
||||||
|
|
||||||
return self.getRSSFeed(url)
|
return self.getRSSFeed(url, items=['entries', 'feed'])
|
||||||
|
|
||||||
provider = NyaaProvider()
|
provider = NyaaProvider()
|
||||||
|
@ -184,6 +184,6 @@ class OmgwtfnzbsCache(tvcache.TVCache):
|
|||||||
|
|
||||||
logger.log(self.provider.name + u" cache update URL: " + rss_url, logger.DEBUG)
|
logger.log(self.provider.name + u" cache update URL: " + rss_url, logger.DEBUG)
|
||||||
|
|
||||||
return self.getRSSFeed(rss_url)
|
return self.getRSSFeed(rss_url, items=['entries', 'feed'])
|
||||||
|
|
||||||
provider = OmgwtfnzbsProvider()
|
provider = OmgwtfnzbsProvider()
|
||||||
|
@ -107,11 +107,11 @@ class TorrentRssProvider(generic.TorrentProvider):
|
|||||||
if not cookie_validator.match(self.cookies):
|
if not cookie_validator.match(self.cookies):
|
||||||
return (False, 'Cookie is not correctly formatted: ' + self.cookies)
|
return (False, 'Cookie is not correctly formatted: ' + self.cookies)
|
||||||
|
|
||||||
data = self.cache._getRSSData()
|
data = self.cache._getRSSData()['entries']
|
||||||
if not data or not len(data) > 0:
|
if not data:
|
||||||
return (False, 'No items found in the RSS feed ' + self.url)
|
return (False, 'No items found in the RSS feed ' + self.url)
|
||||||
|
|
||||||
(title, url) = self._get_title_and_url(data.entries[0])
|
(title, url) = self._get_title_and_url(data[0])
|
||||||
|
|
||||||
if not title:
|
if not title:
|
||||||
return (False, 'Unable to get title from first item')
|
return (False, 'Unable to get title from first item')
|
||||||
@ -168,4 +168,4 @@ class TorrentRssCache(tvcache.TVCache):
|
|||||||
if self.provider.cookies:
|
if self.provider.cookies:
|
||||||
request_headers = {'Cookie': self.provider.cookies}
|
request_headers = {'Cookie': self.provider.cookies}
|
||||||
|
|
||||||
return self.getRSSFeed(self.provider.url, request_headers=request_headers)
|
return self.getRSSFeed(self.provider.url, request_headers=request_headers, items=['entries', 'feed'])
|
@ -305,6 +305,6 @@ class SCCCache(tvcache.TVCache):
|
|||||||
|
|
||||||
def _getRSSData(self):
|
def _getRSSData(self):
|
||||||
search_params = {'RSS': ['']}
|
search_params = {'RSS': ['']}
|
||||||
return self.provider._doSearch(search_params)
|
return {'entries': self.provider._doSearch(search_params)}
|
||||||
|
|
||||||
provider = SCCProvider()
|
provider = SCCProvider()
|
||||||
|
@ -254,7 +254,7 @@ class SpeedCDCache(tvcache.TVCache):
|
|||||||
|
|
||||||
def _getRSSData(self):
|
def _getRSSData(self):
|
||||||
search_params = {'RSS': ['']}
|
search_params = {'RSS': ['']}
|
||||||
return self.provider._doSearch(search_params)
|
return {'entries': self.provider._doSearch(search_params)}
|
||||||
|
|
||||||
provider = SpeedCDProvider()
|
provider = SpeedCDProvider()
|
||||||
|
|
||||||
|
@ -260,7 +260,7 @@ class T411Cache(tvcache.TVCache):
|
|||||||
|
|
||||||
def _getRSSData(self):
|
def _getRSSData(self):
|
||||||
search_params = {'RSS': ['']}
|
search_params = {'RSS': ['']}
|
||||||
return self.provider._doSearch(search_params)
|
return {'entries': self.provider._doSearch(search_params)}
|
||||||
|
|
||||||
|
|
||||||
provider = T411Provider()
|
provider = T411Provider()
|
||||||
|
@ -340,7 +340,7 @@ class ThePirateBayCache(tvcache.TVCache):
|
|||||||
|
|
||||||
def _getRSSData(self):
|
def _getRSSData(self):
|
||||||
search_params = {'RSS': ['rss']}
|
search_params = {'RSS': ['rss']}
|
||||||
return self.provider._doSearch(search_params)
|
return {'entries': self.provider._doSearch(search_params)}
|
||||||
|
|
||||||
class ThePirateBayWebproxy:
|
class ThePirateBayWebproxy:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
|
@ -164,7 +164,7 @@ class TokyoToshokanCache(tvcache.TVCache):
|
|||||||
|
|
||||||
logger.log(u"TokyoToshokan cache update URL: " + url, logger.DEBUG)
|
logger.log(u"TokyoToshokan cache update URL: " + url, logger.DEBUG)
|
||||||
|
|
||||||
return self.getRSSFeed(url)
|
return self.getRSSFeed(url, items=['entries', 'feed'])
|
||||||
|
|
||||||
|
|
||||||
provider = TokyoToshokanProvider()
|
provider = TokyoToshokanProvider()
|
||||||
|
@ -276,7 +276,7 @@ class TorrentBytesCache(tvcache.TVCache):
|
|||||||
|
|
||||||
def _getRSSData(self):
|
def _getRSSData(self):
|
||||||
search_params = {'RSS': ['']}
|
search_params = {'RSS': ['']}
|
||||||
return self.provider._doSearch(search_params)
|
return {'entries': self.provider._doSearch(search_params)}
|
||||||
|
|
||||||
|
|
||||||
provider = TorrentBytesProvider()
|
provider = TorrentBytesProvider()
|
||||||
|
@ -282,8 +282,6 @@ class TorrentDayCache(tvcache.TVCache):
|
|||||||
|
|
||||||
def _getRSSData(self):
|
def _getRSSData(self):
|
||||||
search_params = {'RSS': ['']}
|
search_params = {'RSS': ['']}
|
||||||
return self.provider._doSearch(search_params)
|
return {'entries': self.provider._doSearch(search_params)}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
provider = TorrentDayProvider()
|
provider = TorrentDayProvider()
|
||||||
|
@ -277,7 +277,7 @@ class TorrentLeechCache(tvcache.TVCache):
|
|||||||
|
|
||||||
def _getRSSData(self):
|
def _getRSSData(self):
|
||||||
search_params = {'RSS': ['']}
|
search_params = {'RSS': ['']}
|
||||||
return self.provider._doSearch(search_params)
|
return {'entries': self.provider._doSearch(search_params)}
|
||||||
|
|
||||||
|
|
||||||
provider = TorrentLeechProvider()
|
provider = TorrentLeechProvider()
|
||||||
|
@ -60,11 +60,12 @@ class TvTorrentsProvider(generic.TorrentProvider):
|
|||||||
return True
|
return True
|
||||||
|
|
||||||
def _checkAuthFromData(self, data):
|
def _checkAuthFromData(self, data):
|
||||||
if not data:
|
if not (data.entries and data.feed):
|
||||||
return self._checkAuth()
|
return self._checkAuth()
|
||||||
|
|
||||||
if "User can't be found" in data.feed.get('title', None) or "Invalid Hash" in data.feed.get('title', None):
|
title = data.feed.get('title', None)
|
||||||
logger.log(u"Incorrect authentication credentials for " + self.name + " : " + str(data.feed.title),
|
if "User can't be found" in title or "Invalid Hash" in title:
|
||||||
|
logger.log(u"Incorrect authentication credentials for " + self.name + " : " + str(title),
|
||||||
logger.DEBUG)
|
logger.DEBUG)
|
||||||
raise AuthException(
|
raise AuthException(
|
||||||
u"Your authentication credentials for " + self.name + " are incorrect, check your config")
|
u"Your authentication credentials for " + self.name + " are incorrect, check your config")
|
||||||
@ -89,7 +90,7 @@ class TvTorrentsCache(tvcache.TVCache):
|
|||||||
rss_url = self.provider.url + 'RssServlet?digest=' + provider.digest + '&hash=' + provider.hash + '&fname=true&exclude=(' + ignore_regex + ')'
|
rss_url = self.provider.url + 'RssServlet?digest=' + provider.digest + '&hash=' + provider.hash + '&fname=true&exclude=(' + ignore_regex + ')'
|
||||||
logger.log(self.provider.name + u" cache update URL: " + rss_url, logger.DEBUG)
|
logger.log(self.provider.name + u" cache update URL: " + rss_url, logger.DEBUG)
|
||||||
|
|
||||||
return self.getRSSFeed(rss_url)
|
return self.getRSSFeed(rss_url, items=['entries', 'feed'])
|
||||||
|
|
||||||
def _checkAuth(self, data):
|
def _checkAuth(self, data):
|
||||||
return self.provider._checkAuthFromData(data)
|
return self.provider._checkAuthFromData(data)
|
||||||
|
@ -43,30 +43,31 @@ class WombleCache(tvcache.TVCache):
|
|||||||
self.minTime = 15
|
self.minTime = 15
|
||||||
|
|
||||||
def updateCache(self):
|
def updateCache(self):
|
||||||
|
# check if we should update
|
||||||
# delete anything older then 7 days
|
|
||||||
self._clearCache()
|
|
||||||
|
|
||||||
if not self.shouldUpdate():
|
if not self.shouldUpdate():
|
||||||
return
|
return
|
||||||
|
|
||||||
|
# clear cache
|
||||||
|
self._clearCache()
|
||||||
|
|
||||||
|
# set updated
|
||||||
|
self.setLastUpdate()
|
||||||
|
|
||||||
cl = []
|
cl = []
|
||||||
for url in [self.provider.url + 'rss/?sec=tv-sd&fr=false', self.provider.url + 'rss/?sec=tv-hd&fr=false']:
|
for url in [self.provider.url + 'rss/?sec=tv-sd&fr=false', self.provider.url + 'rss/?sec=tv-hd&fr=false']:
|
||||||
logger.log(u"Womble's Index cache update URL: " + url, logger.DEBUG)
|
logger.log(u"Womble's Index cache update URL: " + url, logger.DEBUG)
|
||||||
|
|
||||||
# By now we know we've got data and no auth errors, all we need to do is put it in the database
|
for item in self.getRSSFeed(url, items=['entries', 'feed'])['entries'] or []:
|
||||||
for item in self.getRSSFeed(url).get('entries', []):
|
ci = self._parseItem(item)
|
||||||
ci = self._parseItem(item.title, item.url)
|
|
||||||
if ci is not None:
|
if ci is not None:
|
||||||
cl.append(ci)
|
cl.append(ci)
|
||||||
|
|
||||||
if len(cl) > 0:
|
if len(cl) > 0:
|
||||||
myDB = self._getDB()
|
myDB = self._getDB()
|
||||||
myDB.mass_action(cl)
|
myDB.mass_action(cl)
|
||||||
self.setLastUpdate()
|
|
||||||
|
|
||||||
def _checkAuth(self, data):
|
def _checkAuth(self, data):
|
||||||
return data.feed.get('title', None) != 'Invalid Link'
|
return data if data.feed.title != 'Invalid Link' else None
|
||||||
|
|
||||||
provider = WombleProvider()
|
provider = WombleProvider()
|
||||||
|
|
||||||
|
@ -33,7 +33,7 @@ class RSSFeeds:
|
|||||||
finally:
|
finally:
|
||||||
self.rssDB.close()
|
self.rssDB.close()
|
||||||
|
|
||||||
def getFeed(self, url, post_data=None, request_headers=None):
|
def getFeed(self, url, post_data=None, request_headers=None, items=[]):
|
||||||
parsed = list(urlparse.urlparse(url))
|
parsed = list(urlparse.urlparse(url))
|
||||||
parsed[2] = re.sub("/{2,}", "/", parsed[2]) # replace two or more / with one
|
parsed[2] = re.sub("/{2,}", "/", parsed[2]) # replace two or more / with one
|
||||||
|
|
||||||
@ -42,8 +42,15 @@ class RSSFeeds:
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
fc = Cache(self.rssDB)
|
fc = Cache(self.rssDB)
|
||||||
feed = fc.fetch(url, False, False, request_headers)
|
resp = fc.fetch(url, False, False, request_headers)
|
||||||
|
|
||||||
return feed
|
data = {}
|
||||||
|
for item in items:
|
||||||
|
try:
|
||||||
|
data[item] = resp[item]
|
||||||
|
except:
|
||||||
|
data[item] = None
|
||||||
|
|
||||||
|
return data
|
||||||
finally:
|
finally:
|
||||||
self.rssDB.close()
|
self.rssDB.close()
|
@ -375,7 +375,6 @@ def searchForNeededEpisodes():
|
|||||||
|
|
||||||
providers = [x for x in sickbeard.providers.sortedProviderList(sickbeard.RANDOMIZE_PROVIDERS) if x.isActive() and x.enable_daily]
|
providers = [x for x in sickbeard.providers.sortedProviderList(sickbeard.RANDOMIZE_PROVIDERS) if x.isActive() and x.enable_daily]
|
||||||
for curProvider in providers:
|
for curProvider in providers:
|
||||||
# spawn separate threads for each provider so we don't need to wait for providers with slow network operation
|
|
||||||
threads += [threading.Thread(target=curProvider.cache.updateCache, name=origThreadName + " :: [" + curProvider.name + "]")]
|
threads += [threading.Thread(target=curProvider.cache.updateCache, name=origThreadName + " :: [" + curProvider.name + "]")]
|
||||||
|
|
||||||
# start the thread we just created
|
# start the thread we just created
|
||||||
@ -393,13 +392,6 @@ def searchForNeededEpisodes():
|
|||||||
|
|
||||||
# pick a single result for each episode, respecting existing results
|
# pick a single result for each episode, respecting existing results
|
||||||
for curEp in curFoundResults:
|
for curEp in curFoundResults:
|
||||||
|
|
||||||
if curEp.show.paused:
|
|
||||||
logger.log(
|
|
||||||
u"Show " + curEp.show.name + " is paused, ignoring all RSS items for " + curEp.prettyName(),
|
|
||||||
logger.DEBUG)
|
|
||||||
continue
|
|
||||||
|
|
||||||
# find the best result for the current episode
|
# find the best result for the current episode
|
||||||
bestResult = None
|
bestResult = None
|
||||||
for curResult in curFoundResults[curEp]:
|
for curResult in curFoundResults[curEp]:
|
||||||
@ -442,6 +434,7 @@ def searchProviders(show, episodes, manualSearch=False):
|
|||||||
finalResults = []
|
finalResults = []
|
||||||
|
|
||||||
didSearch = False
|
didSearch = False
|
||||||
|
threads = []
|
||||||
|
|
||||||
# build name cache for show
|
# build name cache for show
|
||||||
sickbeard.name_cache.buildNameCache(show)
|
sickbeard.name_cache.buildNameCache(show)
|
||||||
@ -449,6 +442,18 @@ def searchProviders(show, episodes, manualSearch=False):
|
|||||||
origThreadName = threading.currentThread().name
|
origThreadName = threading.currentThread().name
|
||||||
|
|
||||||
providers = [x for x in sickbeard.providers.sortedProviderList(sickbeard.RANDOMIZE_PROVIDERS) if x.isActive() and x.enable_backlog]
|
providers = [x for x in sickbeard.providers.sortedProviderList(sickbeard.RANDOMIZE_PROVIDERS) if x.isActive() and x.enable_backlog]
|
||||||
|
for curProvider in providers:
|
||||||
|
threads += [threading.Thread(target=curProvider.cache.updateCache,
|
||||||
|
name=origThreadName + " :: [" + curProvider.name + "]")]
|
||||||
|
|
||||||
|
# start the thread we just created
|
||||||
|
for t in threads:
|
||||||
|
t.start()
|
||||||
|
|
||||||
|
# wait for all threads to finish
|
||||||
|
for t in threads:
|
||||||
|
t.join()
|
||||||
|
|
||||||
for providerNum, curProvider in enumerate(providers):
|
for providerNum, curProvider in enumerate(providers):
|
||||||
if curProvider.anime_only and not show.is_anime:
|
if curProvider.anime_only and not show.is_anime:
|
||||||
logger.log(u"" + str(show.name) + " is not an anime, skiping", logger.DEBUG)
|
logger.log(u"" + str(show.name) + " is not an anime, skiping", logger.DEBUG)
|
||||||
@ -470,7 +475,6 @@ def searchProviders(show, episodes, manualSearch=False):
|
|||||||
logger.log(u"Performing season pack search for " + show.name)
|
logger.log(u"Performing season pack search for " + show.name)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
curProvider.cache.updateCache()
|
|
||||||
searchResults = curProvider.findSearchResults(show, episodes, search_mode, manualSearch)
|
searchResults = curProvider.findSearchResults(show, episodes, search_mode, manualSearch)
|
||||||
except exceptions.AuthException, e:
|
except exceptions.AuthException, e:
|
||||||
logger.log(u"Authentication error: " + ex(e), logger.ERROR)
|
logger.log(u"Authentication error: " + ex(e), logger.ERROR)
|
||||||
|
@ -96,12 +96,10 @@ class TVCache():
|
|||||||
myDB.action("DELETE FROM [" + self.providerID + "] WHERE 1")
|
myDB.action("DELETE FROM [" + self.providerID + "] WHERE 1")
|
||||||
|
|
||||||
def _get_title_and_url(self, item):
|
def _get_title_and_url(self, item):
|
||||||
# override this in the provider if daily search has a different data layout to backlog searches
|
|
||||||
return self.provider._get_title_and_url(item)
|
return self.provider._get_title_and_url(item)
|
||||||
|
|
||||||
def _getRSSData(self):
|
def _getRSSData(self):
|
||||||
data = None
|
return None
|
||||||
return data
|
|
||||||
|
|
||||||
def _checkAuth(self, data):
|
def _checkAuth(self, data):
|
||||||
return True
|
return True
|
||||||
@ -110,49 +108,37 @@ class TVCache():
|
|||||||
return True
|
return True
|
||||||
|
|
||||||
def updateCache(self):
|
def updateCache(self):
|
||||||
|
# check if we should update
|
||||||
if not self.shouldUpdate():
|
if not self.shouldUpdate():
|
||||||
return
|
return
|
||||||
|
|
||||||
try:
|
try:
|
||||||
if self._checkAuth(None):
|
data = self._getRSSData()
|
||||||
data = self._getRSSData()
|
if self._checkAuth(data):
|
||||||
if not data or not len(data) > 0:
|
|
||||||
return
|
|
||||||
|
|
||||||
# clear cache
|
# clear cache
|
||||||
self._clearCache()
|
self._clearCache()
|
||||||
|
|
||||||
# set updated
|
# set updated
|
||||||
self.setLastUpdate()
|
self.setLastUpdate()
|
||||||
|
|
||||||
try:
|
cl = []
|
||||||
items = data.get('entries', [])
|
for item in data['entries']:
|
||||||
except:
|
ci = self._parseItem(item)
|
||||||
items = data
|
if ci is not None:
|
||||||
|
cl.append(ci)
|
||||||
|
|
||||||
if self._checkAuth(items):
|
if len(cl) > 0:
|
||||||
cl = []
|
myDB = self._getDB()
|
||||||
for item in items:
|
myDB.mass_action(cl)
|
||||||
title, url = self._get_title_and_url(item)
|
|
||||||
ci = self._parseItem(title, url)
|
|
||||||
if ci is not None:
|
|
||||||
cl.append(ci)
|
|
||||||
|
|
||||||
if len(cl) > 0:
|
|
||||||
myDB = self._getDB()
|
|
||||||
myDB.mass_action(cl)
|
|
||||||
|
|
||||||
else:
|
|
||||||
raise AuthException(
|
|
||||||
u"Your authentication credentials for " + self.provider.name + " are incorrect, check your config")
|
|
||||||
except AuthException, e:
|
except AuthException, e:
|
||||||
logger.log(u"Authentication error: " + ex(e), logger.ERROR)
|
logger.log(u"Authentication error: " + ex(e), logger.ERROR)
|
||||||
except Exception, e:
|
except Exception, e:
|
||||||
logger.log(u"Error while searching " + self.provider.name + ", skipping: " + ex(e), logger.ERROR)
|
logger.log(u"Error while searching " + self.provider.name + ", skipping: " + ex(e), logger.ERROR)
|
||||||
logger.log(traceback.format_exc(), logger.DEBUG)
|
logger.log(traceback.format_exc(), logger.DEBUG)
|
||||||
|
|
||||||
def getRSSFeed(self, url, post_data=None, request_headers=None):
|
def getRSSFeed(self, url, post_data=None, request_headers=None, items=[]):
|
||||||
return RSSFeeds(self.providerID).getFeed(url, post_data, request_headers)
|
return RSSFeeds(self.providerID).getFeed(url, post_data, request_headers, items)
|
||||||
|
|
||||||
def _translateTitle(self, title):
|
def _translateTitle(self, title):
|
||||||
return u'' + title.replace(' ', '.')
|
return u'' + title.replace(' ', '.')
|
||||||
@ -160,7 +146,8 @@ class TVCache():
|
|||||||
def _translateLinkURL(self, url):
|
def _translateLinkURL(self, url):
|
||||||
return url.replace('&', '&')
|
return url.replace('&', '&')
|
||||||
|
|
||||||
def _parseItem(self, title, url):
|
def _parseItem(self, item):
|
||||||
|
title, url = self._get_title_and_url(item)
|
||||||
|
|
||||||
self._checkItemAuth(title, url)
|
self._checkItemAuth(title, url)
|
||||||
|
|
||||||
|
File diff suppressed because it is too large
Load Diff
@ -12,7 +12,7 @@ from sickbeard.helpers import create_https_certificates
|
|||||||
from tornado.web import Application, StaticFileHandler, RedirectHandler, HTTPError
|
from tornado.web import Application, StaticFileHandler, RedirectHandler, HTTPError
|
||||||
from tornado.httpserver import HTTPServer
|
from tornado.httpserver import HTTPServer
|
||||||
from tornado.ioloop import IOLoop
|
from tornado.ioloop import IOLoop
|
||||||
|
from tornado.routes import route
|
||||||
|
|
||||||
class MultiStaticFileHandler(StaticFileHandler):
|
class MultiStaticFileHandler(StaticFileHandler):
|
||||||
def initialize(self, paths, default_filename=None):
|
def initialize(self, paths, default_filename=None):
|
||||||
@ -62,8 +62,8 @@ class SRWebServer(threading.Thread):
|
|||||||
self.video_root = None
|
self.video_root = None
|
||||||
|
|
||||||
# web root
|
# web root
|
||||||
self.options['web_root'] = ('/' + self.options['web_root'].lstrip('/')) if self.options[
|
self.options['web_root'] = ('/' + self.options['web_root'].lstrip('/')) if self.options['web_root'] else '/'
|
||||||
'web_root'] else ''
|
sickbeard.WEB_ROOT = self.options['web_root'].strip('/')
|
||||||
|
|
||||||
# tornado setup
|
# tornado setup
|
||||||
self.enable_https = self.options['enable_https']
|
self.enable_https = self.options['enable_https']
|
||||||
@ -90,33 +90,31 @@ class SRWebServer(threading.Thread):
|
|||||||
autoreload=False,
|
autoreload=False,
|
||||||
gzip=True,
|
gzip=True,
|
||||||
xheaders=sickbeard.HANDLE_REVERSE_PROXY,
|
xheaders=sickbeard.HANDLE_REVERSE_PROXY,
|
||||||
cookie_secret='61oETzKXQAGaYdkL5gEmGeJJFuYh7EQnp2XdTP1o/Vo='
|
cookie_secret='61oETzKXQAGaYdkL5gEmGeJJFuYh7EQnp2XdTP1o/Vo=',
|
||||||
|
username=self.options['username'],
|
||||||
|
password=self.options['password'],
|
||||||
)
|
)
|
||||||
|
|
||||||
# Main Handler
|
# Main Handlers
|
||||||
self.app.add_handlers(".*$", [
|
self.app.add_handlers(".*$", [] + route.get_routes())
|
||||||
(r'%s/api/(.*)(/?)' % self.options['web_root'], webapi.Api),
|
|
||||||
(r'%s/(.*)(/?)' % self.options['web_root'], webserve.MainHandler),
|
|
||||||
(r'(.*)', webserve.MainHandler)
|
|
||||||
])
|
|
||||||
|
|
||||||
# Static Path Handler
|
# Static Path Handlers
|
||||||
self.app.add_handlers(".*$", [
|
self.app.add_handlers(".*$", [
|
||||||
(r'%s/(favicon\.ico)' % self.options['web_root'], MultiStaticFileHandler,
|
(r'%s(favicon\.ico)' % self.options['web_root'], MultiStaticFileHandler,
|
||||||
{'paths': [os.path.join(self.options['data_root'], 'images/ico/favicon.ico')]}),
|
{'paths': [os.path.join(self.options['data_root'], 'images/ico/favicon.ico')]}),
|
||||||
(r'%s/%s/(.*)(/?)' % (self.options['web_root'], 'images'), MultiStaticFileHandler,
|
(r'%s%s/(.*)(/?)' % (self.options['web_root'], 'images'), MultiStaticFileHandler,
|
||||||
{'paths': [os.path.join(self.options['data_root'], 'images'),
|
{'paths': [os.path.join(self.options['data_root'], 'images'),
|
||||||
os.path.join(sickbeard.CACHE_DIR, 'images')]}),
|
os.path.join(sickbeard.CACHE_DIR, 'images')]}),
|
||||||
(r'%s/%s/(.*)(/?)' % (self.options['web_root'], 'css'), MultiStaticFileHandler,
|
(r'%s%s/(.*)(/?)' % (self.options['web_root'], 'css'), MultiStaticFileHandler,
|
||||||
{'paths': [os.path.join(self.options['data_root'], 'css')]}),
|
{'paths': [os.path.join(self.options['data_root'], 'css')]}),
|
||||||
(r'%s/%s/(.*)(/?)' % (self.options['web_root'], 'js'), MultiStaticFileHandler,
|
(r'%s%s/(.*)(/?)' % (self.options['web_root'], 'js'), MultiStaticFileHandler,
|
||||||
{'paths': [os.path.join(self.options['data_root'], 'js')]}),
|
{'paths': [os.path.join(self.options['data_root'], 'js')]}),
|
||||||
])
|
])
|
||||||
|
|
||||||
# Static Videos Path
|
# Static Videos Path
|
||||||
if self.video_root:
|
if self.video_root:
|
||||||
self.app.add_handlers(".*$", [
|
self.app.add_handlers(".*$", [
|
||||||
(r'%s/%s/(.*)' % (self.options['web_root'], 'videos'), MultiStaticFileHandler,
|
(r'%s%s/(.*)' % (self.options['web_root'], 'videos'), MultiStaticFileHandler,
|
||||||
{'paths': [self.video_root]}),
|
{'paths': [self.video_root]}),
|
||||||
])
|
])
|
||||||
|
|
||||||
|
22
tornado/routes.py
Normal file
22
tornado/routes.py
Normal file
@ -0,0 +1,22 @@
|
|||||||
|
import tornado.web
|
||||||
|
|
||||||
|
class route(object):
|
||||||
|
_routes = []
|
||||||
|
|
||||||
|
def __init__(self, uri, name=None):
|
||||||
|
self._uri = uri
|
||||||
|
self.name = name
|
||||||
|
|
||||||
|
def __call__(self, _handler):
|
||||||
|
"""gets called when we class decorate"""
|
||||||
|
name = self.name and self.name or _handler.__name__
|
||||||
|
self._routes.append(tornado.web.url(self._uri, _handler, name=name))
|
||||||
|
return _handler
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def get_routes(self):
|
||||||
|
self._routes.reverse()
|
||||||
|
return self._routes
|
||||||
|
|
||||||
|
def route_redirect(from_, to, name=None):
|
||||||
|
route._routes.append(tornado.web.url(from_, tornado.web.RedirectHandler, dict(url=to), name=name))
|
Loading…
Reference in New Issue
Block a user