From 418b5117e4a9559474ca0d59e2b810cfb74cad52 Mon Sep 17 00:00:00 2001 From: Sarah Date: Fri, 8 Apr 2022 23:00:53 +0100 Subject: [PATCH] add update references --- update_wiki.py | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) diff --git a/update_wiki.py b/update_wiki.py index 06669a6..d1f0d0e 100644 --- a/update_wiki.py +++ b/update_wiki.py @@ -1,5 +1,6 @@ from bookstack import Bookstack from update_films_by_country import get_films_by_country, build_page as build_films_by_country_page, FILMS_BY_COUNTRY_PAGE_ID +from update_film_references import get_films_by_reference, build_page as build_film_references_page, FILM_BY_REFERENCES_PAGE_ID from update_films_by_year import get_films_by_year, build_page as build_films_by_year_page, FILMS_BY_YEAR_PAGE_ID from vcinema_utils import VCinemaUtils from update_viewings_csv import update_viewings_csv @@ -8,7 +9,7 @@ import argparse from progress.bar import IncrementalBar -def update_wiki(token_id, token_secret, update_csv, update_films_by_year, update_films_by_country): +def update_wiki(token_id, token_secret, update_csv, update_films_by_year, update_films_by_country, update_film_references): if update_csv: print("Updating CSV") update_viewings_csv(token_id, token_secret) @@ -23,11 +24,14 @@ def update_wiki(token_id, token_secret, update_csv, update_films_by_year, update if update_films_by_country: data_fields.append("countries") + if update_film_references: + data_fields.append("keywords") + viewing_count = len(viewings) with IncrementalBar('Retrieving movie data', max=viewing_count, suffix='%(percent).1f%% - %(eta)ds remaining', check_tty=False) as bar: VCinemaUtils.add_imdb_data_to_viewings(viewings, data_fields, bar) - update_page_count = sum([update_films_by_year, update_films_by_country]) + update_page_count = sum([update_films_by_year, update_films_by_country, update_film_references]) with IncrementalBar('Processing viewing data', max=update_page_count, suffix='%(percent).1f%% - %(eta)ds remaining', check_tty=False) as bar: if update_films_by_year: @@ -36,6 +40,8 @@ def update_wiki(token_id, token_secret, update_csv, update_films_by_year, update if update_films_by_country: films_by_country = get_films_by_country(viewings) bar.next() + if update_film_references: + films_by_reference = get_films_by_reference(viewings) bar.finish() @@ -46,6 +52,8 @@ def update_wiki(token_id, token_secret, update_csv, update_films_by_year, update if update_films_by_country: films_by_country_page = build_films_by_country_page(films_by_country) bar.next() + if update_film_references: + films_by_reference_page = build_film_references_page(films_by_reference) bar.finish() @@ -56,6 +64,9 @@ def update_wiki(token_id, token_secret, update_csv, update_films_by_year, update if update_films_by_country: Bookstack.update_page(VCinemaUtils.JACKNET_WIKI_URL, token_id, token_secret, FILMS_BY_COUNTRY_PAGE_ID, markdown=films_by_country_page) bar.next() + if update_film_references: + Bookstack.update_page(VCinemaUtils.JACKNET_WIKI_URL, token_id, token_secret, FILM_BY_REFERENCES_PAGE_ID, markdown=films_by_reference_page) + bar.next() bar.finish() @@ -68,7 +79,8 @@ if __name__ == '__main__': parser.add_argument('update_csv', help='Update viewings.csv file, default: True', default=True) parser.add_argument('update_films_by_year', help='Update films by release year page, default: True', default=True) parser.add_argument('update films_by_country', help='Update films by country page, default: True', default=True) + parser.add_argument('update_films_by_reference', help='Update films by reference page, default: True', default=True) args = parser.parse_args() - update_wiki(args.token_id, args.token_secret, args.update_csv, args.update_films_by_year, args.update_films_by_country) + update_wiki(args.token_id, args.token_secret, args.update_csv, args.update_films_by_year, args.update_films_by_country, args.update_films_by_reference)