Merge branch 'master' of git.jacknet.io:sarah/vcinema

get_hidden_themes.py

@@ -0,0 +1,37 @@
+from imdb_utils import IMDbUtils
+
+import argparse
+from progress.bar import IncrementalBar
+
+
+def get_hidden_themes(imdb_ids):
+    hidden_themes = set()
+    is_first = True
+
+    with IncrementalBar('Retrieving movie data', max=len(imdb_ids), suffix='%(percent).1f%% - %(eta)ds remaining', check_tty=False) as bar:
+        for imdb_id in imdb_ids:
+            movie_data = IMDbUtils.get_movie_keywords(imdb_id)
+
+            if 'keywords' in movie_data:
+                keywords = set(movie_data['keywords'])
+                if is_first:
+                    hidden_themes = keywords
+                    is_first = False
+                else:
+                    hidden_themes = hidden_themes.intersection(keywords)
+
+        bar.next()
+
+    return hidden_themes
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument('imdb_ids', nargs="+", default=[])
+
+    args = parser.parse_args()
+
+    hidden_themes = get_hidden_themes(args.imdb_ids)
+
+    print(hidden_themes)
+