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

add update references

Browse Source
This commit is contained in:
Sarah 2022-04-08 23:00:53 +01:00
parent 0ab9d81a88
commit 418b5117e4
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
update_wiki.py
View File

@ -1,5 +1,6 @@
from bookstack import Bookstack 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_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 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 vcinema_utils import VCinemaUtils
from update_viewings_csv import update_viewings_csv from update_viewings_csv import update_viewings_csv
@ -8,7 +9,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, update_films_by_year, update_films_by_country, update_film_references):
if update_csv: if update_csv:
print("Updating CSV") print("Updating CSV")
update_viewings_csv(token_id, token_secret) 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: if update_films_by_country:
data_fields.append("countries") data_fields.append("countries")
if update_film_references:
data_fields.append("keywords")
viewing_count = len(viewings) viewing_count = len(viewings)
with IncrementalBar('Retrieving movie data', max=viewing_count, suffix='%(percent).1f%% - %(eta)ds remaining', check_tty=False) as bar: 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) 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: 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: 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: if update_films_by_country:
films_by_country = get_films_by_country(viewings) films_by_country = get_films_by_country(viewings)
bar.next() bar.next()
if update_film_references:
films_by_reference = get_films_by_reference(viewings)
bar.finish() 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: if update_films_by_country:
films_by_country_page = build_films_by_country_page(films_by_country) films_by_country_page = build_films_by_country_page(films_by_country)
bar.next() bar.next()
if update_film_references:
films_by_reference_page = build_film_references_page(films_by_reference)
bar.finish() 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: 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) Bookstack.update_page(VCinemaUtils.JACKNET_WIKI_URL, token_id, token_secret, FILMS_BY_COUNTRY_PAGE_ID, markdown=films_by_country_page)
bar.next() 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() 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_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_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_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() 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)