Identity and Computed comparison skipped when compare_server_default is not set, as of 1.18.0

Abierto
  • ¿Algo no está bien?
#1,877 1 comentario 0 reacciones 0 asignados Ver en GitHub

Nadie ha tomado este issue todavía.

  • #1878 de @Fire162 — cerrado sin fusionar
  • #1882 de @Fire162 — cerrado sin fusionar
  • #1885 de @anish-yen — cerrado sin fusionar

Evaluación

Dificultad
3/5
Tiempo estimado
1-2 días
Aptitud para principiantes
50/100
Tipo de issue
Error
Claridad
Bien especificado
Estado de actividad
Activo
Stack tecnológico
postgresql, python, sqlalchemy
Área
database

Línea de trabajo

Start with _user_compare_server_default() in alembic/autogenerate/compare/server_defaults.py, where the early STOP for compare_server_default is False ends the dispatch chain before the computed and identity comparators run. Compare it with the old gate in MigrationContext._compare_server_default() and the call site in alembic/autogenerate/compare.py. Done when Identity and Computed changes produce diffs under default options, with a test that sets compare_server_default to False, since AutogenFixtureTest always sets it to True. Linked PR #1878 was closed unmerged, so check why before reusing it.

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

autogenerate - defaults bug regression

Describe the bug

As of 1.18.0, autogenerate no longer detects Identity changes on an existing column, and no longer emits the "Computed default on ... cannot be modified" warning, unless compare_server_default is set to True or a callable. With the default of compare_server_default=False, adding, removing or altering an Identity produces no diff at all. In 1.17.2 and earlier these were detected regardless of that option.

Expected behavior

Identity and computed comparison should run independently of compare_server_default, as they did through 1.17.2. That option only gated the comparison of plain server defaults.

To Reproduce

import warnings

from sqlalchemy import Column
from sqlalchemy import Computed
from sqlalchemy import create_engine
from sqlalchemy import Identity
from sqlalchemy import Integer
from sqlalchemy import MetaData
from sqlalchemy import Table

import alembic
from alembic.autogenerate import compare_metadata
from alembic.migration import MigrationContext

engine = create_engine("postgresql://scott:tiger@localhost/test")


def table(metadata, *n_args):
    Table(
        "t",
        metadata,
        Column("id", Integer, primary_key=True),
        Column("bar", Integer),
        Column("n", Integer, *n_args, nullable=False),
    )


cases = [
    ("add Identity", (), (Identity(),)),
    ("remove Identity", (Identity(),), ()),
    (
        "change Identity",
        (Identity(increment=1),),
        (Identity(increment=5),),
    ),
    (
        "change Computed",
        (Computed("bar * 5", persisted=True),),
        (Computed("bar * 42", persisted=True),),
    ),
]

print(f"alembic {alembic.__version__}")
for label, in_database, in_model in cases:
    db_metadata = MetaData()
    table(db_metadata, *in_database)
    model_metadata = MetaData()
    table(model_metadata, *in_model)

    with engine.begin() as conn:
        db_metadata.drop_all(conn)
        db_metadata.create_all(conn)

    try:
        with engine.connect() as conn:
            # default options; compare_server_default is not set
            context = MigrationContext.configure(conn)
            with warnings.catch_warnings(record=True) as w:
                warnings.simplefilter("always")
                diffs = compare_metadata(context, model_metadata)
    finally:
        with engine.begin() as conn:
            db_metadata.drop_all(conn)

    diffs = [d[0] for group in diffs for d in group]
    warned = [str(x.message) for x in w if "Computed" in str(x.message)]
    print(f"{label}: diffs={diffs} warnings={warned}")

Output on 1.17.2:

alembic 1.17.2
add Identity: diffs=['modify_default'] warnings=[]
remove Identity: diffs=['modify_default'] warnings=[]
change Identity: diffs=['modify_default'] warnings=[]
change Computed: diffs=[] warnings=['Computed default on t.n cannot be modified']

Output on 1.18.0, and unchanged on current main:

alembic 1.18.0
add Identity: diffs=[] warnings=[]
remove Identity: diffs=[] warnings=[]
change Identity: diffs=[] warnings=[]
change Computed: diffs=[] warnings=[]

Cause

Before the plugin reorganization in e532a7e3, the False check was inside MigrationContext._compare_server_default(), which autogenerate/compare.py called only for plain server defaults. The computed and identity branches ran ahead of it and never consulted the option.

Those branches are now separate comparators on the "column" / "server_default" dispatch chain, and the check moved into _user_compare_server_default() in alembic/autogenerate/compare/server_defaults.py, which is registered with DispatchPriority.FIRST:

if migration_context._user_compare_server_default is False:
    return PriorityDispatchResult.STOP

STOP ends the chain, so _compare_computed_default(), _compare_identity_default() and _dialect_impl_compare_server_default() are all skipped. Only the last of those was gated by the option previously.

The test suite does not catch this because AutogenFixtureTest always configures compare_server_default=True; a test for the fix needs to run with the option at its default.

Versions

  • Alembic: 1.18.0 through 1.20.0, and current main
  • Database: PostgreSQL 18 (the Identity cases need a backend with identity columns; the dispatch logic is not backend specific)
  • SQLAlchemy: 2.1
Lenguaje dominante
Python
Estrellas
4.5k
Forks
391
Sin PR fusionados en 30 d

Preparar el entorno

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Más de sqlalchemy/alembic

Todos los issues de sqlalchemy/alembic

Issues similares

Más issues de Python

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.