使用中间件全局捕获异常,再进行注册使用

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和官方文档
感谢大佬们浏览!!!


Logo

加入社区!打开量化的大门,首批课程上线啦!

更多推荐