tofu-api/tofu_api/dependencies.py

25 lines
578 B
Python

from sqlalchemy.orm import Session
from tofu_api.common.database import SQLAlchemy
class Dependencies:
"""
Container for dependency injection.
"""
_dependency_cache: dict
def __init__(self):
self._dependency_cache = {}
# Database dependencies
def get_sqlalchemy(self) -> SQLAlchemy:
if SQLAlchemy not in self._dependency_cache:
self._dependency_cache[SQLAlchemy] = SQLAlchemy()
return self._dependency_cache[SQLAlchemy]
def get_db_session(self) -> Session:
return self.get_sqlalchemy().session