Steven
d390b7b58a
CI / backend (push) Canceled after 0s
fnsvr-01 的 PostgreSQL 在 8432 端口（不是 5432），实测 PostgreSQL 17.6 / Alpine。 已在该实例建好独立库与角色： - dadaotask（开发）/ dadaotask_test（测试），owner 都是 dadao_owner - dadao_owner 迁移用；dadao_app 应用用（NOSUPERUSER NOBYPASSRLS） - 两个库都已 ALTER SCHEMA public OWNER TO dadao_owner + REVOKE CREATE FROM PUBLIC + GRANT 给 dadao_app 代码改动： - config 增加 test_database_url / test_migration_database_url，并给出 resolved_database_url() / resolved_migration_database_url()：ENV=test 且有测试库时优先用测试库， 否则退回开发库（CI 里直接给 DATABASE_URL 即可，行为不变）。 - db.py / alembic/env.py / tests/conftest.py 统一走这两个解析方法。 原先 conftest 清表用的是 settings.migration_database_url，而应用连的是测试库， 一旦测试库与开发库不同，TRUNCATE 就清到另一个库上——表现为"测试库脏数据越攒越多、 列表类断言随机失败"。这是本次真踩到的坑，不是假想。 - tests/conftest.py 保留 E402 豁免（环境变量必须在 import app 模块前写好）。 验证： - 远程 dadaotask_test 上 pytest 33 passed，跑完表内 0 行（确认清表打在同一个库） - 远程 dadaotask 上 alembic upgrade head 成功，/health 与 /version 正常，db_role 为 dadao_app 且三项全 false - 本机便携 PG 16.4 上同样 33 passed（两个 PG 大版本都过） - ruff check / format 干净 已知限制：该实例没有 pgvector（只有 pg_trgm / pgcrypto），M3 向量检索前必须先定扩展来源， 已写进 2.Design/DEVELOPMENT.md。
69 lines
1.9 KiB
Python
69 lines
1.9 KiB
Python
"""Alembic 环境：用**迁移角色**（owner）连接，与应用的受限角色分开。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from logging.config import fileConfig
|
|
|
|
from sqlalchemy import pool
|
|
from sqlalchemy.engine import Connection
|
|
from sqlalchemy.ext.asyncio import async_engine_from_config
|
|
|
|
from alembic import context
|
|
from app.config import get_settings
|
|
from app.models import Base
|
|
|
|
config = context.config
|
|
if config.config_file_name is not None:
|
|
fileConfig(config.config_file_name)
|
|
|
|
_settings = get_settings()
|
|
# 测试环境用独立测试库（conftest 会在 ENV=test 下跑迁移）
|
|
config.set_main_option("sqlalchemy.url", _settings.resolved_migration_database_url())
|
|
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"},
|
|
compare_type=True,
|
|
)
|
|
with context.begin_transaction():
|
|
context.run_migrations()
|
|
|
|
|
|
def do_run_migrations(connection: Connection) -> None:
|
|
context.configure(
|
|
connection=connection,
|
|
target_metadata=target_metadata,
|
|
compare_type=True,
|
|
# RLS 策略、GRANT 这类"非表结构"变更 autogenerate 看不见，必须手写
|
|
include_schemas=False,
|
|
)
|
|
with context.begin_transaction():
|
|
context.run_migrations()
|
|
|
|
|
|
async def run_async_migrations() -> None:
|
|
connectable = async_engine_from_config(
|
|
config.get_section(config.config_ini_section, {}),
|
|
prefix="sqlalchemy.",
|
|
poolclass=pool.NullPool,
|
|
)
|
|
async with connectable.connect() as connection:
|
|
await connection.run_sync(do_run_migrations)
|
|
await connectable.dispose()
|
|
|
|
|
|
def run_migrations_online() -> None:
|
|
asyncio.run(run_async_migrations())
|
|
|
|
|
|
if context.is_offline_mode():
|
|
run_migrations_offline()
|
|
else:
|
|
run_migrations_online()
|