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 <liverpoo1026.bne@gmail.com>
Co-authored-by: Kevin Hwa <kevin.hwa@polymathian.com>

diff --git a/alembic/op.pyi b/alembic/op.pyi
index 2a03e1e27ade2e4e000adfe48809973e90e5bf8e..d781ac1bcd86da6e0b19761a25485c6e014c1af1 100644 (file)
--- 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 7d69f0932d3f33ae37fa403c76f9049241026259..99132dd661994465a12cfdea7775bfbf3e337af4 100644 (file)
--- 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 0d40dc7853d539ad3c06888338a4fdccfc6ae2ae..3bff50837a63f255b095858394cc31259de23ede 100644 (file)
--- 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"]: