diff options
Diffstat (limited to 'alembic/ddl/postgresql.py')
-rw-r--r-- | alembic/ddl/postgresql.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/alembic/ddl/postgresql.py b/alembic/ddl/postgresql.py index 32674d2..e7c85bd 100644 --- a/alembic/ddl/postgresql.py +++ b/alembic/ddl/postgresql.py @@ -419,7 +419,7 @@ class CreateExcludeConstraintOp(ops.AddConstraintOp): def __init__( self, - constraint_name: Optional[str], + constraint_name: sqla_compat._ConstraintName, table_name: Union[str, quoted_name], elements: Union[ Sequence[Tuple[str, str]], @@ -443,7 +443,6 @@ class CreateExcludeConstraintOp(ops.AddConstraintOp): cls, constraint: ExcludeConstraint ) -> CreateExcludeConstraintOp: constraint_table = sqla_compat._table_for_constraint(constraint) - return cls( constraint.name, constraint_table.name, @@ -451,7 +450,9 @@ class CreateExcludeConstraintOp(ops.AddConstraintOp): (expr, op) for expr, name, op in constraint._render_exprs # type:ignore[attr-defined] # noqa ], - where=constraint.where, + where=cast( + "Optional[Union[BinaryExpression, str]]", constraint.where + ), schema=constraint_table.schema, _orig_constraint=constraint, deferrable=constraint.deferrable, |