Update update_viewings_csv.py

update_film_references.py

@@ -0,0 +1,29 @@
+from collections import OrderedDict
+import string
+
+from vcinema_utils import VCinemaUtils
+
+# Page ID of https://wiki.jacknet.io/books/vcinema/page/references
+FILM_BY_REFERENCES_PAGE_ID = 63
+
+
+def get_films_by_reference(viewings):
+    viewings_filtered_by_keyword = VCinemaUtils.filter_viewings(viewings, "keywords")
+    viewings_filtered_by_reference_keyword = {k: v for k, v in viewings_filtered_by_keyword.items() if k.startswith("reference-to")}
+
+    return viewings_filtered_by_reference_keyword
+
+
+def build_page(reference_keywords):
+    reference_keywords_sorted = OrderedDict(sorted(reference_keywords.items(), key=lambda t: t[0]))
+
+    films_by_reference_table = "| Referenced | Films |\n| - | - |"
+
+    for year in reference_keywords_sorted.keys():
+        films_by_reference_table += "\n"
+        films_by_reference_table += str(string.capwords(year[13:].replace("-", " "))) + " | "
+
+        film_links = ["[{}](https://www.imdb.com/title/tt{}/)".format(film['title'], film['imdb_id']) for film in reference_keywords_sorted[year]]
+        films_by_reference_table += "<br>".join(film_links)
+
+    return films_by_reference_table

update_films_by_year.py

@@ -20,7 +20,9 @@ def build_page(films_by_year, progressbar=None):
     for year in films_by_year_sorted.keys():
         page_table += "\n"
         page_table += str(year) + " | "
-        page_table += "<br>".join("[{}](https://www.imdb.com/title/tt{}/)".format(film['title'], film['imdb_id']) for film in films_by_year_sorted[year])
+
+        film_links = ["[{}](https://www.imdb.com/title/tt{}/)".format(film['title'], film['imdb_id']) for film in films_by_year_sorted[year]]
+        page_table += "<br>".join(film_links)
 
         if progressbar is not None:
             progressbar.next()

update_viewings_csv.py

@@ -1,4 +1,3 @@
-import argparse
 from bs4 import BeautifulSoup
 import hashlib
 
@@ -32,13 +31,3 @@ def update_viewings_csv(token_id, token_secret, check_existing=True):
         print("File updated")
     else:
         print("File already up-to-date")
-
-
-if __name__ == '__main__':
-    parser = argparse.ArgumentParser(description='Update the vcinema.csv file with data from the CSV wiki page.')
-    parser.add_argument('token_id', help='API token ID.')
-    parser.add_argument('token_secret', help='API token secret.')
-
-    args = parser.parse_args()
-
-    update_viewings_csv(args.token_id, args.token_secret)