0x2620/python-ox
2
1
Fork 0
Code Issues Pull requests Releases Wiki Activity

cleanup title in one place

Browse source
This commit is contained in:
j 2012-09-14 11:27:36 +02:00
parent 36c6e6908e
commit 7a8609f5cf
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
ox/web/imdb.py
View file

@ -301,17 +301,20 @@ class Imdb(SiteParser):
self['englishTitle'] = ititle[0][0]
self['title'] = self.get('englishTitle', self['originalTitle'])
def cleanup_title(title):
if title.startswith('"') and title.endswith('"'):
title = title[1:-1]
title = re.sub('\(\#[.\d]+\)', '', title)
return title
for t in ('title', 'englishTitle', 'originalTitle'):
if t in self:
if self[t].startswith('"') and self[t].endswith('"'):
self[t] = self[t][1:-1]
self[t] = re.sub('\(\#[.\d]+\)', '', self[t])
self[t] = cleanup_title(self[t])
if 'alternativeTitles' in self:
if len(self['alternativeTitles']) == 2 and \
isinstance(self['alternativeTitles'][0], basestring):
self['alternativeTitles'] = [self['alternativeTitles']]
self['alternativeTitles'] = [[re.sub('\(\#[.\d]+\)', '', t[0]),
self['alternativeTitles'] = [[cleanup_title(t[0]),
t[1].split(' / ')[0].split('(')[0].strip()]
for t in self['alternativeTitles']]
#self[t] = re.sub('\(\#[.\d]+\)', '', self[t])