Handle locations in generate_library_guids_dict

This commit is contained in:
Luigi311
2022-06-20 16:07:52 -06:00
parent 34d97f8dde
commit afb71d8e00

View File

@@ -90,13 +90,16 @@ def generate_library_guids_dict(user_list: dict, generate_output: int):
show_output_keys = user_list.keys()
show_output_keys = ([ dict(x) for x in list(show_output_keys) ])
for show_key in show_output_keys:
for provider_key, prvider_value in show_key.items():
for provider_key, provider_value in show_key.items():
# Skip title
if provider_key.lower() == "title":
continue
if provider_key.lower() not in show_output_dict:
show_output_dict[provider_key.lower()] = []
show_output_dict[provider_key.lower()].append(prvider_value.lower())
if provider_key.lower() == "locations":
show_output_dict[provider_key.lower()].append(provider_value)
else:
show_output_dict[provider_key.lower()].append(provider_value.lower())
if generate_output in (1, 3):
for show in user_list:
@@ -105,14 +108,20 @@ def generate_library_guids_dict(user_list: dict, generate_output: int):
for episode_key, episode_value in episode.items():
if episode_key.lower() not in episode_output_dict:
episode_output_dict[episode_key.lower()] = []
episode_output_dict[episode_key.lower()].append(episode_value.lower())
if episode_key == "locations":
episode_output_dict[episode_key.lower()].append(episode_value)
else:
episode_output_dict[episode_key.lower()].append(episode_value.lower())
if generate_output == 2:
for movie in user_list:
for movie_key, movie_value in movie.items():
if movie_key.lower() not in movies_output_dict:
movies_output_dict[movie_key.lower()] = []
movies_output_dict[movie_key.lower()].append(movie_value.lower())
if movie_key == "locations":
movies_output_dict[movie_key.lower()].append(movie_value)
else:
movies_output_dict[movie_key.lower()].append(movie_value.lower())
return show_output_dict, episode_output_dict, movies_output_dict