update_film_references.py

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

update_viewings_csv.py

@@ -1,3 +1,4 @@
+import argparse
 from bs4 import BeautifulSoup
 import hashlib
 
@@ -30,4 +31,14 @@ def update_viewings_csv(token_id, token_secret, check_existing=True):
         Bookstack.delete_attachment(VCinemaUtils.JACKNET_WIKI_URL, token_id, token_secret, existing_attachment_id)
         print("File updated")
     else:
-        print("File already up-to-date")
+        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)