Create CordArr

1 files changed, 164 insertions, 0 deletions

diff --git a/code/validate_config.py b/code/validate_config.py
new file mode 100644
index 0000000..191cb03
--- /dev/null
+++ b/code/validate_config.py
@@ -0,0 +1,164 @@
+import configparser
+import sqlite3
+import requests
+from global_variables import LOG, YES_VALUES, NO_VALUES
+
+"""
+Validate all of the options passed into the config.ini file
+"""
+
+
+def validate_config(file_contents):
+    config = configparser.ConfigParser()
+    config.read_string(file_contents)
+
+    errors = 0
+
+    try:
+        # Validate BOT_TOKEN
+        if not config["REQUIRED"]["BOT_TOKEN"]:
+            LOG.error("BOT_TOKEN has not been set.")
+            errors += 1
+
+        # Validate RADARR_HOST_URL
+        if not config["REQUIRED"]["RADARR_HOST_URL"]:
+            LOG.error("RADARR_HOST_URL has not been set.")
+            errors += 1
+
+        # Validate RADARR_API_KEY
+        if not config["REQUIRED"]["RADARR_API_KEY"]:
+            LOG.error("RADARR_API_KEY has not been set.")
+            errors += 1
+
+        radarr_headers = {
+            "Content-Type": "application/json",
+            "X-Api-Key": config["REQUIRED"]["RADARR_API_KEY"],
+        }
+
+        # Make sure connection to Radarr API can be established
+        try:
+            requests.get(config["REQUIRED"]["RADARR_HOST_URL"], headers=radarr_headers)
+        except requests.exceptions.ConnectionError:
+            LOG.error("Could not connect to Radarr API. Please check your RADARR_HOST_URL and RADARR_API_KEY")
+            errors += 1
+
+        # Validate ROOT_FOLDER_PATH
+        if not config["REQUIRED"]["ROOT_FOLDER_PATH"]:
+            LOG.error("ROOT_FOLDER_PATH has not been set.")
+            errors += 1
+
+        # Validate QUALITY_PROFILE_ID
+        data = requests.get(
+            f'{config["REQUIRED"]["RADARR_HOST_URL"]}/api/v3/qualityprofile',
+            headers=radarr_headers,
+        ).json()
+        all_ids = []
+        for entry in data:
+            all_ids.append(str(entry["id"]))
+
+        if (
+            not config["REQUIRED"]["QUALITY_PROFILE_ID"]
+            or config["REQUIRED"]["QUALITY_PROFILE_ID"] not in all_ids
+        ):
+            config["AVAILABLE_QUALITY_IDS"] = {}
+            for entry in data:
+                config["AVAILABLE_QUALITY_IDS"][str(entry["id"])] = entry["name"]
+
+            LOG.error("Empty or invalid QUALITY_PROFILE_ID passed. Pass one of the valid IDs which are now listed within the config.ini file.")
+            errors += 1
+
+        # Validate ENABLE_JELLYFIN_TEMP_ACCOUNTS
+        if not config["REQUIRED"]["ENABLE_JELLYFIN_TEMP_ACCOUNTS"]:
+            LOG.error("ENABLE_JELLYFIN_TEMP_ACCOUNTS has not been set.")
+            errors += 1
+
+        else:
+            # Validate the value of ENABLE_JELLYFIN_TEMP_ACCOUNTS
+            if (config["REQUIRED"]["ENABLE_JELLYFIN_TEMP_ACCOUNTS"].lower() not in YES_VALUES + NO_VALUES):
+                LOG.error("Invalid value passed to ENABLE_JELLYFIN_TEMP_ACCOUNTS. Pass a true/false value.")
+                errors += 1
+
+            if (config["REQUIRED"]["ENABLE_JELLYFIN_TEMP_ACCOUNTS"].lower() in YES_VALUES):
+                # Validate JELLYFIN_URL
+                if not config["JELLYFIN_ACCOUNTS"]["JELLYFIN_URL"]:
+                    LOG.error("Empty URL passed to JELLYFIN_URL. Pass a valid URL (e.g. http://localhost:8096)")
+                    errors += 1
+                # Validate JELLYFIN_API_KEY
+                if not config["JELLYFIN_ACCOUNTS"]["JELLYFIN_API_KEY"]:
+                    LOG.error("Empty JELLYFIN_API_KEY passed. Create a Jellyfin API key in your Jellyfin dashboard and pass it here.")
+                    errors += 1
+                # Validate ACCOUNT_TIME
+                if not config["JELLYFIN_ACCOUNTS"]["ACCOUNT_TIME"]:
+                    LOG.error("Empty ACCOUNT_TIME passed. Pass a valid time in the format of HH:MM:SS (e.g. 00:30:00)")
+                    errors += 1
+                try:
+                    time = int(config["JELLYFIN_ACCOUNTS"]["ACCOUNT_TIME"])
+                except ValueError:
+                    LOG.error("Invalid value passed to ACCOUNT_TIME. Pass a valid integer value (e.g. 24)")
+                    errors += 1
+
+                # Make sure connection to Jellyfin API can be established
+                jellyfin_headers = {
+                    "Content-Type": "application/json",
+                    "Authorization": f"MediaBrowser Client=\"other\", device=\"CordArr\", DeviceId=\"cordarr-device-id\", Version=\"0.0.0\", Token=\"{config['JELLYFIN_ACCOUNTS']['JELLYFIN_API_KEY']}\"",
+                }
+
+                response = requests.get(
+                    f"{config['JELLYFIN_ACCOUNTS']['JELLYFIN_URL']}/Users",
+                    headers=jellyfin_headers,
+                )
+                if response.status_code != 200:
+                    LOG.error("Could not connect to Jellyfin API. Please check your JELLYFIN_URL and JELLYFIN_API_KEY")
+                    errors += 1
+
+        if errors > 0:
+            LOG.info(f"Found {errors} error(s) in the configuration file. Please fix them before restarting the application.")
+            exit()
+
+    except KeyError:
+        LOG.critical("You are missing at least one of the configuration options in your config.ini file. In order to regenerate all options, delete the config.ini file and restart the application.")
+        exit()
+
+
+"""
+This method is called before starting the application - to make and validate the configuration
+"""
+
+
+def create_config():
+    # While here, we can begin by making the database
+    db = sqlite3.connect("cordarr.db")
+    cursor = db.cursor()
+    cursor.execute("CREATE TABLE IF NOT EXISTS movies (user_id, movie_id, movie_title)")
+    cursor.execute("CREATE TABLE IF NOT EXISTS jellyfin_accounts (user_id, jellyfin_user_id, deletion_time, PRIMARY KEY (user_id))")
+    db.commit()
+    db.close()
+
+    # Attempt to open and validate the configuration file
+    try:
+        with open("config.ini", "r") as config:
+            file_contents = config.read()
+            validate_config(file_contents)
+
+    except FileNotFoundError:
+        try:
+            with open("/data/config.ini", "r") as config:
+                file_contents = config.read()
+                validate_config(file_contents)
+
+        except FileNotFoundError:
+            # Create the config.ini file
+            config = configparser.ConfigParser()
+            config["REQUIRED"] = {
+                "BOT_TOKEN": "",
+                "RADARR_HOST_URL": "http://",
+                "RADARR_API_KEY": "",
+                "ROOT_FOLDER_PATH": "",
+                "QUALITY_PROFILE_ID": "",
+                "ENABLE_JELLYFIN_TEMP_ACCOUNTS": "",
+            }
+
+            config["JELLYFIN_ACCOUNTS"] = {
+                "JELLYFIN_URL": "",
+                "JELLYFIN_API_KEY": ""
+            }