update_wiki.py

@@ -34,6 +34,9 @@ def update_wiki(token_id, token_secret, update_csv, pages):
     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)
 
+    update_page_count = sum([update_films_by_year, update_films_by_country, update_film_references,
+                             update_hidden_themes, update_keyword_scores])
+
     print("Processing viewing data")
 
     if update_films_by_year:
@@ -54,6 +57,25 @@ def update_wiki(token_id, token_secret, update_csv, pages):
 
     print("Done!")
 
+    with IncrementalBar('Updating pages', max=update_page_count, check_tty=False) as bar:
+        if update_films_by_year:
+            FilmsByYear.update_page(token_id, token_secret, films_by_year)
+            bar.next()
+        if update_films_by_country:
+            FilmsByCountry.update_page(token_id, token_secret, films_by_country)
+            bar.next()
+        if update_film_references:
+            FilmsByReference.update_page(token_id, token_secret, films_by_reference)
+            bar.next()
+        if update_hidden_themes:
+            HiddenThemes.update_page(token_id, token_secret, hidden_themes)
+            bar.next()
+        if update_keyword_scores:
+            KeywordScores.update_page(token_id, token_secret, keyword_scores)
+            bar.next()
+
+        bar.finish()
+
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser(description='Update wiki pages.')
@@ -67,3 +89,5 @@ if __name__ == '__main__':
         token = json.load(json_file)
 
     update_wiki(token['token_id'], token['token_secret'], not args.do_not_update_csv, args.pages)
+
+    update_wiki(args.token_id, args.token_secret, args.update_csv, args.pages)