diff options
| author | mike bayer <mike_mp@zzzcomputing.com> | 2023-03-01 21:27:09 +0000 |
|---|---|---|
| committer | Gerrit Code Review <gerrit@bbpush.zzzcomputing.com> | 2023-03-01 21:27:09 +0000 |
| commit | 192e4de9964e11f7f4f04472af7a1e09b0356a1c (patch) | |
| tree | 55b3631bc43ebec025f58821de9cb75eff853491 /alembic/ddl | |
| parent | bbc56256572f0753466a15284eaf3602f48d2cc9 (diff) | |
| parent | 714b74478313a69042bf61cc42a585fc6dddf2e4 (diff) | |
| download | alembic-192e4de9964e11f7f4f04472af7a1e09b0356a1c.tar.gz | |
Merge "Improve typing to accommodate sqlalchemy v2" into main
Diffstat (limited to 'alembic/ddl')
| -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, |