Files
FastapiAdmin/backend/app/api/v1/module_system/log/schema.py
T
zhangtao 4e2b668d7b chore: 批量优化项目代码,修复多处细节问题
本次提交包含多项优化和修复:
1. 修复CRUD初始化参数传递、搜索参数处理逻辑
2. 更新环境配置中的大模型相关参数
3. 重构部分服务方法命名,统一代码风格
4. 新增多个枚举类型,补充模型关联关系和加载选项
5. 优化查询参数类实现,完善字段校验逻辑
6. 调整Pydantic模型字段注释和类型定义
7. 简化并移除冗余的CRUD方法实现
8. 新增超级管理员权限装饰器
9. 修复邮件日志模型的租户关联和字段定义
2026-06-20 23:37:58 +08:00

126 lines
4.8 KiB
Python

from dataclasses import dataclass
from fastapi import Query
from pydantic import BaseModel, ConfigDict, Field, field_validator
from app.common.enums import QueueEnum
from app.core.base_params import BaseQueryParam, TenantByQueryParam, UserByQueryParam
from app.core.base_schema import BaseSchema, TenantBySchema, UserBySchema
ALLOWED_REQUEST_METHODS = ["GET", "POST", "PUT", "DELETE", "PATCH", "OPTIONS", "HEAD"]
class LoginLogCreateSchema(BaseModel):
"""新增登录日志"""
username: str = Field(..., min_length=1, max_length=64, description="用户名")
status: int = Field(default=1, ge=1, le=2, description="登录状态(1成功 2失败)")
login_ip: str | None = Field(default=None, max_length=50, description="登录IP地址")
login_location: str | None = Field(default=None, max_length=255, description="登录位置")
request_os: str | None = Field(default=None, max_length=64, description="操作系统")
request_browser: str | None = Field(default=None, max_length=64, description="浏览器")
msg: str | None = Field(default=None, max_length=255, description="提示消息")
@field_validator("username")
@classmethod
def validate_username(cls, v: str) -> str:
v = v.strip()
if not v:
raise ValueError("用户名不能为空")
if len(v) > 64:
raise ValueError("用户名长度不能超过64个字符")
return v
@field_validator("status")
@classmethod
def validate_status(cls, v: int) -> int:
if v not in [1, 2]:
raise ValueError("登录状态只能为1(成功)或2(失败)")
return v
class LoginLogOutSchema(LoginLogCreateSchema, BaseSchema, UserBySchema, TenantBySchema):
"""登录日志响应"""
model_config = ConfigDict(from_attributes=True)
class LoginLogDetailOutSchema(LoginLogOutSchema):
"""登录日志详情响应"""
@dataclass
class LoginLogQueryParam(BaseQueryParam, UserByQueryParam, TenantByQueryParam):
"""登录日志查询参数"""
def __init__(
self,
username: str | None = Query(None, max_length=64, description="用户名"),
*args,
**kwargs,
) -> None:
super().__init__(*args, **kwargs)
if username:
self.username = (QueueEnum.like.value, username)
@dataclass
class OperationLogQueryParam(BaseQueryParam, UserByQueryParam, TenantByQueryParam):
"""操作日志查询参数"""
def __init__(
self,
request_path: str | None = Query(None, description="请求路径"),
request_method: str | None = Query(None, description="请求方式"),
username: str | None = Query(None, description="用户名"),
*args,
**kwargs,
) -> None:
super().__init__(*args, **kwargs)
if request_path:
self.request_path = (QueueEnum.like.value, request_path)
if request_method:
self.request_method = (QueueEnum.eq.value, request_method)
if username:
self.username = (QueueEnum.like.value, username)
class OperationLogOutSchema(BaseSchema, UserBySchema, TenantBySchema):
"""操作日志响应模型"""
model_config = ConfigDict(from_attributes=True)
status: int | None = Field(default=None, description="状态(0:启动 1:停用)")
description: str | None = Field(default=None, description="描述")
request_path: str = Field(..., description="请求路径")
request_method: str = Field(..., description="请求方式")
response_code: int = Field(..., description="响应状态码")
process_time: str | None = Field(default=None, description="处理时间")
class OperationLogDetailOutSchema(OperationLogOutSchema):
"""操作日志详情响应模型"""
request_payload: str | None = Field(default=None, description="请求体")
response_json: str | None = Field(default=None, description="响应体")
class OperationLogCreateSchema(BaseModel):
request_path: str = Field(..., min_length=1, max_length=255, description="请求路径")
request_method: str = Field(..., description="请求方式")
request_payload: str | None = Field(None, description="请求体")
response_code: int = Field(200, ge=100, le=599, description="响应状态码")
response_json: str | None = Field(None, description="响应体")
process_time: str | None = Field(None, max_length=20, description="处理时间")
created_id: int | None = Field(None, description="创建人ID")
updated_id: int | None = Field(None, description="更新人ID")
description: str | None = Field(None, description="备注")
@field_validator("request_method")
@classmethod
def validate_request_method(cls, value: str) -> str:
upper_value = value.upper()
if upper_value not in ALLOWED_REQUEST_METHODS:
raise ValueError(f"请求方式必须是: {', '.join(ALLOWED_REQUEST_METHODS)}")
return upper_value