Compare commits

..

3 Commits

Author SHA1 Message Date
Sarah
2435776796 fix typos in docs 2022-04-15 17:12:22 +01:00
Sarah
3aeb4bf8ee fix keyword score table generation 2022-04-15 17:10:09 +01:00
Sarah
935f1e25ca change hidden theme calculation 2022-04-15 17:09:56 +01:00
3 changed files with 10 additions and 13 deletions

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

View File

@ -68,7 +68,7 @@ def build_page(keyword_data, minimum_score=1.0):
row_data = [] row_data = []
row_data.append(str(keyword)) row_data.append(str(keyword))
row_data.append(str(len(data['vcinema_films']))) row_data.append(str(len(data['vcinema_films'])))
row_data.append(str(len(data['total']))) row_data.append(str(data['total']))
row_data.append(str(round(data['score'], 3))) row_data.append(str(round(data['score'], 3)))
table += " | ".join(row_data) table += " | ".join(row_data)

View File

@ -100,12 +100,12 @@ if __name__ == '__main__':
parser.add_argument('token_id', help='API token ID.') parser.add_argument('token_id', help='API token ID.')
parser.add_argument('token_secret', help='API token secret.') parser.add_argument('token_secret', help='API token secret.')
parser.add_argument('-update_csv', help='Update viewings.csv file, default: True', default=False) parser.add_argument('--update_csv', help='Update viewings.csv file, default: True', default=True, required=False)
parser.add_argument('--update_films_by_year', help='Update films by release year page, default: True', default=False, required=False) parser.add_argument('--update_films_by_year', help='Update films by release year page, default: True', default=True, required=False)
parser.add_argument('--update_films_by_country', help='Update films by country page, default: True', default=True, required=False) parser.add_argument('--update_films_by_country', help='Update films by country page, default: True', default=True, required=False)
parser.add_argument('--update_films_by_reference', help='Update films by reference page, default: True', default=False, required=False) parser.add_argument('--update_films_by_reference', help='Update films by reference page, default: True', default=True, required=False)
parser.add_argument('--update_hidden_themes', help='Update hidden themese page, default: True', default=False, required=False) parser.add_argument('--update_hidden_themes', help='Update hidden themes page, default: True', default=True, required=False)
parser.add_argument('--update_keyword_scores', help='Upate keyword scores page - THIS TAKES A WHILE, default: False', default=False, required=False) parser.add_argument('--update_keyword_scores', help='Update keyword scores page - THIS TAKES A WHILE, default: False', default=False, required=False)
args = parser.parse_args() args = parser.parse_args()