diff --git a/scrapeit/wikipedia.py b/scrapeit/wikipedia.py index fbd9704..ff3597a 100644 --- a/scrapeit/wikipedia.py +++ b/scrapeit/wikipedia.py @@ -26,11 +26,13 @@ def searchByImdb(imdb_id, title=None, director=None): data = read_url(url) soup = BeautifulSoup(data) edit_url = soup('a', dict(href=re.compile('action=edit'), - title=re.compile('You can edit this page.')))[0]['href'] - edit_url = "http://en.wikipedia.org%s" % edit_url.replace('&', '&') - data = read_url(edit_url) - w_imdb_id = data.find('imdb_id') - if w_imdb_id > 0: - if imdb_id in data[w_imdb_id:w_imdb_id+50]: - return url + title=re.compile('You can edit this page.'))) + if edit_url: + edit_url = edit_url[0]['href'] + edit_url = "http://en.wikipedia.org%s" % edit_url.replace('&', '&') + data = read_url(edit_url) + w_imdb_id = data.find('imdb_id') + if w_imdb_id > 0: + if imdb_id in data[w_imdb_id:w_imdb_id+50]: + return url return ''