diff --git a/wiki_pages/FilmsByReference.py b/wiki_pages/FilmsByReference.py index a0a06b7..57726a6 100644 --- a/wiki_pages/FilmsByReference.py +++ b/wiki_pages/FilmsByReference.py @@ -27,6 +27,9 @@ def get_films_by_reference(viewings): if keyword.startswith("a-"): keyword = keyword[2:] + if keyword.endswith("-character"): + keyword = keyword[:-10] + referenced = keyword.replace("-", " ") try: @@ -41,15 +44,19 @@ def get_films_by_reference(viewings): page_title = page_title[0].upper() + page_title[1:] page_url = VCinemaUtils.generate_wikipedia_url(page_title) except wikipedia.PageError as _: - try: - referenced_page = wikipedia.page(title=searches[0], auto_suggest=False) + if len(searches) > 0: + try: + referenced_page = wikipedia.page(title=searches[0], auto_suggest=False) - page_title = referenced_page.title - page_url = referenced_page.url - except wikipedia.DisambiguationError as e: - page_title = e.title - page_title = page_title[0].upper() + page_title[1:] - page_url = VCinemaUtils.generate_wikipedia_url(page_title) + page_title = referenced_page.title + page_url = referenced_page.url + except wikipedia.DisambiguationError as e: + page_title = e.title + page_title = page_title[0].upper() + page_title[1:] + page_url = VCinemaUtils.generate_wikipedia_url(page_title) + else: + page_title = referenced.title() + page_url = None if page_title in films_by_reference.keys(): films_by_reference[page_title]["keywords"].append(keyword)