aswi2021jmsd-gitlab/src/config/connection_provider.py @ 3195e946

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):

    @singleton
    @provider
    def connect(self, configuration: Configuration) -> Connection:
        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