P2-4: dependencies.py 添加返回类型注解
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user