fix:暂时移除alembic,需要默认创建数据库

This commit is contained in:
lan
2022-12-14 11:16:30 +08:00
parent 1994ba5917
commit 2a102bfb1f
+3 -5
View File
@@ -1,20 +1,18 @@
import datetime import datetime
from alembic import command
from alembic.config import Config
from sqlalchemy import Boolean, Column, Integer, String, DateTime from sqlalchemy import Boolean, Column, Integer, String, DateTime
from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.ext.asyncio import create_async_engine from sqlalchemy.ext.asyncio import create_async_engine
from sqlalchemy.ext.asyncio.session import AsyncSession from sqlalchemy.ext.asyncio.session import AsyncSession
import settings import settings
engine = create_async_engine(settings.DATABASE_URL) engine = create_async_engine(settings.DATABASE_URL)
Base = declarative_base() Base = declarative_base()
async def init_models(): async def init_models():
config = Config() async with engine.begin() as conn:
config.set_main_option("script_location", "migrations") await conn.run_sync(Base.metadata.create_all)
command.upgrade(config, "head")
async def get_session(): async def get_session():