sarah/vcinema
1
2
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Compare commits

base: sarah:4043970511c8da99cbd29d6e70a29021769f5aa3
Branches Tags
sarah:master
sarah:task/refactor-viewings-representation
sarah:feature/add-year-roundup-scripts
sarah:feature/improve-table-generation
..
compare: sarah:00b3f490633350340c13b4c04a52ae34a3ae1765
Branches Tags
sarah:task/refactor-viewings-representation
sarah:feature/add-year-roundup-scripts
sarah:feature/improve-table-generation
sarah:master

No commits in common. "4043970511c8da99cbd29d6e70a29021769f5aa3" and "00b3f490633350340c13b4c04a52ae34a3ae1765" have entirely different histories.
4043970511 ... 00b3f49063

1 changed files with 9 additions and 19 deletions
Show Stats Expand all files Collapse all files

28
wiki_pages/FilmsByReference.py
View File

@ -26,9 +26,6 @@ def get_films_by_reference(viewings):
if keyword.startswith("a-"): if keyword.startswith("a-"):
keyword = keyword[2:] keyword = keyword[2:]
if keyword.endswith("-character"):
keyword = keyword[:-10]
referenced = keyword.replace("-", " ") referenced = keyword.replace("-", " ")
try: try:
@ -43,19 +40,15 @@ def get_films_by_reference(viewings):
page_title = page_title[0].upper() + page_title[1:] page_title = page_title[0].upper() + page_title[1:]
page_url = VCinemaUtils.generate_wikipedia_url(page_title) page_url = VCinemaUtils.generate_wikipedia_url(page_title)
except wikipedia.PageError as _: except wikipedia.PageError as _:
if len(searches) > 0: try:
try: referenced_page = wikipedia.page(title=searches[0], auto_suggest=False)
referenced_page = wikipedia.page(title=searches[0], auto_suggest=False)
page_title = referenced_page.title page_title = referenced_page.title
page_url = referenced_page.url page_url = referenced_page.url
except wikipedia.DisambiguationError as e: except wikipedia.DisambiguationError as e:
page_title = e.title page_title = e.title
page_title = page_title[0].upper() + page_title[1:] page_title = page_title[0].upper() + page_title[1:]
page_url = VCinemaUtils.generate_wikipedia_url(page_title) page_url = VCinemaUtils.generate_wikipedia_url(page_title)
else:
page_title = referenced.title()
page_url = None
if page_title in films_by_reference.keys(): if page_title in films_by_reference.keys():
films_by_reference[page_title]["keywords"].append(keyword) films_by_reference[page_title]["keywords"].append(keyword)
@ -89,10 +82,7 @@ def build_page(films_by_reference):
reference_url = referenced["url"] reference_url = referenced["url"]
referenced_films = referenced["films"] referenced_films = referenced["films"]
if reference_url is None: row_data.append(VCinemaUtils.generate_markdown_link(reference, reference_url))
row_data.append(reference)
else:
row_data.append(VCinemaUtils.generate_markdown_link(reference, reference_url))
row_data.append(VCinemaUtils.get_film_list(referenced_films)) row_data.append(VCinemaUtils.get_film_list(referenced_films))
table += " | ".join(row_data) table += " | ".join(row_data)