from sqlite_framework.session.session import SqliteSession

from bot.storage.data_source.data_source import StorageDataSource


class SqliteStorageDataSource(StorageDataSource):
    def __init__(self, database_filename: str, debug: bool):
        super().__init__()
        self.session = SqliteSession(database_filename, debug)

    def init(self):
        self.session.init()

    def context_manager(self):
        return self.session.context_manager()

from sqlite_framework.log.logger import SqliteLogger
from sqlite_framework.session.session import SqliteSession

from bot.storage.data_source.data_source import StorageDataSource


class SqliteStorageDataSource(StorageDataSource):
    def __init__(self, database_filename: str, debug: bool, logger: SqliteLogger):
        super().__init__()
        self.session = SqliteSession(database_filename, debug)
        self.logger = logger

    def init(self):
        self.session.init()

    def context_manager(self):
        return self.session.context_manager()

Add logger to SqliteStorageDataSource