import os from api.backend.database.common import connect, QUERIES, insert import logging from api.backend.auth.auth_utils import get_password_hash LOG = logging.getLogger(__name__) def init_database(): cursor = connect() for query in QUERIES["init"].strip().split(";"): if query.strip(): LOG.info(f"Executing query: {query}") _ = cursor.execute(query) if os.environ.get("REGISTRATION_ENABLED", "True") == "False": default_user_email = os.environ.get("DEFAULT_USER_EMAIL") default_user_password = os.environ.get("DEFAULT_USER_PASSWORD") default_user_full_name = os.environ.get("DEFAULT_USER_FULL_NAME") if ( not default_user_email or not default_user_password or not default_user_full_name ): LOG.error( "DEFAULT_USER_EMAIL, DEFAULT_USER_PASSWORD, or DEFAULT_USER_FULL_NAME is not set!" ) exit(1) query = "INSERT INTO users (email, hashed_password, full_name) VALUES (?, ?, ?)" _ = insert( query, ( default_user_email, get_password_hash(default_user_password), default_user_full_name, ), ) cursor.close()