From 9b01e5fa7178333f2e78ee0fc1322112307b51dd Mon Sep 17 00:00:00 2001 From: liverpool1026 Date: Fri, 8 Oct 2021 05:46:58 +1000 Subject: Fix issue 928 create_check_constraint condition argument typing fix (#929) * Update create_check_constraint method typing * Pass str type down Fixes: #928 Co-authored-by: Kevin Hwa Co-authored-by: Kevin Hwa --- alembic/op.pyi | 2 +- alembic/operations/ops.py | 4 ++-- alembic/operations/schemaobj.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/alembic/op.pyi b/alembic/op.pyi index 2a03e1e..d781ac1 100644 --- a/alembic/op.pyi +++ b/alembic/op.pyi @@ -422,7 +422,7 @@ def bulk_insert( def create_check_constraint( constraint_name: Optional[str], table_name: str, - condition: "BinaryExpression", + condition: Union[str, "BinaryExpression"], schema: Optional[str] = None, **kw ) -> Optional["Table"]: diff --git a/alembic/operations/ops.py b/alembic/operations/ops.py index 7d69f09..99132dd 100644 --- a/alembic/operations/ops.py +++ b/alembic/operations/ops.py @@ -732,7 +732,7 @@ class CreateCheckConstraintOp(AddConstraintOp): self, constraint_name: Optional[str], table_name: str, - condition: Union["TextClause", "ColumnElement[Any]"], + condition: Union[str, "TextClause", "ColumnElement[Any]"], schema: Optional[str] = None, **kw ) -> None: @@ -778,7 +778,7 @@ class CreateCheckConstraintOp(AddConstraintOp): operations: "Operations", constraint_name: Optional[str], table_name: str, - condition: "BinaryExpression", + condition: Union[str, "BinaryExpression"], schema: Optional[str] = None, **kw ) -> Optional["Table"]: diff --git a/alembic/operations/schemaobj.py b/alembic/operations/schemaobj.py index 0d40dc7..3bff508 100644 --- a/alembic/operations/schemaobj.py +++ b/alembic/operations/schemaobj.py @@ -137,7 +137,7 @@ class SchemaObjects: self, name: Optional[str], source: str, - condition: Union["TextClause", "ColumnElement[Any]"], + condition: Union[str, "TextClause", "ColumnElement[Any]"], schema: Optional[str] = None, **kw ) -> Union["CheckConstraint"]: -- cgit v1.2.1