You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
116 lines
4.1 KiB
116 lines
4.1 KiB
from couchpotato.core.logger import CPLog
|
|
from couchpotato.core.providers.base import NZBProvider
|
|
|
|
log = CPLog(__name__)
|
|
|
|
|
|
class NZBMatrix(NZBProvider):
|
|
|
|
urls = {
|
|
'download': 'https://api.nzbmatrix.com/v1.1/download.php?id=%s%s',
|
|
'detail': 'https://nzbmatrix.com/nzb-details.php?id=%s&hit=1',
|
|
'search': 'http://rss.nzbmatrix.com/rss.php',
|
|
}
|
|
|
|
cat_ids = [
|
|
([42], ['720p', '1080p']),
|
|
([2], ['cam', 'ts', 'dvdrip', 'tc', 'r5', 'scr']),
|
|
([54], ['brrip']),
|
|
([1], ['dvdr']),
|
|
]
|
|
cat_backup_id = 2
|
|
|
|
def __init__(self, config):
|
|
log.info('Using NZBMatrix provider')
|
|
|
|
self.config = config
|
|
|
|
def find(self, movie, quality, type, retry = False):
|
|
|
|
self.cleanCache();
|
|
|
|
results = []
|
|
if not self.enabled() or not self.isAvailable(self.searchUrl):
|
|
return results
|
|
|
|
catId = self.getCatId(type)
|
|
arguments = urlencode({
|
|
'term': movie.imdb,
|
|
'subcat': catId,
|
|
'username': self.conf('username'),
|
|
'apikey': self.conf('apikey'),
|
|
'searchin': 'weblink',
|
|
'english': 1 if self.conf('english') else 0,
|
|
})
|
|
url = "%s?%s" % (self.searchUrl, arguments)
|
|
cacheId = str(movie.imdb) + '-' + str(catId)
|
|
singleCat = (len(self.catIds.get(catId)) == 1 and catId != self.catBackupId)
|
|
|
|
try:
|
|
cached = False
|
|
if(self.cache.get(cacheId)):
|
|
data = True
|
|
cached = True
|
|
log.info('Getting RSS from cache: %s.' % cacheId)
|
|
else:
|
|
log.info('Searching: %s' % url)
|
|
data = self.urlopen(url)
|
|
self.cache[cacheId] = {
|
|
'time': time.time()
|
|
}
|
|
|
|
except (IOError, URLError):
|
|
log.error('Failed to open %s.' % url)
|
|
return results
|
|
|
|
if data:
|
|
try:
|
|
try:
|
|
if cached:
|
|
xml = self.cache[cacheId]['xml']
|
|
else:
|
|
xml = self.getItems(data)
|
|
self.cache[cacheId]['xml'] = xml
|
|
except:
|
|
log.debug('No valid xml or to many requests.. You never know with %s.' % self.name)
|
|
return results
|
|
|
|
for nzb in xml:
|
|
|
|
title = self.gettextelement(nzb, "title")
|
|
if 'error' in title.lower(): continue
|
|
|
|
id = int(self.gettextelement(nzb, "link").split('&')[0].partition('id=')[2])
|
|
size = self.gettextelement(nzb, "description").split('<br /><b>')[2].split('> ')[1]
|
|
date = str(self.gettextelement(nzb, "description").split('<br /><b>')[3].partition('Added:</b> ')[2])
|
|
|
|
new = self.feedItem()
|
|
new.id = id
|
|
new.type = 'nzb'
|
|
new.name = title
|
|
new.date = int(time.mktime(parse(date).timetuple()))
|
|
new.size = self.parseSize(size)
|
|
new.url = self.downloadLink(id)
|
|
new.detailUrl = self.detailLink(id)
|
|
new.content = self.gettextelement(nzb, "description")
|
|
new.score = self.calcScore(new, movie)
|
|
new.checkNZB = True
|
|
|
|
if new.date > time.time() - (int(self.config.get('NZB', 'retention')) * 24 * 60 * 60):
|
|
if self.isCorrectMovie(new, movie, type, imdbResults = True, singleCategory = singleCat):
|
|
results.append(new)
|
|
log.info('Found: %s' % new.name)
|
|
else:
|
|
log.info('Found outside retention: %s' % new.name)
|
|
|
|
return results
|
|
except SyntaxError:
|
|
log.error('Failed to parse XML response from NZBMatrix.com')
|
|
|
|
return results
|
|
|
|
def getApiExt(self):
|
|
return '&username=%s&apikey=%s' % (self.conf('username'), self.conf('apikey'))
|
|
|
|
def isEnabled(self):
|
|
return self.conf('enabled') and self.conf('username') and self.conf('apikey')
|
|
|