62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
"""通用工具函数"""
|
||
|
||
import logging
|
||
|
||
from fastapi import Request
|
||
from sqlalchemy.orm import Session
|
||
|
||
from yuxi.storage.postgres.models_business 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")
|
||
|
||
# 禁用默认的uvicorn访问日志（因为我们使用自定义中间件）
|
||
uvicorn_access_logger.handlers.clear()
|
||
|
||
# 创建格式化器
|
||
formatter = logging.Formatter(fmt="%(asctime)s %(levelname)s: %(message)s", datefmt="%m-%d %H:%M:%S")
|
||
|
||
# 为uvicorn主日志设置格式化器
|
||
for handler in uvicorn_logger.handlers:
|
||
handler.setFormatter(formatter)
|
||
|
||
|
||
async def log_operation(db: Session, user_id: int, operation: str, details: str = None, request: Request = None):
|
||
"""记录用户操作日志"""
|
||
try:
|
||
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)
|
||
await db.commit()
|
||
except Exception:
|
||
# 日志写入失败不影响主业务
|
||
pass
|
||
|
||
|
||
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
|