ghplvh
981bee5a46
- src/ 按域组织骨架，示例域 items 双后端实现（SQLAlchemy2.0 async / Beanie 2.x） - DB_BACKEND 环境变量切换，router 只依赖 Repo Protocol - alembic 迁移（含初始 items 迁移）、uv 依赖管理、ruff、pytest 异步测试 - Dockerfile + docker-compose（app/mysql/mongo） - AGENTS.md 重写为本模板开发规范；原最佳实践文档归档 docs/ - 验证：mongodb/mysql 双后端 pytest 全绿 + uvicorn 真实冒烟通过
54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
"""Alembic async env：仅用于 MySQL 后端。MongoDB 无 schema 迁移概念。"""
|
||
|
||
import asyncio
|
||
from logging.config import fileConfig
|
||
|
||
from sqlalchemy.ext.asyncio import async_engine_from_config
|
||
|
||
from alembic import context
|
||
|
||
config = context.config
|
||
if config.config_file_name is not None:
|
||
fileConfig(config.config_file_name)
|
||
|
||
# 从应用配置注入 DSN + metadata
|
||
import src.items.mysql
|
||
from src.config import settings
|
||
from src.database import Base
|
||
|
||
config.set_main_option("sqlalchemy.url", settings.MYSQL_DSN)
|
||
target_metadata = Base.metadata
|
||
|
||
|
||
def run_migrations_offline() -> None:
|
||
context.configure(
|
||
url=config.get_main_option("sqlalchemy.url"),
|
||
target_metadata=target_metadata,
|
||
literal_binds=True,
|
||
dialect_opts={"paramstyle": "named"},
|
||
)
|
||
with context.begin_transaction():
|
||
context.run_migrations()
|
||
|
||
|
||
def do_run_migrations(connection) -> None:
|
||
context.configure(connection=connection, target_metadata=target_metadata)
|
||
with context.begin_transaction():
|
||
context.run_migrations()
|
||
|
||
|
||
async def run_migrations_online() -> None:
|
||
connectable = async_engine_from_config(
|
||
config.get_section(config.config_ini_section, {}),
|
||
prefix="sqlalchemy.",
|
||
)
|
||
async with connectable.connect() as connection:
|
||
await connection.run_sync(do_run_migrations)
|
||
await connectable.dispose()
|
||
|
||
|
||
if context.is_offline_mode():
|
||
run_migrations_offline()
|
||
else:
|
||
asyncio.run(run_migrations_online())
|