1 files changed, 72 insertions, 0 deletions
diff --git a/app/commands/movie_show_response_newaccount.py b/app/commands/movie_show_response_newaccount.py
new file mode 100644
index 0000000..6922560
--- /dev/null
+++ b/app/commands/movie_show_response_newaccount.py
@@ -0,0 +1,72 @@
+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