Add CTE prefixes (#157)

sqlalchemy/sqlalchemy#5043 added support for CTE prefix clauses in PostgreSQL 12. This PR fixes the dreadful `"CTE" has no attribute "prefix_with"` error for me.
dropbox · Jan 8, 2021 · b3c1613 · b3c1613
1 parent c748ba3
commit b3c1613
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/sqlalchemy-stubs/sql/selectable.pyi b/sqlalchemy-stubs/sql/selectable.pyi
@@ -122,11 +122,12 @@ class TableSample(Alias):
     def __init__(self, selectable: FromClause, sampling: Union[float, Function[float]], name: Optional[str] = ...,
                  seed: Optional[Any] = ...) -> None: ...
 
-class CTE(Generative, HasSuffixes, Alias):
+class CTE(Generative, HasPrefixes, HasSuffixes, Alias):
     __visit_name__: str = ...
     recursive: bool = ...
     def __init__(self, selectable: Select, name: Optional[str] = ..., recursive: bool = ...,
-                 _cte_alias: Optional[Any] = ..., _restates: Any = ..., _suffixes: Optional[Any] = ...) -> None: ...
+                 _cte_alias: Optional[Any] = ..., _restates: Any = ..., _prefixes: Optional[Any] = ...,
+                 _suffixes: Optional[Any] = ...) -> None: ...
     def alias(self, name: Optional[str] = ..., flat: bool = ...) -> CTE: ...
     def union(self, other: Select) -> CTE: ...
     def union_all(self, other: Select) -> CTE: ...