mirror of
https://github.com/moparisthebest/SickRage
synced 2024-11-11 03:45:01 -05:00
122 lines
4.2 KiB
Python
122 lines
4.2 KiB
Python
# Author: Nic Wolfe <nic@wolfeden.ca>
|
|
# URL: http://code.google.com/p/sickbeard/
|
|
#
|
|
# This file is part of Sick Beard.
|
|
#
|
|
# Sick Beard is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Sick Beard is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with Sick Beard. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import urllib
|
|
import generic
|
|
import sickbeard
|
|
|
|
try:
|
|
import xml.etree.cElementTree as etree
|
|
except ImportError:
|
|
import xml.etree.ElementTree as etree
|
|
|
|
from sickbeard import exceptions, logger
|
|
from sickbeard import tvcache, show_name_helpers
|
|
|
|
|
|
class NZBsRUSProvider(generic.NZBProvider):
|
|
def __init__(self):
|
|
generic.NZBProvider.__init__(self, "NZBs'R'US")
|
|
self.cache = NZBsRUSCache(self)
|
|
self.url = 'https://www.nzbsrus.com/'
|
|
self.supportsBacklog = True
|
|
|
|
def isEnabled(self):
|
|
return sickbeard.NZBSRUS
|
|
|
|
def _checkAuth(self):
|
|
if sickbeard.NZBSRUS_UID in (None, "") or sickbeard.NZBSRUS_HASH in (None, ""):
|
|
raise exceptions.AuthException("NZBs'R'US authentication details are empty, check your config")
|
|
|
|
def _get_season_search_strings(self, show, season, wantedEp, searchSeason=False):
|
|
return ['^' + x for x in show_name_helpers.makeSceneSeasonSearchString(show, season)]
|
|
|
|
def _get_episode_search_strings(self, ep_obj):
|
|
return ['^' + x for x in show_name_helpers.makeSceneSearchString(ep_obj)]
|
|
|
|
def _doSearch(self, search, show=None, age=None):
|
|
params = {'uid': sickbeard.NZBSRUS_UID,
|
|
'key': sickbeard.NZBSRUS_HASH,
|
|
'xml': 1,
|
|
'age': sickbeard.USENET_RETENTION,
|
|
'lang0': 1, # English only from CouchPotato
|
|
'lang1': 1,
|
|
'lang3': 1,
|
|
'c91': 1, # TV:HD
|
|
'c104': 1, # TV:SD-x264
|
|
'c75': 1, # TV:XviD
|
|
'searchtext': search}
|
|
|
|
if not params['age']:
|
|
params['age'] = 500
|
|
|
|
searchURL = self.url + 'api.php?' + urllib.urlencode(params)
|
|
logger.log(u"NZBS'R'US search url: " + searchURL, logger.DEBUG)
|
|
|
|
data = self.getURL(searchURL)
|
|
if not data:
|
|
return []
|
|
|
|
if not data.startswith('<?xml'): # Error will be a single line of text
|
|
logger.log(u"NZBs'R'US error: " + data, logger.ERROR)
|
|
return []
|
|
|
|
root = etree.fromstring(data)
|
|
if root is None:
|
|
logger.log(u"Error trying to parse NZBS'R'US XML data.", logger.ERROR)
|
|
logger.log(u"RSS data: " + data, logger.DEBUG)
|
|
return []
|
|
return root.findall('./results/result')
|
|
|
|
def _get_title_and_url(self, element):
|
|
if element.find('title'): # RSS feed
|
|
title = element.find('title').text
|
|
url = element.find('link').text.replace('&', '&')
|
|
else: # API item
|
|
title = element.find('name').text
|
|
nzbID = element.find('id').text
|
|
key = element.find('key').text
|
|
url = self.url + 'nzbdownload_rss.php' + '/' + \
|
|
nzbID + '/' + sickbeard.NZBSRUS_UID + '/' + key + '/'
|
|
return (title, url)
|
|
|
|
|
|
class NZBsRUSCache(tvcache.TVCache):
|
|
def __init__(self, provider):
|
|
tvcache.TVCache.__init__(self, provider)
|
|
# only poll NZBs'R'US every 15 minutes max
|
|
self.minTime = 15
|
|
|
|
def _getRSSData(self):
|
|
url = self.provider.url + 'rssfeed.php?'
|
|
urlArgs = {'cat': '91,75,104', # HD,XviD,SD-x264
|
|
'i': sickbeard.NZBSRUS_UID,
|
|
'h': sickbeard.NZBSRUS_HASH}
|
|
|
|
url += urllib.urlencode(urlArgs)
|
|
logger.log(u"NZBs'R'US cache update URL: " + url, logger.DEBUG)
|
|
|
|
data = self.provider.getURL(url)
|
|
return data
|
|
|
|
def _checkAuth(self, data):
|
|
return data != 'Invalid Link'
|
|
|
|
|
|
provider = NZBsRUSProvider()
|