add arg handling

This commit is contained in:
Sarah 2022-04-08 21:41:46 +01:00
parent f44cc0a4c1
commit b4ed2381d5

View File

@ -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)