SQLAlchemy: Annotate text()

python · AlexWaygood · Jan 11, 2023 · Jan 11, 2023 · Jan 11, 2023 · Jan 11, 2023
commit 752e5c8b15c9cbc1af32a9c0d00d78fdfd429fc6
diff --git a/stubs/SQLAlchemy/sqlalchemy/sql/elements.pyi b/stubs/SQLAlchemy/sqlalchemy/sql/elements.pyi
@@ -1,3 +1,4 @@
+from _typeshed import Incomplete
 from typing import Any, Generic, TypeVar
 from typing_extensions import Literal
 
@@ -136,7 +137,7 @@ class TextClause(
     def __and__(self, other): ...
     key: Any
     text: Any
-    def __init__(self, text, bind: Any | None = ...): ...
+    def __init__(self, text: str, bind: Incomplete | None = None) -> None: ...
     def bindparams(self, *binds, **names_to_values) -> None: ...
     def columns(self, *cols, **types): ...
     @property

diff --git a/stubs/SQLAlchemy/sqlalchemy/sql/expression.pyi b/stubs/SQLAlchemy/sqlalchemy/sql/expression.pyi
@@ -163,7 +163,9 @@ lateral: Incomplete
 or_: Incomplete
 bindparam: Incomplete
 select: Incomplete
-text: Incomplete
+
+def text(text: str, bind: Incomplete | None = None) -> TextClause: ...
+
 table: Incomplete
 column: Incomplete
 over: Incomplete