From 78ff7c60d4865d3c23f37eae070038ab917ed186 Mon Sep 17 00:00:00 2001
From: chenwj <654891551@qq.com>
Date: Wed, 11 Jan 2023 11:53:45 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E6=94=B9=E5=BC=80=E5=8F=91?=
 =?UTF-8?q?=E7=8E=AF=E5=A2=83=E6=97=A5=E5=BF=97=E9=85=8D=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

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数据库中间件