aswi2021jmsd-gitlab4/src/db/setup_database.py @ 9a6c9613

import sqlite3

from src.constants import DATABASE_FILE_LOCATION
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:
        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()