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

This commit is contained in:
lan
2022-12-14 11:16:30 +08:00
parent 77c21a274d
commit c517be78e9
+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():