Create get_countries.py

roundup_scripts/get_countries.py

@@ -0,0 +1,36 @@
+import argparse
+from datetime import datetime, timedelta
+import json
+from progress.bar import IncrementalBar
+
+from vcinema_utils import VCinemaUtils
+
+
+def get_countries(token_id, token_secret, start_date, end_date):
+    print("Getting viewings")
+    viewings = VCinemaUtils.get_vcinema_viewings(token_id, token_secret)
+
+    with IncrementalBar('Retrieving movie data', max=len(viewings), suffix='%(percent).1f%% - %(eta)ds remaining', check_tty=False) as bar:
+        VCinemaUtils.add_imdb_data_to_viewings(viewings, ["countries"], bar)
+
+    in_range_viewings = VCinemaUtils.get_viewings_in_date_range(viewings, start_date, end_date)
+
+    in_range_countries = VCinemaUtils.filter_viewings(in_range_viewings, "countries")
+
+    print(f"Countries in range {start_date.strftime('%d-%m-%Y')} - {end_date.strftime('%d-%m-%Y')}:")
+    for country, films in in_range_countries.items():
+        print(f"{country}:{len(films)}")
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument('start_date', type=lambda s: datetime.strptime(s, '%d-%m-%Y'))
+    parser.add_argument('end_date', type=lambda s: datetime.strptime(s, '%d-%m-%Y'))
+
+    args = parser.parse_args()
+
+    with open('../token.json') as json_file:
+        token = json.load(json_file)
+
+    get_countries(token['token_id'], token['token_secret'], args.start_date, args.end_date)