aswi2021jmsd-gitlab/src/db/setup_database.py @ 4ec8e4ae

import os
import sqlite3
from pathlib import Path

from src.constants import DATABASE_FILE_LOCATION, DB_DIR
from src.db.init_queries import SCHEMA_SQL, DEFAULT_VALUES_SQL
from src.utils.logger import Logger


def setup_database():
    Logger.debug(f"Database is setting.")

    try:
        # TODO improve creating DB dir
        if not os.path.exists(DB_DIR):
            os.makedirs("../../" + DB_DIR)

        # TODO improve creating DB file (now via FileAnchor)
        co = sqlite3.connect(DATABASE_FILE_LOCATION.shortest_relative_path())
        cu = co.cursor()
        cu.executescript(SCHEMA_SQL)
        cu.executescript(DEFAULT_VALUES_SQL)
    except sqlite3.Error as e:
        Logger.error(f"Unknown error during database setting.")
        raise e

    return co


if __name__ == '__main__':
    setup_database()