Use generate_library_guids_dict instead of library type

This commit is contained in:
Luigi311
2022-07-05 18:09:08 -06:00
parent de619de923
commit a3f3db8f4e
4 changed files with 29 additions and 31 deletions

View File

@@ -80,13 +80,12 @@ def check_skip_logic(library_title, library_type, blacklist_library, whitelist_l
return skip_reason
def generate_library_guids_dict(user_list: dict, generate_output: int):
# if generate_output is 0 then only generate shows, if 1 then only generate episodes, if 2 then generate movies, if 3 then generate shows and episodes
def generate_library_guids_dict(user_list: dict):
show_output_dict = {}
episode_output_dict = {}
movies_output_dict = {}
if generate_output in (0, 3):
try:
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:
@@ -101,8 +100,10 @@ def generate_library_guids_dict(user_list: dict, generate_output: int):
show_output_dict[provider_key.lower()].append(show_location)
else:
show_output_dict[provider_key.lower()].append(provider_value.lower())
except:
pass
if generate_output in (1, 3):
try:
for show in user_list:
for season in user_list[show]:
for episode in user_list[show][season]:
@@ -114,8 +115,10 @@ def generate_library_guids_dict(user_list: dict, generate_output: int):
episode_output_dict[episode_key.lower()].append(episode_location)
else:
episode_output_dict[episode_key.lower()].append(episode_value.lower())
except:
pass
if generate_output == 2:
try:
for movie in user_list:
for movie_key, movie_value in movie.items():
if movie_key.lower() not in movies_output_dict:
@@ -125,6 +128,8 @@ def generate_library_guids_dict(user_list: dict, generate_output: int):
movies_output_dict[movie_key.lower()].append(movie_location)
else:
movies_output_dict[movie_key.lower()].append(movie_value.lower())
except:
pass
return show_output_dict, episode_output_dict, movies_output_dict