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

black formatting

pull/33/head
Luigi311 2022-12-23 23:11:38 -07:00
parent 111e284cc8
commit 1eb92cf7c1
3 changed files with 30 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/jellyfin.py
View File

@ -104,11 +104,20 @@ class Jellyfin:
and "MediaSources" in movie and "MediaSources" in movie
and movie["MediaSources"] is not {} 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: 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: 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 # Create a dictionary for the movie with its title
movie_guids = {"title": movie["Name"]} movie_guids = {"title": movie["Name"]}
@ -133,7 +142,10 @@ class Jellyfin:
# Append the movie dictionary to the list for the given user and library # Append the movie dictionary to the list for the given user and library
user_watched[user_name][library_title].append(movie_guids) 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 # TV Shows
if library_type == "Series": if library_type == "Series":
@ -158,7 +170,10 @@ class Jellyfin:
# Create a list of tasks to retrieve the seasons of each watched show # Create a list of tasks to retrieve the seasons of each watched show
seasons_tasks = [] seasons_tasks = []
for show in watched_shows_filtered: 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 = { show_guids = {
k.lower(): v for k, v in show["ProviderIds"].items() k.lower(): v for k, v in show["ProviderIds"].items()
} }
@ -277,15 +292,16 @@ class Jellyfin:
][season_dict["Identifiers"]["season_name"]] = season_dict[ ][season_dict["Identifiers"]["season_name"]] = season_dict[
"Episodes" "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( logger(
f"Jellyfin: Got watched for {user_name} in library {library_title}", 1 f"Jellyfin: Got watched for {user_name} in library {library_title}", 1
) )
if library_title in user_watched[user_name]: if library_title in user_watched[user_name]:
logger( logger(f"Jellyfin: {user_watched[user_name][library_title]}", 3)
f"Jellyfin: {user_watched[user_name][library_title]}", 3
)
return user_watched return user_watched
except Exception as e: except Exception as e:

4
src/main.py
View File

@ -15,7 +15,6 @@ from src.jellyfin import Jellyfin
load_dotenv(override=True) load_dotenv(override=True)
def setup_users( def setup_users(
server_1, server_2, blacklist_users, whitelist_users, user_mapping=None 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): for i, baseurl in enumerate(jellyfin_baseurl):
baseurl = baseurl.strip() baseurl = baseurl.strip()
if baseurl[-1] == '/': if baseurl[-1] == "/":
baseurl = baseurl[:-1] baseurl = baseurl[:-1]
servers.append( servers.append(
( (
@ -212,7 +211,6 @@ def generate_server_connections():
) )
) )
return servers return servers

5
src/plex.py
View File

@ -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] = {}
user_watched[user_name][library.title][show_guids] = episode_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) logger(f"Plex: Got watched for {user_name} in library {library.title}", 1)
if library.title in user_watched[user_name]: if library.title in user_watched[user_name]: