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)