新增 alembic 管理数据库架构

database.py

@@ -1,18 +1,23 @@
 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
 
-engine = create_async_engine("sqlite+aiosqlite:///database.db")
+import settings
+
+engine = create_async_engine(f"sqlite+aiosqlite:///{settings.DATABASE_FILE}")
 
 Base = declarative_base()
 
 
 async def init_models():
-    async with engine.begin() as conn:
-        await conn.run_sync(Base.metadata.create_all)
+    config = Config()
+    config.set_main_option("script_location", "migrations")
+    command.upgrade(config, "head")
 
 
 async def get_session():