From b4ed2381d5397052dfa9394803c9c0a9f1759f19 Mon Sep 17 00:00:00 2001 From: Sarah Date: Fri, 8 Apr 2022 21:41:46 +0100 Subject: [PATCH] add arg handling --- update_wiki.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/update_wiki.py b/update_wiki.py index 140da39..4f441db 100644 --- a/update_wiki.py +++ b/update_wiki.py @@ -8,7 +8,7 @@ import argparse from progress.bar import IncrementalBar -def update_wiki(token_id, token_secret, update_csv=False, update_films_by_year=True, update_films_by_country=True): +def update_wiki(token_id, token_secret, update_csv, update_films_by_year, update_films_by_country): if update_csv: print("Updating CSV") update_viewings_csv(token_id, token_secret) @@ -69,6 +69,10 @@ if __name__ == '__main__': parser.add_argument('token_id', help='API token ID.') parser.add_argument('token_secret', help='API token secret.') + 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) + args = parser.parse_args() - update_wiki(args.token_id, args.token_secret) + update_wiki(args.token_id, args.token_secret, args.update_csv, args.update_films_by_year, args.update_films_by_country)