151 lines
5.0 KiB
Python
151 lines
5.0 KiB
Python
import redis
|
|
from app.config import get_settings
|
|
from fastapi import Depends
|
|
from sqlalchemy.orm import Session
|
|
|
|
from packages.adapters.redis import NoopSessionStore, SessionStore, get_session_store
|
|
from packages.adapters.smtp import EmailConfig, NoopEmailService, get_email_service
|
|
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.generated_video_repository import (
|
|
SQLAlchemyGeneratedVideoRepository,
|
|
)
|
|
from packages.adapters.sqlalchemy_impl.generation_task_repository import (
|
|
SQLAlchemyGenerationTaskRepository,
|
|
)
|
|
from packages.adapters.sqlalchemy_impl.ingest_job_repository import (
|
|
SQLAlchemyIngestJobRepository,
|
|
)
|
|
from packages.adapters.sqlalchemy_impl.project_repository import (
|
|
SQLAlchemyProjectRepository,
|
|
)
|
|
from packages.adapters.sqlalchemy_impl.project_title_repository import (
|
|
SQLAlchemyProjectTitleRepository,
|
|
)
|
|
from packages.adapters.sqlalchemy_impl.session import build_session_factory
|
|
from packages.adapters.sqlalchemy_impl.user_repository import SQLAlchemyUserRepository
|
|
from packages.adapters.sqlalchemy_impl.workspace_invitation_repository import (
|
|
SQLAlchemyWorkspaceInvitationRepository,
|
|
)
|
|
from packages.adapters.sqlalchemy_impl.workspace_member_repository import (
|
|
SQLAlchemyWorkspaceMemberRepository,
|
|
)
|
|
from packages.adapters.sqlalchemy_impl.workspace_repository import (
|
|
SQLAlchemyWorkspaceRepository,
|
|
)
|
|
from packages.ports.user_repository import UserRepository
|
|
from packages.ports.workspace_invitation_repository import WorkspaceInvitationRepository
|
|
from packages.ports.workspace_member_repository import WorkspaceMemberRepository
|
|
from packages.ports.workspace_repository import WorkspaceRepository
|
|
|
|
_engine, _SessionLocal = build_session_factory(get_settings().DATABASE_URL)
|
|
|
|
|
|
def get_db_session() -> Session:
|
|
"""Get database session with automatic cleanup."""
|
|
session: Session = _SessionLocal()
|
|
try:
|
|
yield session
|
|
finally:
|
|
session.close()
|
|
|
|
|
|
def get_asset_repository(
|
|
session: Session = Depends(get_db_session),
|
|
) -> SQLAlchemyAssetRepository:
|
|
return SQLAlchemyAssetRepository(session)
|
|
|
|
|
|
def get_asset_library_repository(
|
|
session: Session = Depends(get_db_session),
|
|
) -> SQLAlchemyAssetLibraryRepository:
|
|
return SQLAlchemyAssetLibraryRepository(session)
|
|
|
|
|
|
def get_ingest_job_repository(
|
|
session: Session = Depends(get_db_session),
|
|
) -> SQLAlchemyIngestJobRepository:
|
|
return SQLAlchemyIngestJobRepository(session)
|
|
|
|
|
|
def get_classification_job_repository(
|
|
session: Session = Depends(get_db_session),
|
|
) -> SQLAlchemyClassificationJobRepository:
|
|
return SQLAlchemyClassificationJobRepository(session)
|
|
|
|
|
|
def get_generation_task_repository(
|
|
session: Session = Depends(get_db_session),
|
|
) -> SQLAlchemyGenerationTaskRepository:
|
|
return SQLAlchemyGenerationTaskRepository(session)
|
|
|
|
|
|
def get_generated_video_repository(
|
|
session: Session = Depends(get_db_session),
|
|
) -> SQLAlchemyGeneratedVideoRepository:
|
|
return SQLAlchemyGeneratedVideoRepository(session)
|
|
|
|
|
|
def get_project_repository(
|
|
session: Session = Depends(get_db_session),
|
|
) -> SQLAlchemyProjectRepository:
|
|
return SQLAlchemyProjectRepository(session)
|
|
|
|
|
|
def get_project_title_repository(
|
|
session: Session = Depends(get_db_session),
|
|
) -> SQLAlchemyProjectTitleRepository:
|
|
return SQLAlchemyProjectTitleRepository(session)
|
|
|
|
|
|
def get_user_repository(
|
|
session: Session = Depends(get_db_session),
|
|
) -> UserRepository:
|
|
return SQLAlchemyUserRepository(session)
|
|
|
|
|
|
def get_workspace_repository(
|
|
session: Session = Depends(get_db_session),
|
|
) -> WorkspaceRepository:
|
|
return SQLAlchemyWorkspaceRepository(session)
|
|
|
|
|
|
def get_workspace_member_repository(
|
|
session: Session = Depends(get_db_session),
|
|
) -> WorkspaceMemberRepository:
|
|
return SQLAlchemyWorkspaceMemberRepository(session)
|
|
|
|
|
|
def get_workspace_invitation_repository(
|
|
session: Session = Depends(get_db_session),
|
|
) -> WorkspaceInvitationRepository:
|
|
return SQLAlchemyWorkspaceInvitationRepository(session)
|
|
|
|
|
|
def get_auth_session_store() -> SessionStore:
|
|
if not get_settings().ENABLE_REDIS_SESSIONS:
|
|
return NoopSessionStore()
|
|
return SessionStore(redis_client=redis.from_url(get_settings().REDIS_URL, decode_responses=True))
|
|
|
|
|
|
def get_auth_email_service() -> EmailConfig:
|
|
if not get_settings().ENABLE_EMAIL_DELIVERY:
|
|
return NoopEmailService()
|
|
return get_email_service(
|
|
EmailConfig(
|
|
smtp_host=get_settings().SMTP_HOST,
|
|
smtp_port=get_settings().SMTP_PORT,
|
|
smtp_user=get_settings().SMTP_USER,
|
|
smtp_password=get_settings().SMTP_PASSWORD,
|
|
from_email=get_settings().SMTP_FROM_EMAIL,
|
|
from_name=get_settings().SMTP_FROM_NAME,
|
|
use_tls=get_settings().SMTP_USE_TLS,
|
|
),
|
|
enabled=True,
|
|
)
|