aswi2021jmsd-gitlab4/src/config/connection_provider.py @ dea7f12f

import sqlite3
from sqlite3 import Connection

from injector import Module, provider, singleton

from src.config.configuration import Configuration
from src.db.init_queries import DEFAULT_VALUES_SQL
from src.db.setup_database import SCHEMA_SQL
from src.utils.logger import Logger


class ConnectionProvider(Module):
    """
    Dependency injection module that provides database connection singleton
    """

    @singleton
    @provider
    def connect(self, configuration: Configuration) -> Connection:
        """
        Create an SQLite connection based on the given configuration
        :param configuration: Configuration class with application config data
        :return: connection singleton
        """

        Logger.debug(f"Creating a database connection [{configuration.connection_string}].")

        try:
            co = sqlite3.connect(database=configuration.connection_string, check_same_thread=False)
            cu = co.cursor()
            cu.executescript(SCHEMA_SQL)  # TODO change setup_database not to drop tables if they exist
            cu.executescript(DEFAULT_VALUES_SQL)
        except sqlite3.Error as e:
            Logger.error(f"Unknown error during database setting.")
            raise e

        return co