diff --git a/oxweb/imdb.py b/oxweb/imdb.py index 16fc977..535c788 100644 --- a/oxweb/imdb.py +++ b/oxweb/imdb.py @@ -79,6 +79,8 @@ def getMovieInfo(imdbId): txt = [cleanUp(k) for k in txt.split('|')] elif ', ' in txt: txt = [cleanUp(k) for k in txt.split(', ')] + elif title in ('country', 'language'): + txt = [cleanUp(txt), ] if title == 'original air date': info['series_episode_info'] = txt.split('\n')[-1].strip() txt = txt.split('\n')[0].strip() @@ -605,7 +607,7 @@ class IMDb: self.IMDbDict = IMDbDict if IMDbDict['episode_of']: - episode_of =getMovieInfo(IMDbDict['episode_of']) + episode_of = getMovieInfo(IMDbDict['episode_of']) for key in ('country', 'language'): if not IMDbDict[key]: IMDbDict[key] = episode_of[key]