From f06961c5a566f16be26bbd70b5682ba16eb1ed9b Mon Sep 17 00:00:00 2001 From: Wu Clan Date: Thu, 7 Aug 2025 16:51:13 +0800 Subject: [PATCH] Update the naming of table creation function --- backend/core/registrar.py | 4 ++-- backend/database/db.py | 6 +++++- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/backend/core/registrar.py b/backend/core/registrar.py index 439a2420..0a23b979 100644 --- a/backend/core/registrar.py +++ b/backend/core/registrar.py @@ -18,7 +18,7 @@ from backend.common.log import set_custom_logfile, setup_logging from backend.core.conf import settings from backend.core.path_conf import STATIC_DIR, UPLOAD_DIR -from backend.database.db import create_table +from backend.database.db import create_tables from backend.database.redis import redis_client from backend.middleware.access_middleware import AccessMiddleware from backend.middleware.jwt_auth_middleware import JwtAuthMiddleware @@ -40,7 +40,7 @@ async def register_init(app: FastAPI) -> AsyncGenerator[None, None]: :return: """ # 创建数据库表 - await create_table() + await create_tables() # 初始化 limiter await FastAPILimiter.init( redis=redis_client, diff --git a/backend/database/db.py b/backend/database/db.py index 529bd9c1..1eaf0dc9 100644 --- a/backend/database/db.py +++ b/backend/database/db.py @@ -75,7 +75,7 @@ async def get_db() -> AsyncGenerator[AsyncSession, None]: yield session -async def create_table() -> None: +async def create_tables() -> None: """创建数据库表""" async with async_engine.begin() as coon: await coon.run_sync(MappedBase.metadata.create_all) @@ -86,7 +86,11 @@ def uuid4_str() -> str: return str(uuid4()) +# SQLA 数据库链接 SQLALCHEMY_DATABASE_URL = create_database_url() + +# SALA 异步引擎和会话 async_engine, async_db_session = create_async_engine_and_session(SQLALCHEMY_DATABASE_URL) + # Session Annotated CurrentSession = Annotated[AsyncSession, Depends(get_db)]