diff --git a/src/jellyfin.py b/src/jellyfin.py index 84fb130..9658427 100644 --- a/src/jellyfin.py +++ b/src/jellyfin.py @@ -104,11 +104,20 @@ class Jellyfin: and "MediaSources" in movie and movie["MediaSources"] is not {} ): - logger(f"Jellyfin: Adding {movie['Name']} to {user_name} watched list", 3) + logger( + f"Jellyfin: Adding {movie['Name']} to {user_name} watched list", + 3, + ) if "ProviderIds" in movie: - logger(f"Jellyfin: {movie['Name']} {movie['ProviderIds']} {movie['MediaSources']}", 3) + logger( + f"Jellyfin: {movie['Name']} {movie['ProviderIds']} {movie['MediaSources']}", + 3, + ) else: - logger(f"Jellyfin: {movie['Name']} {movie['MediaSources']['Path']}", 3) + logger( + f"Jellyfin: {movie['Name']} {movie['MediaSources']['Path']}", + 3, + ) # Create a dictionary for the movie with its title movie_guids = {"title": movie["Name"]} @@ -133,7 +142,10 @@ class Jellyfin: # Append the movie dictionary to the list for the given user and library user_watched[user_name][library_title].append(movie_guids) - logger(f"Jellyfin: Added {movie_guids} to {user_name} watched list", 3) + logger( + f"Jellyfin: Added {movie_guids} to {user_name} watched list", + 3, + ) # TV Shows if library_type == "Series": @@ -158,7 +170,10 @@ class Jellyfin: # Create a list of tasks to retrieve the seasons of each watched show seasons_tasks = [] for show in watched_shows_filtered: - logger(f"Jellyfin: Adding {show['Name']} to {user_name} watched list", 3) + logger( + f"Jellyfin: Adding {show['Name']} to {user_name} watched list", + 3, + ) show_guids = { k.lower(): v for k, v in show["ProviderIds"].items() } @@ -277,15 +292,16 @@ class Jellyfin: ][season_dict["Identifiers"]["season_name"]] = season_dict[ "Episodes" ] - logger(f"Jellyfin: Added {season_dict['Episodes']} to {user_name} {season_dict['Identifiers']['show_guids']} watched list", 1) + logger( + f"Jellyfin: Added {season_dict['Episodes']} to {user_name} {season_dict['Identifiers']['show_guids']} watched list", + 1, + ) logger( f"Jellyfin: Got watched for {user_name} in library {library_title}", 1 ) if library_title in user_watched[user_name]: - logger( - f"Jellyfin: {user_watched[user_name][library_title]}", 3 - ) + logger(f"Jellyfin: {user_watched[user_name][library_title]}", 3) return user_watched except Exception as e: diff --git a/src/main.py b/src/main.py index 9302f3e..153ec71 100644 --- a/src/main.py +++ b/src/main.py @@ -15,7 +15,6 @@ from src.jellyfin import Jellyfin load_dotenv(override=True) - def setup_users( server_1, server_2, blacklist_users, whitelist_users, user_mapping=None ): @@ -203,7 +202,7 @@ def generate_server_connections(): for i, baseurl in enumerate(jellyfin_baseurl): baseurl = baseurl.strip() - if baseurl[-1] == '/': + if baseurl[-1] == "/": baseurl = baseurl[:-1] servers.append( ( @@ -212,7 +211,6 @@ def generate_server_connections(): ) ) - return servers diff --git a/src/plex.py b/src/plex.py index 87724da..a91b81d 100644 --- a/src/plex.py +++ b/src/plex.py @@ -101,7 +101,10 @@ def get_user_library_watched(user, user_plex, library): user_watched[user_name][library.title][show_guids] = {} user_watched[user_name][library.title][show_guids] = episode_guids - logger(f"Plex: Added {episode_guids} to {user_name} {show_guids} watched list", 3) + logger( + f"Plex: Added {episode_guids} to {user_name} {show_guids} watched list", + 3, + ) logger(f"Plex: Got watched for {user_name} in library {library.title}", 1) if library.title in user_watched[user_name]: