bel/JellyPlex-Watched
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Handle locations in generate_library_guids_dict

Browse Source
This commit is contained in:
Luigi311
2022-06-20 16:07:52 -06:00
parent 34d97f8dde
commit afb71d8e00
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/functions.py
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 = user_list.keys()
show_output_keys = ([ dict(x) for x in list(show_output_keys) ]) show_output_keys = ([ dict(x) for x in list(show_output_keys) ])
for show_key in 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 # Skip title
if provider_key.lower() == "title": if provider_key.lower() == "title":
continue continue
if provider_key.lower() not in show_output_dict: if provider_key.lower() not in show_output_dict:
show_output_dict[provider_key.lower()] = [] 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): if generate_output in (1, 3):
for show in user_list: for show in user_list:
@@ -105,6 +108,9 @@ def generate_library_guids_dict(user_list: dict, generate_output: int):
for episode_key, episode_value in episode.items(): for episode_key, episode_value in episode.items():
if episode_key.lower() not in episode_output_dict: if episode_key.lower() not in episode_output_dict:
episode_output_dict[episode_key.lower()] = [] episode_output_dict[episode_key.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()) episode_output_dict[episode_key.lower()].append(episode_value.lower())
if generate_output == 2: if generate_output == 2:
@@ -112,6 +118,9 @@ def generate_library_guids_dict(user_list: dict, generate_output: int):
for movie_key, movie_value in movie.items(): for movie_key, movie_value in movie.items():
if movie_key.lower() not in movies_output_dict: if movie_key.lower() not in movies_output_dict:
movies_output_dict[movie_key.lower()] = [] movies_output_dict[movie_key.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()) movies_output_dict[movie_key.lower()].append(movie_value.lower())
return show_output_dict, episode_output_dict, movies_output_dict return show_output_dict, episode_output_dict, movies_output_dict