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

fix countries

Browse source
This commit is contained in:
j 2018-01-14 18:33:59 +01:00
parent e480b8dcbf
commit 2a03726c39
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
ox/web/imdb.py
View file

@ -289,6 +289,11 @@ class Imdb(SiteParser):
isinstance(self['alternativeTitles'][0], string_types): isinstance(self['alternativeTitles'][0], string_types):
self['alternativeTitles'] = [self['alternativeTitles']] self['alternativeTitles'] = [self['alternativeTitles']]
for key in ('country', 'genre'):
if key in self:
self[key] = [x[0] if len(x) == 1 and isinstance(x, list) else x for x in self[key]]
self[key] = list(filter(lambda x: x.lower() != 'home', self[key]))
#normalize country names #normalize country names
if 'country' in self: if 'country' in self:
self['country'] = [normalize_country_name(c) or c for c in self['country']] self['country'] = [normalize_country_name(c) or c for c in self['country']]
@ -372,9 +377,6 @@ class Imdb(SiteParser):
self['connections'] = cc self['connections'] = cc
for key in ('country', 'genre'):
if key in self:
self[key] = list(filter(lambda x: x.lower() != 'home', self[key]))
if 'isSeries' in self: if 'isSeries' in self:
del self['isSeries'] del self['isSeries']
self['isSeries'] = True self['isSeries'] = True