790c068eb6
- 修改 README 中的模型配置路径，确保指向正确的配置文件位置 - 新增 MinIO 存储模块文档，提供基本使用和高级使用示例 - 更新内容审查机制和品牌信息配置的相关路径 - 移除旧的用户模型和线程模型，重构数据库模型管理 - 优化代码结构，确保数据库管理器和模型的导入路径一致
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
"""通用工具函数"""
|
|
|
|
import logging
|
|
|
|
from fastapi import Request
|
|
from sqlalchemy.orm import Session
|
|
|
|
from src.storage.db.models import OperationLog, User
|
|
|
|
|
|
def setup_logging():
|
|
"""配置应用程序日志格式"""
|
|
# 配置日志格式
|
|
logging.basicConfig(
|
|
level=logging.INFO, format="%(asctime)s %(levelname)s: %(message)s", datefmt="%Y-%m-%d %H:%M:%S", force=True
|
|
)
|
|
|
|
# 确保uvicorn的日志也使用相同格式
|
|
uvicorn_logger = logging.getLogger("uvicorn")
|
|
uvicorn_access_logger = logging.getLogger("uvicorn.access")
|
|
|
|
# 创建格式化器
|
|
formatter = logging.Formatter(fmt="%(asctime)s %(levelname)s: %(message)s", datefmt="%m-%d %H:%M:%S")
|
|
|
|
# 为所有处理器设置格式化器
|
|
for handler in uvicorn_logger.handlers:
|
|
handler.setFormatter(formatter)
|
|
for handler in uvicorn_access_logger.handlers:
|
|
handler.setFormatter(formatter)
|
|
|
|
|
|
def log_operation(db: Session, user_id: int, operation: str, details: str = None, request: Request = None):
|
|
"""记录用户操作日志"""
|
|
ip_address = None
|
|
if request:
|
|
ip_address = request.client.host if request.client else None
|
|
|
|
log = OperationLog(user_id=user_id, operation=operation, details=details, ip_address=ip_address)
|
|
db.add(log)
|
|
db.commit()
|
|
|
|
|
|
def get_user_dict(user: User, include_password: bool = False) -> dict:
|
|
"""获取用户字典表示"""
|
|
return user.to_dict(include_password)
|
|
|
|
|
|
def convert_serializable(obj):
|
|
"""将对象转换为可序列化的格式"""
|
|
if isinstance(obj, list | tuple):
|
|
return [convert_serializable(item) for item in obj]
|
|
if isinstance(obj, dict):
|
|
return {k: convert_serializable(v) for k, v in obj.items()}
|
|
if hasattr(obj, "__dict__"):
|
|
return convert_serializable(vars(obj))
|
|
return obj
|