cleanup episode titles

This commit is contained in:
j 2012-09-23 15:12:07 +02:00
parent 46f9b36851
commit 7ae6556ff4

View file

@ -293,10 +293,13 @@ class Imdb(SiteParser):
type = type.strip() type = type.strip()
stop_word = False stop_word = False
for key in ( for key in (
'alternative title',
'complete title', 'complete title',
'recut version', 'recut version',
'script title', 'script title',
'working title', 'working title',
'reissue title',
'IMAX version'
): ):
if key in type: if key in type:
stop_word = True stop_word = True
@ -402,6 +405,8 @@ class Imdb(SiteParser):
del self['_director'] del self['_director']
if 'isSeries' in self: if 'isSeries' in self:
del self['isSeries'] del self['isSeries']
if 'episodeTitle' in self:
self['episodeTitle'] = re.sub('Episode \#\d+\.\d+', '', self['episodeTitle'])
if 'series' in self: if 'series' in self:
if 'episodeTitle' in self: if 'episodeTitle' in self:
@ -410,6 +415,8 @@ class Imdb(SiteParser):
if 'episodeTitle' in self and 'season' in self and 'episode' in self: if 'episodeTitle' in self and 'season' in self and 'episode' in self:
self['title'] = "%s (S%02dE%02d) %s" % ( self['title'] = "%s (S%02dE%02d) %s" % (
self['seriesTitle'], self['season'], self['episode'], self['episodeTitle']) self['seriesTitle'], self['season'], self['episode'], self['episodeTitle'])
if 'title' in self:
self['title'] = self['title'].strip()
if 'director' in self: if 'director' in self:
self['episodeDirector'] = self['director'] self['episodeDirector'] = self['director']
@ -433,8 +440,6 @@ class Imdb(SiteParser):
if 'creator' in self: if 'creator' in self:
self['seriesDirector'] = self['creator'] self['seriesDirector'] = self['creator']
if 'originalTitle' in self:
del self['originalTitle']
else: else:
for key in ('seriesTitle', 'episodeTitle', 'season', 'episode'): for key in ('seriesTitle', 'episodeTitle', 'season', 'episode'):
if key in self: if key in self: