From c913e7cd8958d1aa3c21ed979e3bcbcabeafc5dd Mon Sep 17 00:00:00 2001 From: xiaoxia Date: Fri, 26 Jun 2026 18:37:11 +0800 Subject: [PATCH] =?UTF-8?q?P2-4:=20dependencies.py=20=E6=B7=BB=E5=8A=A0?= =?UTF-8?q?=E8=BF=94=E5=9B=9E=E7=B1=BB=E5=9E=8B=E6=B3=A8=E8=A7=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/api/app/dependencies.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/apps/api/app/dependencies.py b/apps/api/app/dependencies.py index e0c0d57ac..a1d45a039 100644 --- a/apps/api/app/dependencies.py +++ b/apps/api/app/dependencies.py @@ -1,3 +1,5 @@ +from typing import Generator + import redis from app.config import settings from fastapi import Depends @@ -40,7 +42,8 @@ from packages.ports.workspace_repository import WorkspaceRepository _engine, _SessionLocal = build_session_factory(settings.DATABASE_URL) -def get_db_session(): +def get_db_session() -> Generator[Session, None, None]: + """获取数据库会话""" session: Session = _SessionLocal() try: yield session @@ -120,13 +123,13 @@ def get_workspace_invitation_repository( return SQLAlchemyWorkspaceInvitationRepository(session) -def get_auth_session_store(): - if not settings.ENABLE_REDIS_SESSIONS: +def get_auth_session_store() -> NoopSessionStore | SessionStore: + if not settings.ENABLE_REDIS_SESSION: return NoopSessionStore() return SessionStore(redis_client=redis.from_url(settings.REDIS_URL, decode_responses=True)) -def get_auth_email_service(): +def get_auth_email_service() -> NoopEmailService: if not settings.ENABLE_EMAIL_DELIVERY: return NoopEmailService() return get_email_service(