moparisthebest/SickRage
1
0
Fork 0
mirror of https://github.com/moparisthebest/SickRage synced 2024-12-13 11:32:20 -05:00
Code Issues Releases Wiki Activity

Fix for country codes not being properly accessed from imdb causing no icon to display when looking at show info

Browse Source
This commit is contained in:
echel0n 2014-04-24 13:21:04 -07:00
parent f8ec897010
commit 0cc5ba4eb7
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
sickbeard/tv.py
View File

@ -784,7 +784,7 @@ class TVShow(object):
'runtimes': '', 'runtimes': '',
'genres': [], 'genres': [],
'countries': '', 'countries': '',
'country_codes': '', 'country_codes': [],
'certificates': [], 'certificates': [],
'rating': '', 'rating': '',
'votes': '', 'votes': '',
@ -797,12 +797,13 @@ class TVShow(object):
i = imdb.IMDb() i = imdb.IMDb()
imdbTv = i.get_movie(str(re.sub("[^0-9]", "", self.imdbid))) imdbTv = i.get_movie(str(re.sub("[^0-9]", "", self.imdbid)))
for key in filter(lambda x: x in imdbTv.keys(), imdb_info.keys()): test = imdbTv.keys()
for key in filter(lambda x: x.replace('_', ' ') in imdbTv.keys(), imdb_info.keys()):
# Store only the first value for string type # Store only the first value for string type
if type(imdb_info[key]) == type('') and type(imdbTv.get(key)) == type([]): if type(imdb_info[key]) == type('') and type(imdbTv.get(key)) == type([]):
imdb_info[key] = imdbTv.get(key)[0] imdb_info[key] = imdbTv.get(key.replace('_', ' '))[0]
else: else:
imdb_info[key] = imdbTv.get(key) imdb_info[key] = imdbTv.get(key.replace('_', ' '))
#Filter only the value #Filter only the value
if imdb_info['runtimes']: if imdb_info['runtimes']:
@ -821,6 +822,11 @@ class TVShow(object):
else: else:
imdb_info['genres'] = '' imdb_info['genres'] = ''
if imdb_info['country_codes']:
imdb_info['country_codes'] = '|'.join(imdb_info['country_codes'])
else:
imdb_info['country_codes'] = ''
#Get only the production country certificate if any #Get only the production country certificate if any
if imdb_info['certificates'] and imdb_info['countries']: if imdb_info['certificates'] and imdb_info['countries']:
dct = {} dct = {}
@ -839,8 +845,7 @@ class TVShow(object):
#Rename dict keys without spaces for DB upsert #Rename dict keys without spaces for DB upsert
self.imdb_info = dict( self.imdb_info = dict(
(k.replace(' ', '_'), float(v) if hasattr(v, 'keys') else v) for k, v in imdb_info.items()) (k.replace(' ', '_'), f(v) if hasattr(v, 'keys') else v) for k, v in imdb_info.items())
logger.log(str(self.indexerid) + u": Obtained info from IMDb ->" + str(self.imdb_info), logger.DEBUG) logger.log(str(self.indexerid) + u": Obtained info from IMDb ->" + str(self.imdb_info), logger.DEBUG)
def nextEpisode(self): def nextEpisode(self):