diff --git a/backend/app/core/logger.py b/backend/app/core/logger.py new file mode 100644 index 0000000..8581fb9 --- /dev/null +++ b/backend/app/core/logger.py @@ -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") diff --git a/backend/app/main.py b/backend/app/main.py index 371e831..80ba2b2 100644 --- a/backend/app/main.py +++ b/backend/app/main.py @@ -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) diff --git a/backend/app/models/schemas/__init__.py b/backend/app/models/schemas/__init__.py deleted file mode 100644 index e69de29..0000000 diff --git a/backend/your_workflow b/backend/your_workflow new file mode 100644 index 0000000..cb386eb Binary files /dev/null and b/backend/your_workflow differ diff --git a/backend/your_workflow-shm b/backend/your_workflow-shm new file mode 100644 index 0000000..fe9ac28 Binary files /dev/null and b/backend/your_workflow-shm differ diff --git a/backend/app/models/__init__.py b/backend/your_workflow-wal similarity index 100% rename from backend/app/models/__init__.py rename to backend/your_workflow-wal