Skip to content

Commit 8a4d6c8

Browse files
plotskimueslo
authored andcommitted
Try harder to find TV show banner
1 parent 9a9f9f0 commit 8a4d6c8

1 file changed

Lines changed: 8 additions & 5 deletions

File tree

pythonbits/tvdb.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -29,11 +29,14 @@ def get_rating(banner):
2929
if banner['subKey'] == str(season_number)])
3030
return season_banners[best_banner['resolution']][best_banner['id']]['_bannerpath']
3131
except (IndexError, KeyError):
32-
# failing that, use show banner. if there's no banner at all we
33-
# just error out for now
34-
series_banners = self.show['_banners']['series']
35-
best_banner = best_banner(series_banners['raw'])
36-
return series_banners[best_banner['resolution']][best_banner['id']]['_bannerpath']
32+
for key in ('poster', 'series', 'fanart'):
33+
try:
34+
series_banners = self.show['_banners'][key]
35+
best_banner = best_banner(series_banners['raw'])
36+
return series_banners[best_banner['resolution']][best_banner['id']]['_bannerpath']
37+
except (IndexError, KeyError):
38+
pass
39+
raise Exception('Unable to find cover')
3740

3841
def summary(self):
3942
return {

0 commit comments

Comments
 (0)