op_container: ops.OpContainer,
autogen_context: AutogenContext,
) -> str:
-
buf = StringIO()
printer = PythonPrinter(buf)
def _render_create_table_comment(
autogen_context: AutogenContext, op: ops.CreateTableCommentOp
) -> str:
-
templ = (
"{prefix}create_table_comment(\n"
"{indent}'{tname}',\n"
def _render_drop_table_comment(
autogen_context: AutogenContext, op: ops.DropTableCommentOp
) -> str:
-
templ = (
"{prefix}drop_table_comment(\n"
"{indent}'{tname}',\n"
def _add_fk_constraint(
autogen_context: AutogenContext, op: ops.CreateForeignKeyOp
) -> str:
-
args = [repr(_render_gen_name(autogen_context, op.constraint_name))]
if not autogen_context._has_batch:
args.append(repr(_ident(op.source_table)))
def _drop_constraint(
autogen_context: AutogenContext, op: ops.DropConstraintOp
) -> str:
-
if autogen_context._has_batch:
template = "%(prefix)sdrop_constraint" "(%(name)r, type_=%(type)r)"
else:
@renderers.dispatch_for(ops.AddColumnOp)
def _add_column(autogen_context: AutogenContext, op: ops.AddColumnOp) -> str:
-
schema, tname, column = op.schema, op.table_name, op.column
if autogen_context._has_batch:
template = "%(prefix)sadd_column(%(column)s)"
@renderers.dispatch_for(ops.DropColumnOp)
def _drop_column(autogen_context: AutogenContext, op: ops.DropColumnOp) -> str:
-
schema, tname, column_name = op.schema, op.table_name, op.column_name
if autogen_context._has_batch:
def _alter_column(
autogen_context: AutogenContext, op: ops.AlterColumnOp
) -> str:
-
tname = op.table_name
cname = op.column_name
server_default = op.modify_server_default
is_server_default: bool = False,
) -> str:
if isinstance(value, sql.ClauseElement):
-
if wrap_in_text:
template = "%(prefix)stext(%(sql)r)"
else:
opts.append(
("name", _render_gen_name(autogen_context, constraint.name))
)
- pg_nulls_not_distinct = constraint.dialect_options["postgresql"]["nulls_not_distinct"]
+ pg_nulls_not_distinct = constraint.dialect_options["postgresql"][
+ "nulls_not_distinct"
+ ]
if pg_nulls_not_distinct is not None:
- opts.append(
- ("postgresql_nulls_not_distinct", pg_nulls_not_distinct)
- )
+ opts.append(("postgresql_nulls_not_distinct", pg_nulls_not_distinct))
if alter:
args = [repr(_render_gen_name(autogen_context, constraint.name))]
opts: List[Tuple[str, Any]] = []
if column.server_default:
-
rendered = _render_server_default( # type:ignore[assignment]
column.server_default, autogen_context
)
def _populate_render_fk_opts(
constraint: ForeignKeyConstraint, opts: List[Tuple[str, str]]
) -> None:
-
if constraint.onupdate:
opts.append(("onupdate", repr(constraint.onupdate)))
if constraint.ondelete:
)
def test_generic_array_type(self):
-
eq_ignore_whitespace(
autogenerate.render._repr_type(
types.ARRAY(Integer), self.autogen_context
)
def test_render_variant(self):
-
self.autogen_context.opts["user_module_prefix"] = None
type_ = (
class RenderNamingConventionTest(TestBase):
def setUp(self):
-
convention = {
"ix": "ix_%(custom)s_%(column_0_label)s",
"uq": "uq_%(custom)s_%(table_name)s_%(column_0_name)s",
t = Table("t", m, Column("c", Integer))
eq_ignore_whitespace(
autogenerate.render._render_unique_constraint(
- UniqueConstraint(t.c.c, name="uq_1", deferrable="XYZ",
- postgresql_nulls_not_distinct=True),
+ UniqueConstraint(
+ t.c.c,
+ name="uq_1",
+ deferrable="XYZ",
+ postgresql_nulls_not_distinct=True,
+ ),
self.autogen_context,
None,
),