From 1e970b4fb5d7e8e05c0813deeabcb71657420d02 Mon Sep 17 00:00:00 2001 From: Sarah Date: Fri, 22 Apr 2022 22:01:09 +0100 Subject: [PATCH] Create generate_map_timelapse.py --- generate_map_timelapse.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 generate_map_timelapse.py diff --git a/generate_map_timelapse.py b/generate_map_timelapse.py new file mode 100644 index 0000000..f487725 --- /dev/null +++ b/generate_map_timelapse.py @@ -0,0 +1,34 @@ +from wiki_pages import FilmsByCountry +from vcinema_utils import VCinemaUtils + +import argparse +from progress.bar import IncrementalBar + + +def generate_map_timelapse(token_id, token_secret): + print("Getting viewings") + viewings = VCinemaUtils.get_vcinema_viewings(token_id, token_secret, combine_repeat_viewings=False) + + viewing_count = len(viewings) + 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, ['countries'], bar) + + date_viewings = VCinemaUtils.filter_viewings(viewings, "date_watched") + + for date, viewings in date_viewings.items(): + date_viewings_countries = VCinemaUtils.filter_viewings(viewings, "countries") + + print(date_viewings_countries) + + with open("map-{}.png".format(date), "wb") as f: + f.write(FilmsByCountry.draw_map(date_viewings_countries, file_name="map-{}.svg".format(date))) + + +if __name__ == '__main__': + parser = argparse.ArgumentParser(description='Update wiki pages.') + parser.add_argument('token_id', help='API token ID.') + parser.add_argument('token_secret', help='API token secret.') + + args = parser.parse_args() + + generate_map_timelapse(args.token_id, args.token_secret)