sarah/vcinema
1
2
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

change hidden theme calculation

Browse Source
This commit is contained in:
Sarah 2022-04-15 17:09:56 +01:00
parent 14c3e2a76c
commit 935f1e25ca
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
get_hidden_themes.py
View File

@ -5,8 +5,7 @@ from progress.bar import IncrementalBar
def get_hidden_themes(imdb_ids): def get_hidden_themes(imdb_ids):
hidden_themes = set() film_keywords = []
is_first = True
with IncrementalBar('Retrieving movie data', max=len(imdb_ids), suffix='%(percent).1f%% - %(eta)ds remaining', check_tty=False) as bar: 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: for imdb_id in imdb_ids:
@ -14,14 +13,12 @@ def get_hidden_themes(imdb_ids):
if 'keywords' in movie_data: if 'keywords' in movie_data:
keywords = set(movie_data['keywords']) keywords = set(movie_data['keywords'])
if is_first: film_keywords.append(keywords)
hidden_themes = keywords
is_first = False
else:
hidden_themes = hidden_themes.intersection(keywords)
bar.next() bar.next()
hidden_themes = set.intersection(*film_keywords)
return hidden_themes return hidden_themes