diff --git a/main.py b/main.py index 27812d9..f67c767 100644 --- a/main.py +++ b/main.py @@ -2,7 +2,7 @@ import logging.config as logging_config import os import fastapi_plugins -from fastapi import FastAPI +from fastapi import FastAPI, Request from fastapi.exceptions import HTTPException, RequestValidationError from fastapi_sqlalchemy import DBSessionMiddleware @@ -17,12 +17,13 @@ def create_app(): @app.on_event("startup") async def startup_event(): # 创建日志文件夹和临时文件上传文件夹 - if not os.path.exists("logs"): - os.mkdir("logs") if not os.path.exists("files"): os.mkdir("files") # 日志配置 - # logging_config.fileConfig('conf/log.ini') + if os.getenv("FAST_API_ENV", "dev") == "prod": + if not os.path.exists("logs"): + os.mkdir("logs") + logging_config.fileConfig('conf/log.ini') # 初始化配置文件 mysql_config, redis_config = init_config() # 添加sqlalchemy数据库中间件