P2-4: dependencies.py 添加返回类型注解

This commit is contained in:
2026-06-26 18:37:11 +08:00
parent a85790f24a
commit c913e7cd89
+7 -4
View File
@@ -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(