使用中间件全局捕获异常,再进行注册使用
from fastapi import FastAPI, Request
from pathlib import Path
from loguru import logger
def create_app():
api = FastAPI(title=settings.TITLE, description=settings.DESC, version='1.0.0', docs_url='/docs')
# 跨域
register_cors(api)
# redis
register_redis(api)
# 路由
register_router(api)
# 后台
register_admin(api)
# 定时任务
# register_aps(api, settings.SQLALCHEMY_DATABASE_URL2)
# 日志
register_log(api)
return api
def register_log(api: FastAPI):
@api.middleware('http')
async def add_log(request: Request, call_next):
url = request.url
log_dir = BACKEND_DIR / 'logs'
log_path = log_dir / f'{time.strftime("%Y-%m-%d")}.log'
# 不在终端打印
logger.remove(handler_id=None)
# diagnose backtrace 为true 日志内将增加回溯, format也可以格式化处理
# message "\n\n{time} {level} {message}" 将是默认内容很多所以没用
logger.add(log_path, rotation='0:00', enqueue=True, serialize=False,
encoding="utf-8", retention="7 days", diagnose=False, backtrace=True, format=format_log)
try:
response = await call_next(request)
except Exception as e:
logger.exception(f"{url}What?!\n{e}")
else:
return response
参考文章:Python - 日志管理模块: Loguru的使用_bailang_zhizun的博客-CSDN博客_loguru和官方文档
感谢大佬们浏览!!!