Parker
237aec245e
Temporary jellyfin accounts can now be made through messaging. Commands were moved out and into their own files and functions for organization.
73 lines
3.4 KiB
Python
73 lines
3.4 KiB
Python
import datetime
|
|
import requests
|
|
import random
|
|
import string
|
|
import sqlite3
|
|
|
|
import initialize_variables
|
|
from create_message import create_message
|
|
|
|
|
|
def movie_show_response_newaccount(from_number, message):
|
|
if from_number not in initialize_variables.temp_new_account_requests.keys():
|
|
create_message(from_number, "There is no current request that you can decide on. It might be that your /newaccount command timed out due since you took too long to response. Please try again. If this issue persists, please contact Parker.")
|
|
return
|
|
|
|
# If its been 5 minutes since prompt was sent, alert user of timed out request
|
|
if (datetime.datetime.now() - initialize_variables.temp_new_account_requests[from_number]).total_seconds() / 60 > 5:
|
|
del initialize_variables.temp_new_account_requests[from_number]
|
|
create_message(from_number, "You waited too long and therefore your request has timed out.\n\nPlease try again by re-running the /newaccount command. If this issue persists, please contact Parker.")
|
|
return
|
|
|
|
if message.strip().lower() == "show":
|
|
active_time = 24
|
|
|
|
elif message.strip().lower() == "movie":
|
|
active_time = 4
|
|
|
|
else:
|
|
create_message(from_number, "You did not enter a valid response. Please re-send the /newaccount command and try again. If you believe this is an error, please contact Parker.")
|
|
return
|
|
|
|
# Otherwise, all checks have been completed
|
|
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=active_time)
|
|
# Create new Jellyfin account
|
|
request_1 = requests.post(f'{initialize_variables.jellyfin_url}/Users/New', headers=initialize_variables.jellyfin_headers, json={'Name': username, 'Password': password})
|
|
if request_1.status_code != 200:
|
|
create_message(from_number, "Error creating Jellyfin account. Please try again. If the error persists, contact Parker.")
|
|
return
|
|
|
|
user_id = request_1.json()['Id']
|
|
# Get account policy and make edits
|
|
request_2 = requests.get(f'{initialize_variables.jellyfin_url}/Users/{user_id}', headers=initialize_variables.jellyfin_headers)
|
|
if request_2.status_code != 200:
|
|
create_message(from_number, "Error creating Jellyfin account. Please try again. If the error persists, contact Parker.")
|
|
return
|
|
|
|
policy = request_2.json()['Policy']
|
|
policy['SyncPlayAccess'] = 'JoinGroups'
|
|
policy['EnableContentDownloading'] = False
|
|
policy['InvalidLoginAttemptCount'] = 3
|
|
policy['MaxActiveSessions'] = 1
|
|
# Update user with new policy
|
|
request_3 = requests.post(f'{initialize_variables.jellyfin_url}/Users/{user_id}/Policy', headers=initialize_variables.jellyfin_headers, json=policy)
|
|
if request_3.status_code != 204:
|
|
create_message(from_number, "Error creating Jellyfin account. Please try again. If the error persists, contact Parker.")
|
|
return
|
|
|
|
# Add information to the database
|
|
db = sqlite3.connect(initialize_variables.db_path)
|
|
cursor = db.cursor()
|
|
cursor.execute('''
|
|
INSERT INTO jellyfin_accounts (user_id, deletion_time)
|
|
VALUES(?, ?)
|
|
''', (user_id, deletion_time))
|
|
db.commit()
|
|
db.close()
|
|
|
|
create_message(from_number, f"Username: {username}\nPassword: {password}\n\nYour account will expire in {active_time} hours.")
|
|
return
|