Support user mapping
This commit is contained in:
Luigi311
204
main.py
204
main.py
@@ -1,78 +1,78 @@
|
||||
import copy, os, traceback
|
||||
import copy, os, traceback, json
|
||||
from dotenv import load_dotenv
|
||||
from time import sleep
|
||||
|
||||
from src.functions import logger, str_to_bool
|
||||
from src.functions import logger, str_to_bool, search_mapping
|
||||
from src.plex import Plex
|
||||
from src.jellyfin import Jellyfin
|
||||
|
||||
load_dotenv(override=True)
|
||||
|
||||
def cleanup_watched(watched_list_1, watched_list_2):
|
||||
def cleanup_watched(watched_list_1, watched_list_2, user_mapping):
|
||||
modified_watched_list_1 = copy.deepcopy(watched_list_1)
|
||||
|
||||
# remove entries from plex_watched that are in jellyfin_watched
|
||||
for user in watched_list_1:
|
||||
if user in modified_watched_list_1:
|
||||
for library in watched_list_1[user]:
|
||||
if library in modified_watched_list_1[user]:
|
||||
for item in watched_list_1[user][library]:
|
||||
if item in modified_watched_list_1[user][library]:
|
||||
if user in watched_list_2 and library in watched_list_2[user]:
|
||||
# Movies
|
||||
if isinstance(watched_list_1[user][library], list):
|
||||
for watch_list_1_key, watch_list_1_value in item.items():
|
||||
for watch_list_2_item in watched_list_2[user][library]:
|
||||
for watch_list_2_item_key, watch_list_2_item_value in watch_list_2_item.items():
|
||||
if watch_list_1_key == watch_list_2_item_key and watch_list_1_value == watch_list_2_item_value:
|
||||
if item in modified_watched_list_1[user][library]:
|
||||
modified_watched_list_1[user][library].remove(item)
|
||||
|
||||
# TV Shows
|
||||
elif isinstance(watched_list_1[user][library], dict):
|
||||
if item in watched_list_2[user][library]:
|
||||
for season in watched_list_1[user][library][item]:
|
||||
if season in watched_list_2[user][library][item]:
|
||||
for episode in watched_list_1[user][library][item][season]:
|
||||
for watch_list_1_episode_key, watch_list_1_episode_value in episode.items():
|
||||
for watch_list_2_episode in watched_list_2[user][library][item][season]:
|
||||
for watch_list_2_episode_key, watch_list_2_episode_value in watch_list_2_episode.items():
|
||||
if watch_list_1_episode_key == watch_list_2_episode_key and watch_list_1_episode_value == watch_list_2_episode_value:
|
||||
if episode in modified_watched_list_1[user][library][item][season]:
|
||||
modified_watched_list_1[user][library][item][season].remove(episode)
|
||||
|
||||
# If season is empty, remove season
|
||||
if len(modified_watched_list_1[user][library][item][season]) == 0:
|
||||
if season in modified_watched_list_1[user][library][item]:
|
||||
del modified_watched_list_1[user][library][item][season]
|
||||
for user_1 in watched_list_1:
|
||||
user_2 = search_mapping(user_mapping, user_1)
|
||||
if user_1 in modified_watched_list_1:
|
||||
for library in watched_list_1[user_1]:
|
||||
if library in modified_watched_list_1[user_1]:
|
||||
for item in watched_list_1[user_1][library]:
|
||||
if item in modified_watched_list_1[user_1][library]:
|
||||
if user_1 in watched_list_2:
|
||||
user = user_1
|
||||
elif user_2 in watched_list_2:
|
||||
user = user_2
|
||||
else:
|
||||
print(f"User {user_1} and {user_2} not found in watched list 2")
|
||||
user = None
|
||||
|
||||
if user:
|
||||
if library in watched_list_2[user]:
|
||||
# Movies
|
||||
if isinstance(watched_list_1[user_1][library], list):
|
||||
for watch_list_1_key, watch_list_1_value in item.items():
|
||||
for watch_list_2_item in watched_list_2[user][library]:
|
||||
for watch_list_2_item_key, watch_list_2_item_value in watch_list_2_item.items():
|
||||
if watch_list_1_key == watch_list_2_item_key and watch_list_1_value == watch_list_2_item_value:
|
||||
if item in modified_watched_list_1[user_1][library]:
|
||||
modified_watched_list_1[user_1][library].remove(item)
|
||||
|
||||
# TV Shows
|
||||
elif isinstance(watched_list_1[user_1][library], dict):
|
||||
if item in watched_list_2[user][library]:
|
||||
for season in watched_list_1[user_1][library][item]:
|
||||
if season in watched_list_2[user][library][item]:
|
||||
for episode in watched_list_1[user_1][library][item][season]:
|
||||
for watch_list_1_episode_key, watch_list_1_episode_value in episode.items():
|
||||
for watch_list_2_episode in watched_list_2[user][library][item][season]:
|
||||
for watch_list_2_episode_key, watch_list_2_episode_value in watch_list_2_episode.items():
|
||||
if watch_list_1_episode_key == watch_list_2_episode_key and watch_list_1_episode_value == watch_list_2_episode_value:
|
||||
if episode in modified_watched_list_1[user_1][library][item][season]:
|
||||
modified_watched_list_1[user_1][library][item][season].remove(episode)
|
||||
|
||||
# If season is empty, remove season
|
||||
if len(modified_watched_list_1[user_1][library][item][season]) == 0:
|
||||
if season in modified_watched_list_1[user_1][library][item]:
|
||||
del modified_watched_list_1[user_1][library][item][season]
|
||||
|
||||
# If the show is empty, remove the show
|
||||
if len(modified_watched_list_1[user][library][item]) == 0:
|
||||
if item in modified_watched_list_1[user][library]:
|
||||
del modified_watched_list_1[user][library][item]
|
||||
# If the show is empty, remove the show
|
||||
if len(modified_watched_list_1[user_1][library][item]) == 0:
|
||||
if item in modified_watched_list_1[user_1][library]:
|
||||
del modified_watched_list_1[user_1][library][item]
|
||||
|
||||
# If library is empty then remove it
|
||||
if len(modified_watched_list_1[user][library]) == 0:
|
||||
if library in modified_watched_list_1[user]:
|
||||
del modified_watched_list_1[user][library]
|
||||
if len(modified_watched_list_1[user_1][library]) == 0:
|
||||
if library in modified_watched_list_1[user_1]:
|
||||
del modified_watched_list_1[user_1][library]
|
||||
|
||||
# If user is empty delete user
|
||||
if len(modified_watched_list_1[user]) == 0:
|
||||
del modified_watched_list_1[user]
|
||||
if len(modified_watched_list_1[user_1]) == 0:
|
||||
del modified_watched_list_1[user_1]
|
||||
|
||||
return modified_watched_list_1
|
||||
|
||||
def main():
|
||||
logfile = os.getenv("LOGFILE","log.log")
|
||||
# Delete logfile if it exists
|
||||
if os.path.exists(logfile):
|
||||
os.remove(logfile)
|
||||
|
||||
dryrun = str_to_bool(os.getenv("DRYRUN", "False"))
|
||||
logger(f"Dryrun: {dryrun}", 1)
|
||||
plex = Plex()
|
||||
jellyfin = Jellyfin()
|
||||
|
||||
def setup_black_white_lists():
|
||||
blacklist_library = os.getenv("BLACKLIST_LIBRARY")
|
||||
if blacklist_library:
|
||||
if len(blacklist_library) > 0:
|
||||
@@ -130,43 +130,95 @@ def main():
|
||||
whitelist_users = []
|
||||
logger(f"Whitelist Users: {whitelist_users}", 1)
|
||||
|
||||
return blacklist_library, whitelist_library, blacklist_library_type, whitelist_library_type, blacklist_users, whitelist_users
|
||||
|
||||
def setup_users(plex, jellyfin, blacklist_users, whitelist_users, user_mapping=None):
|
||||
|
||||
# generate list of users from plex.users
|
||||
plex_users = [ x.title.lower() for x in plex.users ]
|
||||
jellyfin_users = [ key.lower() for key in jellyfin.users.keys() ]
|
||||
|
||||
# combined list of overlapping users from plex and jellyfin
|
||||
users = [x for x in plex_users if x in jellyfin_users]
|
||||
users = {}
|
||||
|
||||
for plex_user in plex_users:
|
||||
if user_mapping:
|
||||
jellyfin_plex_mapped_user = search_mapping(user_mapping, plex_user)
|
||||
if jellyfin_plex_mapped_user:
|
||||
users[plex_user] = jellyfin_plex_mapped_user
|
||||
break
|
||||
|
||||
if plex_user in jellyfin_users:
|
||||
users[plex_user] = plex_user
|
||||
|
||||
for jellyfin_user in jellyfin_users:
|
||||
if user_mapping:
|
||||
plex_jellyfin_mapped_user = search_mapping(user_mapping, jellyfin_user)
|
||||
if plex_jellyfin_mapped_user:
|
||||
users[plex_jellyfin_mapped_user] = jellyfin_user
|
||||
break
|
||||
|
||||
if jellyfin_user in plex_users:
|
||||
users[jellyfin_user] = jellyfin_user
|
||||
|
||||
logger(f"User list that exist on both servers {users}", 1)
|
||||
|
||||
users_filtered = []
|
||||
users_filtered = {}
|
||||
for user in users:
|
||||
# whitelist_user is not empty and user lowercase is not in whitelist lowercase
|
||||
if len(whitelist_users) > 0 and user.lower() not in whitelist_users:
|
||||
logger(f"{user} is not in whitelist", 1)
|
||||
else:
|
||||
if user.lower() not in blacklist_users:
|
||||
users_filtered.append(user)
|
||||
if len(whitelist_users) > 0:
|
||||
if user not in whitelist_users and users[user] not in whitelist_users:
|
||||
logger(f"{user} or {users[user]} is not in whitelist", 1)
|
||||
break
|
||||
|
||||
if user not in blacklist_users and users[user] not in blacklist_users:
|
||||
users_filtered[user] = users[user]
|
||||
|
||||
logger(f"Filtered user list {users_filtered}", 1)
|
||||
|
||||
plex_users = []
|
||||
for plex_user in plex.users:
|
||||
if plex_user.title.lower() in users_filtered:
|
||||
if plex_user.title.lower() in users_filtered.keys() or plex_user.title.lower() in users_filtered.values():
|
||||
plex_users.append(plex_user)
|
||||
|
||||
jellyfin_users = {}
|
||||
for jellyfin_user, jellyfin_id in jellyfin.users.items():
|
||||
if jellyfin_user.lower() in users_filtered:
|
||||
if jellyfin_user.lower() in users_filtered.keys() or jellyfin_user.lower() in users_filtered.values():
|
||||
jellyfin_users[jellyfin_user] = jellyfin_id
|
||||
|
||||
if len(plex_users) == 0:
|
||||
raise Exception("No plex users found")
|
||||
|
||||
if len(jellyfin_users) == 0:
|
||||
raise Exception("No jellyfin users found")
|
||||
|
||||
logger(f"plex_users: {plex_users}", 1)
|
||||
logger(f"jellyfin_users: {jellyfin_users}", 1)
|
||||
|
||||
if len(plex_users) == 0:
|
||||
logger("No users found", 2)
|
||||
raise Exception("No users found")
|
||||
return plex_users, jellyfin_users
|
||||
|
||||
if len(jellyfin_users) == 0:
|
||||
logger("No users found", 2)
|
||||
raise Exception("No users found")
|
||||
def main():
|
||||
logfile = os.getenv("LOGFILE","log.log")
|
||||
# Delete logfile if it exists
|
||||
if os.path.exists(logfile):
|
||||
os.remove(logfile)
|
||||
|
||||
dryrun = str_to_bool(os.getenv("DRYRUN", "False"))
|
||||
logger(f"Dryrun: {dryrun}", 1)
|
||||
|
||||
user_mapping = os.getenv("USER_MAPPING")
|
||||
if user_mapping:
|
||||
user_mapping = json.loads(user_mapping.lower())
|
||||
logger(f"User Mapping: {user_mapping}", 1)
|
||||
|
||||
plex = Plex()
|
||||
jellyfin = Jellyfin()
|
||||
|
||||
# Create (black/white)lists
|
||||
blacklist_library, whitelist_library, blacklist_library_type, whitelist_library_type, blacklist_users, whitelist_users = setup_black_white_lists()
|
||||
|
||||
# Create users list
|
||||
plex_users, jellyfin_users = setup_users(plex, jellyfin, blacklist_users, whitelist_users, user_mapping)
|
||||
|
||||
plex_watched = plex.get_plex_watched(plex_users, blacklist_library, whitelist_library, blacklist_library_type, whitelist_library_type)
|
||||
jellyfin_watched = jellyfin.get_jellyfin_watched(jellyfin_users, blacklist_library, whitelist_library, blacklist_library_type, whitelist_library_type)
|
||||
@@ -175,15 +227,15 @@ def main():
|
||||
plex_watched_filtered = copy.deepcopy(plex_watched)
|
||||
jellyfin_watched_filtered = copy.deepcopy(jellyfin_watched)
|
||||
|
||||
plex_watched = cleanup_watched(plex_watched_filtered, jellyfin_watched_filtered)
|
||||
plex_watched = cleanup_watched(plex_watched_filtered, jellyfin_watched_filtered, user_mapping)
|
||||
logger(f"plex_watched that needs to be synced to jellyfin:\n{plex_watched}", 1)
|
||||
|
||||
jellyfin_watched = cleanup_watched(jellyfin_watched_filtered, plex_watched_filtered)
|
||||
jellyfin_watched = cleanup_watched(jellyfin_watched_filtered, plex_watched_filtered, user_mapping)
|
||||
logger(f"jellyfin_watched that needs to be synced to plex:\n{jellyfin_watched}", 1)
|
||||
|
||||
# Update watched status
|
||||
plex.update_watched(jellyfin_watched, dryrun)
|
||||
jellyfin.update_watched(plex_watched, dryrun)
|
||||
plex.update_watched(jellyfin_watched, user_mapping, dryrun)
|
||||
jellyfin.update_watched(plex_watched, user_mapping, dryrun)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
@@ -202,7 +254,7 @@ if __name__ == "__main__":
|
||||
|
||||
|
||||
logger(traceback.format_exc(), 2)
|
||||
logger("Retrying in {sleep_timer}", log_type=0)
|
||||
logger(f"Retrying in {sleep_timer}", log_type=0)
|
||||
|
||||
except KeyboardInterrupt:
|
||||
logger("Exiting", log_type=0)
|
||||
|
||||
Reference in New Issue
Block a user