aswi2021jmsd-gitlab/src/config/connection_provider.py @ 2510f01a

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


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
        """
        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)
        return co