Yeet old runner script as it was useless

Clean up code and styling

onlylegs/utils/startup.py

@@ -0,0 +1,156 @@
+"""
+OnlyLegs - Setup
+Runs when the app detects that there is no user directory
+"""
+import os
+import re
+import platformdirs
+import yaml
+from werkzeug.security import generate_password_hash
+from onlylegs.extensions import db
+from onlylegs.models import Users
+
+
+APPLICATION_ROOT = platformdirs.user_config_dir("onlyLegs")
+REQUIRED_DIRS = {
+    "root": APPLICATION_ROOT,
+    "instance": os.path.join(APPLICATION_ROOT, "instance"),
+    "media": os.path.join(APPLICATION_ROOT, "media"),
+    "uploads": os.path.join(APPLICATION_ROOT, "media", "uploads"),
+    "cache": os.path.join(APPLICATION_ROOT, "media", "cache"),
+    "pfp": os.path.join(APPLICATION_ROOT, "media", "pfp"),
+}
+
+EMAIL_REGEX = re.compile(r"\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b")
+USERNAME_REGEX = re.compile(r"\b[A-Za-z0-9._%+-]+\b")
+
+
+def check_dirs():
+    """
+    Create the user directory
+    """
+
+    for directory in REQUIRED_DIRS.values():
+        if os.path.exists(directory):
+            print("User directory already exists at:", directory)
+            return
+        os.makedirs(directory)
+        print("Created directory at:", directory)
+
+
+def check_env():
+    """
+    Create the .env file with default values
+    """
+    if os.path.exists(os.path.join(APPLICATION_ROOT, ".env")):
+        print("Environment file already exists at:", APPLICATION_ROOT)
+        return
+
+    env_conf = {
+        "FLASK_SECRET": os.urandom(32).hex(),
+    }
+
+    with open(
+        os.path.join(APPLICATION_ROOT, ".env"), encoding="utf-8", mode="w+"
+    ) as file:
+        for key, value in env_conf.items():
+            file.write(key + "=" + value + "\n")
+
+    print(
+        "####################################################"
+        "# A NEW KEY WAS GENERATED FOR YOU! PLEASE NOTE     #"
+        "# DOWN THE FLASK_SECRET KEY LOCATED IN YOUR        #"
+        "# ~/.config/onlylegs/.env FOLDER! LOOSING THIS KEY #"
+        "# WILL RESULT IN YOU BEING UNABLE TO LOG IN!       #"
+        "####################################################",
+        sep="\n",
+    )
+
+
+def check_conf():
+    """
+    Create the YAML config file with default values
+    """
+    if os.path.exists(os.path.join(APPLICATION_ROOT, "conf.yml")):
+        print("Config file already exists at:", APPLICATION_ROOT)
+        return
+
+    can_continue = False
+    username = "admin"
+    name = "Admin"
+    email = "admin@example.com"
+
+    print("No config file found, please enter the following information:")
+    while can_continue:
+        username = input("Admin username: ")
+        name = input("Admin name: ")
+        email = input("Admin email: ")
+
+        if not username or not USERNAME_REGEX.match(username):
+            print("Username is invalid!")
+        if not name:
+            print("Name is invalid!")
+        if not email or not EMAIL_REGEX.match(email):
+            print("Email is invalid!")
+
+        # Check if user is happy with the values
+        is_correct = input("Is this correct? (Y/n): ").lower()
+        if is_correct == "y" or is_correct == "":
+            can_continue = True
+
+    yaml_conf = {
+        "admin": {
+            "name": name,
+            "username": username,
+            "email": email,
+        },
+        "upload": {
+            "allowed-extensions": {
+                "jpg": "jpeg",
+                "jpeg": "jpeg",
+                "png": "png",
+                "webp": "webp",
+            },
+            "max-size": 69,
+            "max-load": 50,
+            "rename": "GWA_{{username}}_{{time}}",
+        },
+        "website": {
+            "name": "OnlyLegs",
+            "motto": "A gallery built for fast and simple image management!",
+            "language": "en",
+        },
+    }
+
+    with open(
+        os.path.join(APPLICATION_ROOT, "conf.yml"), encoding="utf-8", mode="w+"
+    ) as file:
+        yaml.dump(yaml_conf, file, default_flow_style=False)
+
+    print(
+        "####################################################"
+        "# A NEW CONFIG HAS BEEN GENERATED AT:              #"
+        "# ~/.config/onlylegs/conf.yml                      #"
+        "####################################################",
+        sep="\n",
+    )
+
+
+def make_admin_user(app):
+    username = app.config["ADMIN_CONF"]["username"]
+    email = app.config["ADMIN_CONF"]["email"]
+    password = generate_password_hash("changeme!", method="scrypt")
+
+    with app.app_context():
+        db.create_all()
+        db.session.add(Users(username=username, email=email, password=password))
+        db.session.commit()
+
+        print(
+            "####################################################"
+            "# DEFAULT ADMIN USER GENERATED WITH GIVEN USERNAME #"
+            '# THE DEFAULT PASSWORD "changeme!" HAS BEEN USED,  #'
+            "# PLEASE RESET IT IN THE SETTINGS!                 #"
+            "####################################################",
+            sep="\n",
+        )