sarah/vcinema
1
2
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

change how pages to update are specified

Browse Source
This commit is contained in:
Sarah 2022-04-22 00:11:38 +01:00
parent fcb3c9a3b3
commit d9d7948919
1 changed files with 9 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
update_wiki.py
View File

@ -5,8 +5,7 @@ import argparse
from progress.bar import IncrementalBar 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, pages):
update_film_references, update_hidden_themes, update_keyword_scores):
if update_csv: if update_csv:
print("Updating CSV") print("Updating CSV")
ViewingsCsv.update_viewings_csv(token_id, token_secret) 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") print("Getting viewings")
viewings = VCinemaUtils.get_vcinema_viewings(token_id, token_secret) 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 = [] data_fields = []
if update_films_by_year: if update_films_by_year:
data_fields.append("year") data_fields.append("year")
@ -76,14 +81,8 @@ if __name__ == '__main__':
parser.add_argument('token_secret', help='API token secret.') 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_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('--pages', nargs="+", default=['years', 'countries', 'references', 'themes', 'scores'], 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)
args = parser.parse_args() args = parser.parse_args()
update_wiki(args.token_id, args.token_secret, args.update_csv, args.update_films_by_year, update_wiki(args.token_id, args.token_secret, args.update_csv, args.pages)
args.update_films_by_country, args.update_films_by_reference, args.update_hidden_themes,
args.update_keyword_scores)