zzx/video
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化代码风格

Browse Source
This commit is contained in:
ZZX9599
2025-09-08 17:34:23 +08:00
parent 9b3d20511a
commit 8ceb92c572
20 changed files with 223 additions and 192 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
middle/auth_middleware.py
View File

@ -9,8 +9,6 @@ from passlib.context import CryptContext
from ds.config import JWT_CONFIG
from ds.db import db
# 移除这里的 from service.user_service import UserResponse 导入
# ------------------------------
# 密码加密配置
# ------------------------------
@ -23,7 +21,7 @@ SECRET_KEY = JWT_CONFIG["secret_key"]
ALGORITHM = JWT_CONFIG["algorithm"]
ACCESS_TOKEN_EXPIRE_MINUTES = int(JWT_CONFIG["access_token_expire_minutes"])
# OAuth2 依赖（从请求头获取 Token、格式：Bearer <token>）
# OAuth2 依赖（从请求头获取 Token、格式: Bearer <token>）
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/users/login")
@ -63,7 +61,7 @@ def create_access_token(data: dict, expires_delta: Optional[timedelta] = None) -
# ------------------------------
def get_current_user(token: str = Depends(oauth2_scheme)): # 移除返回类型注解
"""从 Token 中解析用户信息、验证通过后返回当前用户"""
# 延迟导入，打破循环依赖
# 延迟导入、打破循环依赖
from schema.user_schema import UserResponse # 在这里导入
# 认证失败异常
@ -101,4 +99,4 @@ def get_current_user(token: str = Depends(oauth2_scheme)): # 移除返回类型
except Exception as e:
raise credentials_exception from e
finally:
db.close_connection(conn, cursor)
db.close_connection(conn, cursor)

8
middle/error_handler.py
View File

@ -8,7 +8,7 @@ from schema.response_schema import APIResponse
async def global_exception_handler(request: Request, exc: Exception):
"""全局异常处理器：所有未捕获的异常都会在这里统一处理"""
"""全局异常处理器: 所有未捕获的异常都会在这里统一处理"""
# 1. 请求参数验证错误（Pydantic 校验失败）
if isinstance(exc, RequestValidationError):
error_details = []
@ -18,7 +18,7 @@ async def global_exception_handler(request: Request, exc: Exception):
status_code=status.HTTP_400_BAD_REQUEST,
content=APIResponse(
code=400,
message=f"请求参数错误：{'; '.join(error_details)}",
message=f"请求参数错误: {'; '.join(error_details)}",
data=None
).model_dump()
)
@ -52,7 +52,7 @@ async def global_exception_handler(request: Request, exc: Exception):
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
content=APIResponse(
code=500,
message=f"数据库错误：{str(exc)}",
message=f"数据库错误: {str(exc)}",
data=None
).model_dump()
)
@ -62,7 +62,7 @@ async def global_exception_handler(request: Request, exc: Exception):
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
content=APIResponse(
code=500,
message=f"服务器内部错误：{str(exc)}",
message=f"服务器内部错误: {str(exc)}",
data=None
).model_dump()
)