From d9d79489193a80b5aae569768cca5b2ef2784801 Mon Sep 17 00:00:00 2001 From: Sarah Date: Fri, 22 Apr 2022 00:11:38 +0100 Subject: [PATCH] change how pages to update are specified --- update_wiki.py | 19 +++++++++---------- 1 file changed, 9 insertions(+), 10 deletions(-) diff --git a/update_wiki.py b/update_wiki.py index 6245a76..8a710d1 100644 --- a/update_wiki.py +++ b/update_wiki.py @@ -5,8 +5,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, - update_film_references, update_hidden_themes, update_keyword_scores): +def update_wiki(token_id, token_secret, update_csv, pages): if update_csv: print("Updating CSV") ViewingsCsv.update_viewings_csv(token_id, token_secret) @@ -14,6 +13,12 @@ def update_wiki(token_id, token_secret, update_csv, update_films_by_year, update print("Getting viewings") viewings = VCinemaUtils.get_vcinema_viewings(token_id, token_secret) + update_films_by_year = 'years' in pages + update_films_by_country = 'countries' in pages + update_film_references = 'references' in pages + update_hidden_themes = 'themes' in pages + update_keyword_scores = 'scores' in pages + data_fields = [] if update_films_by_year: data_fields.append("year") @@ -76,14 +81,8 @@ if __name__ == '__main__': parser.add_argument('token_secret', help='API token secret.') parser.add_argument('--update_csv', help='Update viewings.csv file, default: True', default=True, required=False) - parser.add_argument('--update_films_by_year', help='Update films by release year page, default: True', default=True, required=False) - parser.add_argument('--update_films_by_country', help='Update films by country page, default: True', default=True, required=False) - parser.add_argument('--update_films_by_reference', help='Update films by reference page, default: True', default=True, required=False) - parser.add_argument('--update_hidden_themes', help='Update hidden themes page, default: True', default=True, required=False) - parser.add_argument('--update_keyword_scores', help='Update keyword scores page - THIS TAKES A WHILE, default: False', default=False, required=False) + parser.add_argument('--pages', nargs="+", default=['years', 'countries', 'references', 'themes', 'scores'], required=False) 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, args.update_films_by_reference, args.update_hidden_themes, - args.update_keyword_scores) + update_wiki(args.token_id, args.token_secret, args.update_csv, args.pages)