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.
239 lines
8.2 KiB
239 lines
8.2 KiB
12 years ago
|
from couchpotato.core.event import addEvent
|
||
|
|
from couchpotato.core.helpers.encoding import simplifyString, toUnicode
|
||
|
|
from couchpotato.core.logger import CPLog
|
||
|
|
from couchpotato.core.providers.show.base import ShowProvider
|
||
|
|
from thetvdb.tvdb_api import Tvdb
|
||
|
|
import traceback
|
||
|
|
|
||
|
|
log = CPLog(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class TheTVDb(ShowProvider):
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
#addEvent('show.by_hash', self.byHash)
|
||
|
|
addEvent('show.search', self.search, priority = 1)
|
||
|
|
addEvent('show.info', self.getInfo, priority = 1)
|
||
|
|
#addEvent('show.info_by_thetvdb', self.getInfoByTheTVDBId)
|
||
|
|
|
||
|
|
# Use base wrapper
|
||
|
|
#thetvdbtmdb.configure(self.conf('api_key'))
|
||
|
|
self.tvdb = Tvdb(apikey="7966C02F860586D2", banners=True)
|
||
|
|
|
||
|
|
#def byHash(self, file):
|
||
|
|
#''' Find show by hash '''
|
||
|
|
|
||
|
|
#if self.isDisabled():
|
||
|
|
#return False
|
||
|
|
|
||
|
|
#cache_key = 'tmdb.cache.%s' % simplifyString(file)
|
||
|
|
#results = self.getCache(cache_key)
|
||
|
|
|
||
|
|
#if not results:
|
||
|
|
#log.debug('Searching for show by hash: %s', file)
|
||
|
|
#try:
|
||
|
|
#raw = tmdb.searchByHashingFile(file)
|
||
|
|
|
||
|
|
#results = []
|
||
|
|
#if raw:
|
||
|
|
#try:
|
||
|
|
#results = self.parseShow(raw)
|
||
|
|
#log.info('Found: %s', results['titles'][0] + ' (' + str(results.get('year', 0)) + ')')
|
||
|
|
|
||
|
|
#self.setCache(cache_key, results)
|
||
|
|
#return results
|
||
|
|
#except SyntaxError, e:
|
||
|
|
#log.error('Failed to parse XML response: %s', e)
|
||
|
|
#return False
|
||
|
|
#except:
|
||
|
|
#log.debug('No shows known by hash for: %s', file)
|
||
|
|
#pass
|
||
|
|
|
||
|
|
#return results
|
||
|
|
|
||
|
|
def search(self, q, limit = 12):
|
||
|
|
''' Find show by name
|
||
|
|
show = { 'id': 74713,
|
||
|
|
'language': 'en',
|
||
|
|
'lid': 7,
|
||
|
|
'seriesid': '74713',
|
||
|
|
'seriesname': u'Breaking Bad',}
|
||
|
|
'''
|
||
|
|
|
||
|
|
if self.isDisabled():
|
||
|
|
return False
|
||
|
|
|
||
|
|
search_string = simplifyString(q)
|
||
|
|
cache_key = 'thetvdb.cache.%s.%s' % (search_string, limit)
|
||
|
|
results = self.getCache(cache_key)
|
||
|
|
|
||
|
|
if not results:
|
||
|
|
log.debug('Searching for show: %s', q)
|
||
|
|
|
||
|
|
raw = None
|
||
|
|
try:
|
||
|
|
raw = self.tvdb.search(search_string)
|
||
|
|
|
||
|
|
except: # XXX: Make more specific
|
||
|
|
log.error('Failed searching TheTVDB for "%s": %s', (search_string, traceback.format_exc()))
|
||
|
|
|
||
|
|
results = []
|
||
|
|
if raw:
|
||
|
|
try:
|
||
|
|
nr = 0
|
||
|
|
|
||
|
|
for show in raw:
|
||
|
|
show = self.tvdb[int(show['id'])]
|
||
|
|
results.append(self.parseShow(show))
|
||
|
|
|
||
|
|
nr += 1
|
||
|
|
if nr == limit:
|
||
|
|
break
|
||
|
|
|
||
|
|
log.info('Found: %s', [result['titles'][0] + ' (' + str(result.get('year', 0)) + ')' for result in results])
|
||
|
|
|
||
|
|
self.setCache(cache_key, results)
|
||
|
|
return results
|
||
|
|
except SyntaxError, e:
|
||
|
|
log.error('Failed to parse XML response: %s', e)
|
||
|
|
return False
|
||
|
|
|
||
|
|
return results
|
||
|
|
|
||
|
|
def getInfo(self, identifier = None):
|
||
|
|
|
||
|
|
if not identifier:
|
||
|
|
return {}
|
||
|
|
|
||
|
|
cache_key = 'thetvdb.cache.%s' % identifier
|
||
|
|
result = self.getCache(cache_key)
|
||
|
|
|
||
|
|
if not result:
|
||
|
|
result = {}
|
||
|
|
show = None
|
||
|
|
|
||
|
|
try:
|
||
|
|
log.debug('Getting info: %s', cache_key)
|
||
|
|
#show = thetvdb.imdbLookup(id = identifier)
|
||
|
|
show = self.tvdb[int(identifier)]
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
|
||
|
|
if show:
|
||
|
|
#result = self.parseShow(show[0])
|
||
|
|
result = self.parseShow(show)
|
||
|
|
self.setCache(cache_key, result)
|
||
|
|
|
||
|
|
return result
|
||
|
|
|
||
|
|
#def getInfoByTheTVDBId(self, id = None):
|
||
|
|
|
||
|
|
#cache_key = 'thetvdb.cache.%s' % id
|
||
|
|
#result = self.getCache(cache_key)
|
||
|
|
|
||
|
|
#if not result:
|
||
|
|
#result = {}
|
||
|
|
#show = None
|
||
|
|
|
||
|
|
#try:
|
||
|
|
#log.debug('Getting info: %s', cache_key)
|
||
|
|
#show = tmdb.getShowInfo(id = id)
|
||
|
|
#except:
|
||
|
|
#pass
|
||
|
|
|
||
|
|
#if show:
|
||
|
|
#result = self.parseShow(show)
|
||
|
|
#self.setCache(cache_key, result)
|
||
|
|
|
||
|
|
#return result
|
||
|
|
|
||
|
|
def parseShow(self, show):
|
||
|
|
"""
|
||
|
|
show[74713] = {
|
||
|
|
'actors': u'|Bryan Cranston|Aaron Paul|Dean Norris|RJ Mitte|Betsy Brandt|Anna Gunn|Laura Fraser|Jesse Plemons|Christopher Cousins|Steven Michael Quezada|Jonathan Banks|Giancarlo Esposito|Bob Odenkirk|',
|
||
|
|
'added': None,
|
||
|
|
'addedby': None,
|
||
|
|
'airs_dayofweek': u'Sunday',
|
||
|
|
'airs_time': u'9:00 PM',
|
||
|
|
'banner': u'http://thetvdb.com/banners/graphical/81189-g13.jpg',
|
||
|
|
'contentrating': u'TV-MA',
|
||
|
|
'fanart': u'http://thetvdb.com/banners/fanart/original/81189-28.jpg',
|
||
|
|
'firstaired': u'2008-01-20',
|
||
|
|
'genre': u'|Crime|Drama|Suspense|',
|
||
|
|
'id': u'81189',
|
||
|
|
'imdb_id': u'tt0903747',
|
||
|
|
'language': u'en',
|
||
|
|
'lastupdated': u'1376620212',
|
||
|
|
'network': u'AMC',
|
||
|
|
'networkid': None,
|
||
|
|
'overview': u"Walter White, a struggling high school chemistry teacher is diagnosed with advanced lung cancer. He turns to a life of crime, producing and selling methamphetamine accompanied by a former student, Jesse Pinkman with the aim of securing his family's financial future before he dies.",
|
||
|
|
'poster': u'http://thetvdb.com/banners/posters/81189-22.jpg',
|
||
|
|
'rating': u'9.3',
|
||
|
|
'ratingcount': u'473',
|
||
|
|
'runtime': u'60',
|
||
|
|
'seriesid': u'74713',
|
||
|
|
'seriesname': u'Breaking Bad',
|
||
|
|
'status': u'Continuing',
|
||
|
|
'zap2it_id': u'SH01009396'}
|
||
|
|
"""
|
||
|
|
|
||
|
|
## Images
|
||
|
|
poster = self.getImage(show, type = 'poster', size = 'cover')
|
||
|
|
backdrop = self.getImage(show, type = 'fanart', size = 'w1280')
|
||
|
|
#poster_original = self.getImage(show, type = 'poster', size = 'original')
|
||
|
|
#backdrop_original = self.getImage(show, type = 'backdrop', size = 'original')
|
||
|
|
|
||
|
|
## Genres
|
||
|
|
genres = [] if show['genre'] is None else show['genre'].strip('|').split('|')
|
||
|
|
|
||
|
|
## Year (not really needed for show)
|
||
|
|
year = None
|
||
|
|
|
||
|
|
show_data = {
|
||
|
|
'via_thetvdb': True,
|
||
|
|
'thetvdb_id': int(show['id']),
|
||
|
|
'titles': [show['seriesname'], ],
|
||
|
|
'original_title': show['seriesname'],
|
||
|
|
'images': {
|
||
|
|
'poster': [poster] if poster else [],
|
||
|
|
'backdrop': [backdrop] if backdrop else [],
|
||
|
|
'poster_original': [],
|
||
|
|
'backdrop_original': [],
|
||
|
|
},
|
||
|
|
'imdb': show['imdb_id'],
|
||
|
|
'runtime': show['runtime'],
|
||
|
|
'released': show['firstaired'],
|
||
|
|
'year': year,
|
||
|
|
'plot': show['overview'],
|
||
|
|
'genres': genres,
|
||
|
|
}
|
||
|
|
|
||
|
|
show_data = dict((k, v) for k, v in show_data.iteritems() if v)
|
||
|
|
|
||
|
|
## Add alternative names
|
||
|
|
#for alt in ['original_name', 'alternative_name']:
|
||
|
|
#alt_name = toUnicode(show.get(alt))
|
||
|
|
#if alt_name and not alt_name in show_data['titles'] and alt_name.lower() != 'none' and alt_name != None:
|
||
|
|
#show_data['titles'].append(alt_name)
|
||
|
|
|
||
|
|
return show_data
|
||
|
|
|
||
|
|
def getImage(self, show, type = 'poster', size = 'cover'):
|
||
|
|
""""""
|
||
|
|
# XXX: Need to implement size
|
||
|
|
image_url = ''
|
||
|
|
|
||
|
|
for res, res_data in show['_banners'].get(type, {}).items():
|
||
|
|
for bid, banner_info in res_data.items():
|
||
|
|
image_url = banner_info.get('_bannerpath', '')
|
||
|
|
break
|
||
|
|
|
||
|
|
return image_url
|
||
|
|
|
||
|
|
def isDisabled(self):
|
||
|
|
if self.conf('api_key') == '':
|
||
|
|
log.error('No API key provided.')
|
||
|
|
True
|
||
|
|
else:
|
||
|
|
False
|