feat: 日志模块
This commit is contained in:
parent
ad05f55a04
commit
8cedbc8dca
|
|
@ -0,0 +1,7 @@
|
|||
import logging
|
||||
|
||||
logging.basicConfig(
|
||||
level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s:\n%(message)s"
|
||||
)
|
||||
|
||||
logger = logging.getLogger("workflow")
|
||||
|
|
@ -3,8 +3,9 @@ from fastapi import FastAPI
|
|||
from models.schemas.system import InitStatusResponse, InitConfigRequest
|
||||
from core.config import load_init_config, save_init_config
|
||||
from core.db import init_tortoise, close_tortoise
|
||||
from core.logger import logger
|
||||
from contextlib import asynccontextmanager
|
||||
import logging
|
||||
|
||||
|
||||
async def is_init():
|
||||
init_config = await load_init_config()
|
||||
|
|
@ -16,17 +17,18 @@ async def is_init():
|
|||
async def lifespan(app: FastAPI):
|
||||
|
||||
if await is_init():
|
||||
app.logger.info("初始化数据库")
|
||||
await init_tortoise()
|
||||
logger.info("初始化数据库")
|
||||
init_config = await load_init_config()
|
||||
await init_tortoise(init_config["db_type"], init_config)
|
||||
|
||||
yield
|
||||
|
||||
app.logger.info("关闭数据库")
|
||||
logger.info("关闭数据库")
|
||||
await close_tortoise()
|
||||
|
||||
|
||||
def create_app():
|
||||
app = FastAPI(title="Workflow API", version="1.0.0")
|
||||
app = FastAPI(title="Workflow API", version="1.0.0", lifespan=lifespan)
|
||||
|
||||
@app.get("/")
|
||||
async def root():
|
||||
|
|
@ -48,7 +50,7 @@ def create_app():
|
|||
response_model=InitStatusResponse,
|
||||
)
|
||||
async def init_system(request: InitConfigRequest):
|
||||
app.logger.info("init_system")
|
||||
logger.info("init_system")
|
||||
try:
|
||||
init_config = await load_init_config()
|
||||
if init_config.get('is_init'):
|
||||
|
|
@ -75,8 +77,6 @@ def create_app():
|
|||
|
||||
|
||||
app = create_app()
|
||||
app.logger = logging.getLogger("workflow")
|
||||
app.logger.setLevel(logging.INFO)
|
||||
|
||||
if __name__ == "__main__":
|
||||
uvicorn.run("main:app", host="0.0.0.0", port=8080, reload=True)
|
||||
|
|
|
|||
Binary file not shown.
Binary file not shown.
Loading…
Reference in New Issue