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

add keywords to data

Browse Source
This commit is contained in:
Sarah 2022-04-03 20:27:09 +01:00
parent 981bf3bb41
commit 06ea4e0754
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
vcinema_utils/VCinemaUtils.py
View File

@ -68,11 +68,27 @@ def add_imdb_data(imdb_id, viewings, data_fields, progressbar=None):
progressbar.next()
def add_imdb_keywords(imdb_id, viewings, progressbar=None):
movie = IMDbUtils.get_movie_keywords(imdb_id)
for viewing in viewings:
if viewing['imdb_id'] == movie.movieID:
if 'keywords' in movie:
viewing['keywords'] = movie['keywords']
if progressbar is not None:
progressbar.next()
def add_imdb_data_to_viewings(viewings, field_names, progress_bar=None):
with ThreadPoolExecutor(4) as executor:
future_imdb_tasks = set()
future_imdb_tasks.update(executor.submit(add_imdb_data, viewing['imdb_id'], viewings, field_names, progress_bar) for viewing in viewings)
if ('keywords' in field_names and field_names > 1) or ('keywords' not in field_names and field_names > 0):
future_imdb_tasks.update(executor.submit(add_imdb_data, viewing['imdb_id'], viewings, field_names, progress_bar) for viewing in viewings)
if 'keywords' in field_names:
future_imdb_tasks.update(executor.submit(add_imdb_keywords, viewing['imdb_id'], viewings, progress_bar) for viewing in viewings)
progress_bar.max = len(future_imdb_tasks)