diff --git a/oxweb/wikipedia.py b/oxweb/wikipedia.py index 257b87a..905abed 100644 --- a/oxweb/wikipedia.py +++ b/oxweb/wikipedia.py @@ -56,12 +56,16 @@ def getMovieData(wikipediaUrl): filmbox[key] = value if 'imdb title' in data: filmbox['imdb_id'] = findRe(data, 'imdb title\|.*?(\d*?)\|') + else if 'imdb episode' in data: + filmbox['imdb_id'] = findRe(data, 'imdb episode\|.*?(\d*?)\|') if 'Amg movie' in data: filmbox['amg_id'] = findRe(data, 'Amg movie\|.*?(\d*?)\|') if 'rotten-tomatoes' in data: filmbox['rottentomatoes_id'] = findRe(data, 'rotten-tomatoes\|id\=(.*?)\|') if not filmbox['rottentomatoes_id']: filmbox['rottentomatoes_id'] = findRe(data, 'rotten-tomatoes\|(.*?)\|') + if 'DEFAULTSORT' in data: + filmbox['title_sort'] = findRe(data, '''\{\{DEFAULTSORT:(.*?)\}\}''') return filmbox def getImageUrl(name):