Parker
00d217f453
Jellyfin accounts never actually got deleted from Jellyfin, I forgot to write that part. So now the accounts are properly deleted from both Jellyfin and the DB.
87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
import datetime
|
|
import requests
|
|
import random
|
|
import string
|
|
import sqlite3
|
|
|
|
from global_variables import JELLYFIN_URL, JELLYFIN_HEADERS, ACCOUNT_TIME
|
|
|
|
"""
|
|
Create a new Jellyfin account for the user and return the username and password
|
|
"""
|
|
|
|
|
|
def create_jellyfin_account(user_id):
|
|
username = "".join(random.choices(string.ascii_lowercase + string.digits, k=5))
|
|
password = "".join(random.choices(string.ascii_lowercase + string.digits, k=15))
|
|
|
|
deletion_time = datetime.datetime.now() + datetime.timedelta(hours=ACCOUNT_TIME)
|
|
# Create the new Jellyfin account
|
|
request_1 = requests.post(
|
|
f"{JELLYFIN_URL}/Users/New",
|
|
headers=JELLYFIN_HEADERS,
|
|
json={"Name": username, "Password": password},
|
|
)
|
|
if request_1.status_code != 200:
|
|
return False
|
|
|
|
# Get the user ID of the new account
|
|
jellyfin_user_id = request_1.json()["Id"]
|
|
# Get the account policy and make edits
|
|
request_2 = requests.get(
|
|
f"{JELLYFIN_URL}/Users/{jellyfin_user_id}", headers=JELLYFIN_HEADERS
|
|
)
|
|
if request_2.status_code != 200:
|
|
return False
|
|
|
|
account_policy = request_2.json()
|
|
account_policy["Policy"]["SyncPlayAccess"] = "JoinGroups"
|
|
account_policy["Policy"]["EnableContentDownloading"] = False
|
|
account_policy["Policy"]["InvalidLoginAttemptCount"] = 3
|
|
account_policy["Policy"]["MaxActiveSessions"] = 1
|
|
# Update the user with the newly edited policy
|
|
request_3 = requests.post(
|
|
f"{JELLYFIN_URL}/Users?userId={jellyfin_user_id}",
|
|
headers=JELLYFIN_HEADERS,
|
|
json=account_policy,
|
|
)
|
|
if request_3.status_code != 204:
|
|
return False
|
|
|
|
# Add the information to the database
|
|
db = sqlite3.connect("cordarr.db")
|
|
cursor = db.cursor()
|
|
cursor.execute(
|
|
"INSERT INTO jellyfin_accounts (user_id, jellyfin_user_id, deletion_time) VALUES (?, ?, ?)",
|
|
(user_id, jellyfin_user_id, deletion_time),
|
|
)
|
|
db.commit()
|
|
db.close()
|
|
|
|
return username, password
|
|
|
|
|
|
"""
|
|
Delete a specific Jellyfin account and return True/False
|
|
"""
|
|
|
|
|
|
def delete_jellyfin_account(jellyfin_user_id):
|
|
request = requests.delete(
|
|
f"{JELLYFIN_URL}/Users/{jellyfin_user_id}",
|
|
headers=JELLYFIN_HEADERS,
|
|
)
|
|
# If 204 - account deleted
|
|
# If 404 - account not found
|
|
# Either way, remove account from database
|
|
if request.status_code in (404, 204):
|
|
db = sqlite3.connect("cordarr.db")
|
|
cursor = db.cursor()
|
|
cursor.execute(
|
|
"DELETE FROM jellyfin_accounts WHERE jellyfin_user_id = ?",
|
|
(jellyfin_user_id,),
|
|
)
|
|
db.commit()
|
|
db.close()
|
|
return True
|
|
return False |