Xiaoxia AI
37 lines
1.5 KiB
Python
37 lines
1.5 KiB
Python
from sqlalchemy.orm import Session
|
|
|
|
from packages.adapters.sqlalchemy_impl.asset_library_repository import SQLAlchemyAssetLibraryRepository
|
|
from packages.adapters.sqlalchemy_impl.asset_repository import SQLAlchemyAssetRepository
|
|
from packages.adapters.sqlalchemy_impl.classification_job_repository import SQLAlchemyClassificationJobRepository
|
|
from packages.adapters.sqlalchemy_impl.ingest_job_repository import SQLAlchemyIngestJobRepository
|
|
from packages.adapters.sqlalchemy_impl.project_repository import SQLAlchemyProjectRepository
|
|
from packages.adapters.sqlalchemy_impl.session import SessionLocal
|
|
|
|
|
|
def get_db_session():
|
|
session: Session = SessionLocal()
|
|
try:
|
|
yield session
|
|
finally:
|
|
session.close()
|
|
|
|
|
|
def get_asset_repository(session: Session = next(get_db_session())) -> SQLAlchemyAssetRepository:
|
|
return SQLAlchemyAssetRepository(session)
|
|
|
|
|
|
def get_asset_library_repository(session: Session = next(get_db_session())) -> SQLAlchemyAssetLibraryRepository:
|
|
return SQLAlchemyAssetLibraryRepository(session)
|
|
|
|
|
|
def get_ingest_job_repository(session: Session = next(get_db_session())) -> SQLAlchemyIngestJobRepository:
|
|
return SQLAlchemyIngestJobRepository(session)
|
|
|
|
|
|
def get_classification_job_repository(session: Session = next(get_db_session())) -> SQLAlchemyClassificationJobRepository:
|
|
return SQLAlchemyClassificationJobRepository(session)
|
|
|
|
|
|
def get_project_repository(session: Session = next(get_db_session())) -> SQLAlchemyProjectRepository:
|
|
return SQLAlchemyProjectRepository(session)
|