feat: Jellyfin integration finished
- Finsihed Jellyfin integration with auto-add and auto-remove - Integration toggle in config task: none
This commit is contained in:
175
app/bot/helper/jellyfinhelper.py
Normal file
175
app/bot/helper/jellyfinhelper.py
Normal file
@@ -0,0 +1,175 @@
|
||||
import requests
|
||||
import random
|
||||
import string
|
||||
|
||||
def add_user(jellyfin_url, jellyfin_api_key, username, password, jellyfin_libs):
|
||||
try:
|
||||
url = f"{jellyfin_url}/Users/New"
|
||||
|
||||
querystring = {"api_key":jellyfin_api_key}
|
||||
payload = {
|
||||
"Name": username,
|
||||
"Password": password
|
||||
}
|
||||
headers = {"Content-Type": "application/json"}
|
||||
response = requests.request("POST", url, json=payload, headers=headers, params=querystring)
|
||||
userId = response.json()["Id"]
|
||||
|
||||
if response.status_code != 200:
|
||||
print(f"Error creating new Jellyfin user: {response.text}")
|
||||
return False
|
||||
|
||||
# Grant access to User
|
||||
url = f"{jellyfin_url}/Users/{userId}/Policy"
|
||||
|
||||
querystring = {"api_key":jellyfin_api_key}
|
||||
|
||||
enabled_folders = []
|
||||
server_libs = get_libraries(jellyfin_url, jellyfin_api_key)
|
||||
|
||||
if jellyfin_libs[0] != "all":
|
||||
for lib in jellyfin_libs:
|
||||
found = False
|
||||
for server_lib in server_libs:
|
||||
if lib == server_lib['Name']:
|
||||
enabled_folders.append(server_lib['ItemId'])
|
||||
found = True
|
||||
if not found:
|
||||
print(f"Couldn't find Jellyfin Library: {lib}")
|
||||
|
||||
payload = {
|
||||
"IsAdministrator": False,
|
||||
"IsHidden": True,
|
||||
"IsDisabled": False,
|
||||
"BlockedTags": [],
|
||||
"EnableUserPreferenceAccess": True,
|
||||
"AccessSchedules": [],
|
||||
"BlockUnratedItems": [],
|
||||
"EnableRemoteControlOfOtherUsers": False,
|
||||
"EnableSharedDeviceControl": True,
|
||||
"EnableRemoteAccess": True,
|
||||
"EnableLiveTvManagement": True,
|
||||
"EnableLiveTvAccess": True,
|
||||
"EnableMediaPlayback": True,
|
||||
"EnableAudioPlaybackTranscoding": True,
|
||||
"EnableVideoPlaybackTranscoding": True,
|
||||
"EnablePlaybackRemuxing": True,
|
||||
"ForceRemoteSourceTranscoding": False,
|
||||
"EnableContentDeletion": False,
|
||||
"EnableContentDeletionFromFolders": [],
|
||||
"EnableContentDownloading": True,
|
||||
"EnableSyncTranscoding": True,
|
||||
"EnableMediaConversion": True,
|
||||
"EnabledDevices": [],
|
||||
"EnableAllDevices": True,
|
||||
"EnabledChannels": [],
|
||||
"EnableAllChannels": False,
|
||||
"EnabledFolders": enabled_folders,
|
||||
"EnableAllFolders": jellyfin_libs[0] == "all",
|
||||
"InvalidLoginAttemptCount": 0,
|
||||
"LoginAttemptsBeforeLockout": -1,
|
||||
"MaxActiveSessions": 0,
|
||||
"EnablePublicSharing": True,
|
||||
"BlockedMediaFolders": [],
|
||||
"BlockedChannels": [],
|
||||
"RemoteClientBitrateLimit": 0,
|
||||
"AuthenticationProviderId": "Jellyfin.Server.Implementations.Users.DefaultAuthenticationProvider",
|
||||
"PasswordResetProviderId": "Jellyfin.Server.Implementations.Users.DefaultPasswordResetProvider",
|
||||
"SyncPlayAccess": "CreateAndJoinGroups"
|
||||
}
|
||||
headers = {"content-type": "application/json"}
|
||||
|
||||
response = requests.request("POST", url, json=payload, headers=headers, params=querystring)
|
||||
|
||||
if response.status_code == 200 or response.status_code == 204:
|
||||
return True
|
||||
else:
|
||||
print(f"Error setting user permissions: {response.text}")
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
return False
|
||||
|
||||
def get_libraries(jellyfin_url, jellyfin_api_key):
|
||||
url = f"{jellyfin_url}/Library/VirtualFolders"
|
||||
querystring = {"api_key":jellyfin_api_key}
|
||||
response = requests.request("GET", url, params=querystring)
|
||||
|
||||
return response.json()
|
||||
|
||||
|
||||
def verify_username(jellyfin_url, jellyfin_api_key, username):
|
||||
users = get_users(jellyfin_url, jellyfin_api_key)
|
||||
valid = True
|
||||
for user in users:
|
||||
if user['Name'] == username:
|
||||
valid = False
|
||||
break
|
||||
|
||||
return valid
|
||||
|
||||
def remove_user(jellyfin_url, jellyfin_api_key, jellyfin_username):
|
||||
try:
|
||||
# Get User ID
|
||||
users = get_users(jellyfin_url, jellyfin_api_key)
|
||||
userId = None
|
||||
for user in users:
|
||||
if user['Name'].lower() == jellyfin_username.lower():
|
||||
userId = user['Id']
|
||||
|
||||
if userId is None:
|
||||
# User not found
|
||||
print(f"Error removing user {jellyfin_username} from Jellyfin: Could not find user.")
|
||||
return False
|
||||
|
||||
# Delete User
|
||||
url = f"{jellyfin_url}/Users/{userId}"
|
||||
|
||||
querystring = {"api_key":jellyfin_api_key}
|
||||
response = requests.request("DELETE", url, params=querystring)
|
||||
|
||||
if response.status_code == 204 or response.status_code == 200:
|
||||
return True
|
||||
else:
|
||||
print(f"Error deleting Jellyfin user: {response.text}")
|
||||
except Exception as e:
|
||||
print(e)
|
||||
return False
|
||||
|
||||
def get_users(jellyfin_url, jellyfin_api_key):
|
||||
url = f"{jellyfin_url}/Users"
|
||||
|
||||
querystring = {"api_key":jellyfin_api_key}
|
||||
response = requests.request("GET", url, params=querystring)
|
||||
|
||||
return response.json()
|
||||
|
||||
def generate_password(length, lower=True, upper=True, numbers=True, symbols=True):
|
||||
character_list = []
|
||||
if not (lower or upper or numbers or symbols):
|
||||
raise ValueError("At least one character type must be provided")
|
||||
|
||||
if lower:
|
||||
character_list += string.ascii_lowercase
|
||||
if upper:
|
||||
character_list += string.ascii_uppercase
|
||||
if numbers:
|
||||
character_list += string.digits
|
||||
if symbols:
|
||||
character_list += string.punctuation
|
||||
|
||||
return "".join(random.choice(character_list) for i in range(length))
|
||||
|
||||
def get_config(jellyfin_url, jellyfin_api_key):
|
||||
url = f"{jellyfin_url}/System/Configuration"
|
||||
|
||||
querystring = {"api_key":jellyfin_api_key}
|
||||
response = requests.request("GET", url, params=querystring)
|
||||
return response.json()
|
||||
|
||||
def get_status(jellyfin_url, jellyfin_api_key):
|
||||
url = f"{jellyfin_url}/System/Configuration"
|
||||
|
||||
querystring = {"api_key":jellyfin_api_key}
|
||||
response = requests.request("GET", url, params=querystring)
|
||||
return response.status_code
|
||||
Reference in New Issue
Block a user