vcinema/wiki_pages/FilmsByReference.py

85 lines
3.3 KiB
Python
Raw Normal View History

2022-03-25 22:19:40 +00:00
from collections import OrderedDict
import wikipedia
2022-03-25 22:19:40 +00:00
2022-12-04 12:16:26 +00:00
from bookstack import Bookstack
2022-03-25 22:19:40 +00:00
from vcinema_utils import VCinemaUtils
# Page ID of https://wiki.jacknet.io/books/vcinema/page/references
2022-04-15 17:22:04 +01:00
PAGE_ID = 62
2022-03-25 22:19:40 +00:00
def get_films_by_reference(viewings):
2022-12-04 13:00:58 +00:00
films_by_reference = {}
for viewing in viewings:
if "keywords" in viewing.keys():
for keyword in viewing["keywords"]:
if keyword.startswith("reference-to-"):
for reference in films_by_reference:
if keyword in films_by_reference[reference]["keywords"]:
films_by_reference[reference]["films"].append(viewing)
break
else:
referenced = keyword[13:].replace("-", " ")
if referenced.startswith("a "):
referenced = referenced[2:]
try:
searches = wikipedia.search(referenced, suggestion=False)
referenced_page = wikipedia.page(title=referenced, auto_suggest=False)
page_title = referenced_page.title
page_url = referenced_page.url
except wikipedia.DisambiguationError as e1:
page_title = e1.title
page_url = "https://en.wikipedia.org/wiki/{}".format(e1.title.replace(" ", "_"))
except wikipedia.PageError as _:
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 e2:
page_title = e2.title
page_url = "https://en.wikipedia.org/wiki/{}".format(e2.title.replace(" ", "_"))
if page_title in films_by_reference.keys():
films_by_reference[page_title]["films"].append(viewing)
films_by_reference[page_title]["keywords"].append(keyword)
else:
films_by_reference[page_title] = {"url": page_url,
"keywords": [keyword],
"films": [viewing]}
return films_by_reference
2022-03-25 22:19:40 +00:00
2022-04-16 09:43:51 +01:00
def update_page(token_id, token_secret, films_by_reference_keyword):
page = build_page(films_by_reference_keyword)
Bookstack.update_page(VCinemaUtils.JACKNET_WIKI_URL, token_id, token_secret, PAGE_ID, markdown=page)
2022-12-04 13:00:58 +00:00
def build_page(films_by_reference):
films_by_reference_sorted = OrderedDict(sorted(films_by_reference.items(), key=lambda t: t[0]))
2022-12-04 12:16:26 +00:00
table = "| Referenced | Films |\n| - | - |"
2022-12-04 13:00:58 +00:00
for reference, referenced in films_by_reference_sorted.items():
2022-12-04 12:16:26 +00:00
table += "\n"
row_data = []
reference_url = referenced["url"]
reference_title = reference
referenced_films = referenced["films"]
row_data.append("[{}]({})".format(reference_title, reference_url))
row_data.append(VCinemaUtils.get_film_list(referenced_films))
2022-04-08 22:01:58 +01:00
table += " | ".join(row_data)
return table