def _compare_columns(schema, tname, object_filters, conn_table, metadata_table, diffs, autogen_context, inspector):
name = "%s.%s" % (schema, tname) if schema else tname
metadata_cols_by_name = dict((c.name, c) for c in metadata_table.c)
conn_col_names = dict((c.name, c) for c in conn_table.c)
metadata_col_names = OrderedSet(sorted(metadata_cols_by_name))
for cname in metadata_col_names.difference(conn_col_names):
if _run_filters(metadata_cols_by_name[cname], cname, "column", False, None, object_filters):
diffs.append(("add_column", schema, tname, metadata_cols_by_name[cname]))
log.info("Detected added column '%s.%s'", name, cname)
for cname in set(conn_col_names).difference(metadata_col_names):
if _run_filters(conn_table.c[cname], cname, "column", True, None, object_filters):
diffs.append(("remove_column", schema, tname, conn_table.c[cname]))
log.info("Detected removed column '%s.%s'", name, cname)
for colname in metadata_col_names.intersection(conn_col_names):
metadata_col = metadata_cols_by_name[colname]
conn_col = conn_table.c[colname]
if not _run_filters(metadata_col, colname, "column", False, conn_col, object_filters):
continue
col_diff = []
_compare_type(schema, tname, colname, conn_col, metadata_col, col_diff, autogen_context)
_compare_nullable(schema, tname, colname, conn_col, metadata_col.nullable, col_diff, autogen_context)
_compare_server_default(schema, tname, colname, conn_col, metadata_col, col_diff, autogen_context)
if col_diff:
diffs.append(col_diff)
def _compare_columns(schema, tname, object_filters, conn_table, metadata_table,
diffs, autogen_context, inspector):
name = '%s.%s' % (schema, tname) if schema else tname
metadata_cols_by_name = dict((c.name, c) for c in metadata_table.c)
conn_col_names = dict((c.name, c) for c in conn_table.c)
metadata_col_names = OrderedSet(sorted(metadata_cols_by_name))
for cname in metadata_col_names.difference(conn_col_names):
if _run_filters(metadata_cols_by_name[cname], cname, "column", False,
None, object_filters):
diffs.append(
("add_column", schema, tname, metadata_cols_by_name[cname]))
log.info("Detected added column '%s.%s'", name, cname)
for cname in set(conn_col_names).difference(metadata_col_names):
if _run_filters(conn_table.c[cname], cname, "column", True, None,
object_filters):
diffs.append(("remove_column", schema, tname, conn_table.c[cname]))
log.info("Detected removed column '%s.%s'", name, cname)
for colname in metadata_col_names.intersection(conn_col_names):
metadata_col = metadata_cols_by_name[colname]
conn_col = conn_table.c[colname]
if not _run_filters(metadata_col, colname, "column", False, conn_col,
object_filters):
continue
col_diff = []
_compare_type(schema, tname, colname, conn_col, metadata_col, col_diff,
autogen_context)
_compare_nullable(schema, tname, colname, conn_col,
metadata_col.nullable, col_diff, autogen_context)
_compare_server_default(schema, tname, colname, conn_col, metadata_col,
col_diff, autogen_context)
if col_diff:
diffs.append(col_diff)
def _compare_columns(schema, tname, conn_table, metadata_table, diffs,
autogen_context):
name = '%s.%s' % (schema, tname) if schema else tname
metadata_cols_by_name = dict((c.name, c) for c in metadata_table.c)
conn_col_names = set(conn_table)
metadata_col_names = OrderedSet(sorted(metadata_cols_by_name))
for cname in metadata_col_names.difference(conn_col_names):
diffs.append(
("add_column", schema, tname, metadata_cols_by_name[cname]))
log.info("Detected added column '%s.%s'", name, cname)
for cname in conn_col_names.difference(metadata_col_names):
diffs.append(
("remove_column", schema, tname,
sa_schema.Column(cname,
conn_table[cname]['type'],
nullable=conn_table[cname]['nullable'],
server_default=conn_table[cname]['default'])))
log.info("Detected removed column '%s.%s'", name, cname)
for colname in metadata_col_names.intersection(conn_col_names):
metadata_col = metadata_cols_by_name[colname]
conn_col = conn_table[colname]
col_diff = []
_compare_type(schema, tname, colname, conn_col, metadata_col, col_diff,
autogen_context)
_compare_nullable(schema, tname, colname, conn_col,
metadata_col.nullable, col_diff, autogen_context)
_compare_server_default(schema, tname, colname, conn_col, metadata_col,
col_diff, autogen_context)
if col_diff:
diffs.append(col_diff)
def _compare_columns(
schema,
tname,
conn_table,
metadata_table,
modify_table_ops,
autogen_context,
inspector,
):
name = "%s.%s" % (schema, tname) if schema else tname
metadata_cols_by_name = dict(
(c.name, c) for c in metadata_table.c if not c.system)
conn_col_names = dict(
(c.name, c) for c in conn_table.c
if autogen_context.run_name_filters(c.name, "column", {
"table_name": tname,
"schema_name": schema
}))
metadata_col_names = OrderedSet(sorted(metadata_cols_by_name))
for cname in metadata_col_names.difference(conn_col_names):
if autogen_context.run_object_filters(metadata_cols_by_name[cname],
cname, "column", False, None):
modify_table_ops.ops.append(
ops.AddColumnOp.from_column_and_tablename(
schema, tname, metadata_cols_by_name[cname]))
log.info("Detected added column '%s.%s'", name, cname)
for colname in metadata_col_names.intersection(conn_col_names):
metadata_col = metadata_cols_by_name[colname]
conn_col = conn_table.c[colname]
if not autogen_context.run_object_filters(metadata_col, colname,
"column", False, conn_col):
continue
alter_column_op = ops.AlterColumnOp(tname, colname, schema=schema)
comparators.dispatch("column")(
autogen_context,
alter_column_op,
schema,
tname,
colname,
conn_col,
metadata_col,
)
if alter_column_op.has_changes():
modify_table_ops.ops.append(alter_column_op)
yield
for cname in set(conn_col_names).difference(metadata_col_names):
if autogen_context.run_object_filters(conn_table.c[cname], cname,
"column", True, None):
modify_table_ops.ops.append(
ops.DropColumnOp.from_column_and_tablename(
schema, tname, conn_table.c[cname]))
log.info("Detected removed column '%s.%s'", name, cname)
def _compare_columns(schema, tname, conn_table, metadata_table,
modify_table_ops, autogen_context, inspector):
name = '%s.%s' % (schema, tname) if schema else tname
metadata_cols_by_name = dict((c.name, c) for c in metadata_table.c)
conn_col_names = dict((c.name, c) for c in conn_table.c)
metadata_col_names = OrderedSet(sorted(metadata_cols_by_name))
for cname in metadata_col_names.difference(conn_col_names):
if autogen_context.run_filters(
metadata_cols_by_name[cname], cname,
"column", False, None):
modify_table_ops.ops.append(
ops.AddColumnOp.from_column_and_tablename(
schema, tname, metadata_cols_by_name[cname])
)
log.info("Detected added column '%s.%s'", name, cname)
for colname in metadata_col_names.intersection(conn_col_names):
metadata_col = metadata_cols_by_name[colname]
conn_col = conn_table.c[colname]
if not autogen_context.run_filters(
metadata_col, colname, "column", False,
conn_col):
continue
alter_column_op = ops.AlterColumnOp(
tname, colname, schema=schema)
comparators.dispatch("column")(
autogen_context, alter_column_op,
schema, tname, colname, conn_col, metadata_col
)
if alter_column_op.has_changes():
modify_table_ops.ops.append(alter_column_op)
yield
for cname in set(conn_col_names).difference(metadata_col_names):
if autogen_context.run_filters(
conn_table.c[cname], cname,
"column", True, None):
modify_table_ops.ops.append(
ops.DropColumnOp.from_column_and_tablename(
schema, tname, conn_table.c[cname]
)
)
log.info("Detected removed column '%s.%s'", name, cname)
def _compare_columns(schema, tname, conn_table, metadata_table,
diffs, autogen_context):
name = '%s.%s' % (schema, tname) if schema else tname
metadata_cols_by_name = dict((c.name, c) for c in metadata_table.c)
conn_col_names = set(conn_table)
metadata_col_names = OrderedSet(sorted(metadata_cols_by_name))
for cname in metadata_col_names.difference(conn_col_names):
diffs.append(
("add_column", schema, tname, metadata_cols_by_name[cname])
)
log.info("Detected added column '%s.%s'", name, cname)
for cname in conn_col_names.difference(metadata_col_names):
diffs.append(
("remove_column", schema, tname, sa_schema.Column(
cname,
conn_table[cname]['type'],
nullable=conn_table[cname]['nullable'],
server_default=conn_table[cname]['default']
))
)
log.info("Detected removed column '%s.%s'", name, cname)
for colname in metadata_col_names.intersection(conn_col_names):
metadata_col = metadata_cols_by_name[colname]
conn_col = conn_table[colname]
col_diff = []
_compare_type(schema, tname, colname,
conn_col,
metadata_col,
col_diff, autogen_context
)
_compare_nullable(schema, tname, colname,
conn_col,
metadata_col.nullable,
col_diff, autogen_context
)
_compare_server_default(schema, tname, colname,
conn_col,
metadata_col,
col_diff, autogen_context
)
if col_diff:
diffs.append(col_diff)
def _compare_columns(
schema: Optional[str],
tname: Union["quoted_name", str],
conn_table: "Table",
metadata_table: "Table",
modify_table_ops: "ModifyTableOps",
autogen_context: "AutogenContext",
inspector: "Inspector",
) -> Iterator[None]:
name = "%s.%s" % (schema, tname) if schema else tname
metadata_col_names = OrderedSet(c.name for c in metadata_table.c
if not c.system)
metadata_cols_by_name = {
c.name: c
for c in metadata_table.c if not c.system
}
conn_col_names = {
c.name: c
for c in conn_table.c
if autogen_context.run_name_filters(c.name, "column", {
"table_name": tname,
"schema_name": schema
})
}
for cname in metadata_col_names.difference(conn_col_names):
if autogen_context.run_object_filters(metadata_cols_by_name[cname],
cname, "column", False, None):
modify_table_ops.ops.append(
ops.AddColumnOp.from_column_and_tablename(
schema, tname, metadata_cols_by_name[cname]))
log.info("Detected added column '%s.%s'", name, cname)
for colname in metadata_col_names.intersection(conn_col_names):
metadata_col = metadata_cols_by_name[colname]
conn_col = conn_table.c[colname]
if not autogen_context.run_object_filters(metadata_col, colname,
"column", False, conn_col):
continue
alter_column_op = ops.AlterColumnOp(tname, colname, schema=schema)
comparators.dispatch("column")(
autogen_context,
alter_column_op,
schema,
tname,
colname,
conn_col,
metadata_col,
)
if alter_column_op.has_changes():
modify_table_ops.ops.append(alter_column_op)
yield
for cname in set(conn_col_names).difference(metadata_col_names):
if autogen_context.run_object_filters(conn_table.c[cname], cname,
"column", True, None):
modify_table_ops.ops.append(
ops.DropColumnOp.from_column_and_tablename(
schema, tname, conn_table.c[cname]))
log.info("Detected removed column '%s.%s'", name, cname)
