From 2a102bfb1f98f209720f03ef9221e5fb247c7303 Mon Sep 17 00:00:00 2001 From: lan Date: Wed, 14 Dec 2022 11:16:30 +0800 Subject: [PATCH] =?UTF-8?q?fix:=E6=9A=82=E6=97=B6=E7=A7=BB=E9=99=A4alembic?= =?UTF-8?q?=EF=BC=8C=E9=9C=80=E8=A6=81=E9=BB=98=E8=AE=A4=E5=88=9B=E5=BB=BA?= =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=BA=93?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- database.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/database.py b/database.py index e36bc53..d8eea01 100644 --- a/database.py +++ b/database.py @@ -1,20 +1,18 @@ import datetime -from alembic import command -from alembic.config import Config from sqlalchemy import Boolean, Column, Integer, String, DateTime from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.ext.asyncio import create_async_engine from sqlalchemy.ext.asyncio.session import AsyncSession import settings + engine = create_async_engine(settings.DATABASE_URL) Base = declarative_base() async def init_models(): - config = Config() - config.set_main_option("script_location", "migrations") - command.upgrade(config, "head") + async with engine.begin() as conn: + await conn.run_sync(Base.metadata.create_all) async def get_session():