diff --git a/oxweb/wikipedia.py b/oxweb/wikipedia.py index 100c8db..b6847c3 100644 --- a/oxweb/wikipedia.py +++ b/oxweb/wikipedia.py @@ -51,6 +51,17 @@ def getMovieData(wikipediaUrl): filmbox[key] = value return filmbox +def getImageUrl(name): + data = getUrlUnicode('http://en.wikipedia.org/wiki/Image:' + name) + url = findRe(data, '="(http://upload.wikimedia.org/.*?)"') + return url + +def getMoviePoster(wikipediaUrl): + data = getMovieData(wikipediaUrl) + if 'image' in data: + return getImageUrl(data['image']) + return '' + def getAmgId(wikipediaUrl): data = getMovieData(wikipediaUrl) return data.get('amg_id', '')