crafty-4/app/classes/shared/main_models.py

import logging

from app.classes.models.users import Users, users_helper
from app.classes.shared.helpers import helper
from app.classes.shared.console import console

# To disable warning about unused import ; Users is imported from here in other places
# pylint: disable=self-assigning-variable
Users = Users

try:
    # pylint: disable=unused-import
    from peewee import SqliteDatabase, fn
    from playhouse.shortcuts import model_to_dict

except ModuleNotFoundError as err:
    helper.auto_installer_fix(err)

logger = logging.getLogger(__name__)
peewee_logger = logging.getLogger("peewee")
peewee_logger.setLevel(logging.INFO)
database = SqliteDatabase(
    helper.db_path, pragmas={"journal_mode": "wal", "cache_size": -1024 * 10}
)


class db_builder:
    @staticmethod
    def default_settings():
        logger.info("Fresh Install Detected - Creating Default Settings")
        console.info("Fresh Install Detected - Creating Default Settings")
        default_data = helper.find_default_password()

        username = default_data.get("username", "admin")
        password = default_data.get("password", "crafty")

        users_helper.add_user(
            username=username,
            password=password,
            email="default@example.com",
            superuser=True,
        )

    @staticmethod
    def is_fresh_install():
        try:
            user = users_helper.get_by_id(1)
            if user:
                return False
        except:
            return True


class db_shortcuts:

    # **********************************************************************************
    #                                  Generic Databse Methods
    # **********************************************************************************
    @staticmethod
    def return_rows(query):
        rows = []

        try:
            if query.count() > 0:
                for s in query:
                    rows.append(model_to_dict(s))
        except Exception as e:
            logger.warning(f"Database Error: {e}")

        return rows

    @staticmethod
    def return_db_rows(model):
        data = [model_to_dict(row) for row in model]
        return data


# **********************************************************************************
#                                  Static Accessors
# **********************************************************************************
installer = db_builder()
db_helper = db_shortcuts()