From afb71d8e00c8a5b3eb7c6e3d1e0692ae9e5a61a8 Mon Sep 17 00:00:00 2001 From: Luigi311 Date: Mon, 20 Jun 2022 16:07:52 -0600 Subject: [PATCH] Handle locations in generate_library_guids_dict --- src/functions.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/src/functions.py b/src/functions.py index 4edc73f..64050aa 100644 --- a/src/functions.py +++ b/src/functions.py @@ -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