11# -*- coding: utf-8 -*-
22import tvdb_api
3- from . import imdb
43
54
65class TvdbResult (object ):
@@ -51,24 +50,6 @@ def summary(self):
5150 'imdb_id' : self .show ['imdbId' ],
5251 }
5352
54- def add_show_titles (self , imdb_id , summary ):
55- i = imdb .IMDB ()
56- try :
57- imdb_info = i .get_info (imdb_id )
58- except Exception :
59- summary ['titles' ] = {}
60- else :
61- imdb_sum = imdb_info .summary ()
62- tvdb_title = summary ['title' ]
63- # Original title
64- summary ['title' ] = imdb_sum ['title' ]
65- # dict of international titles
66- summary ['titles' ] = imdb_sum ['titles' ]
67- # "XWW" is IMDb's international title, but unlike TVDB, it doesn't
68- # include the year if there are multiple shows with the same name.
69- if 'XWW' in summary ['titles' ]:
70- summary ['titles' ]['XWW' ] = tvdb_title
71-
7253
7354class TvdbSeason (TvdbResult ):
7455 def summary (self ):
@@ -91,15 +72,12 @@ def summary(self):
9172 s ['url' ] = series_url
9273 s ['cover' ] = self .banner (season_number )
9374 s ['season' ] = season_number
94- s ['imdb_id' ] = self .show ['imdbId' ]
95- self .add_show_titles (s ['imdb_id' ], s )
9675 return s
9776
9877
9978class TvdbEpisode (TvdbResult ):
10079 def summary (self ):
10180 summary = super (TvdbEpisode , self ).summary ()
102- imdb_show_id = summary ['imdb_id' ]
10381 summary .update (** {
10482 'season' : self .episode ['airedSeason' ],
10583 'episode' : self .episode ['episodenumber' ],
@@ -117,14 +95,13 @@ def summary(self):
11795 'url' : 'https://thetvdb.com/series/{}' .format (
11896 self .show ['slug' ]),
11997 'cover' : self .banner (self .episode ['seasonnumber' ])})
120- self .add_show_titles (imdb_show_id , summary )
12198 return summary
12299
123100
124101class TVDB (object ):
125- def __init__ (self ):
126- # todo: selectfirst=False
127- self . tvdb = tvdb_api . Tvdb ( interactive = True , banners = True , actors = True )
102+ def __init__ (self , interactive = True ):
103+ self . tvdb = tvdb_api . Tvdb (
104+ interactive = interactive , banners = True , actors = True )
128105
129106 def search (self , tv_specifier ):
130107 show = self .tvdb [tv_specifier .title ]
0 commit comments