Add dialect typing

Author: bryanforbes

sqlalchemy-stubs/sql/type_api.pyi

@@ -22,6 +22,7 @@ from .sqltypes import TableValueType
 from .visitors import Traversible
 from .visitors import TraversibleType
 from .. import util
+from ..engine import Dialect
 from ..util import compat
 
 BOOLEANTYPE: Boolean
@@ -71,14 +72,18 @@ class TypeEngine(Traversible, Generic[_T]):
     should_evaluate_none: bool = ...
     def evaluates_none(self: _TE) -> _TE: ...
     def copy(self: _TE, **kw: Any) -> _TE: ...
-    def compare_against_backend(self, dialect: Any, conn_type: Any) -> Any: ...
+    def compare_against_backend(
+        self, dialect: Dialect, conn_type: Any
+    ) -> Any: ...
     def copy_value(self, value: _T) -> _T: ...
     def literal_processor(
-        self, dialect: Any
+        self, dialect: Dialect
     ) -> Optional[_LiteralProcessor[_T]]: ...
-    def bind_processor(self, dialect: Any) -> Optional[_BindProcessor[_T]]: ...
+    def bind_processor(
+        self, dialect: Dialect
+    ) -> Optional[_BindProcessor[_T]]: ...
     def result_processor(
-        self, dialect: Any, coltype: Any
+        self, dialect: Dialect, coltype: Any
     ) -> Optional[_ResultProcessor[_T]]: ...
     def column_expression(self, colexpr: Any) -> Any: ...
     def bind_expression(self, bindvalue: Any) -> Any: ...
@@ -90,12 +95,12 @@ class TypeEngine(Traversible, Generic[_T]):
         self, type_: Type[TypeEngine[_U]], dialect_name: str
     ) -> Variant[_U]: ...
     def as_generic(self, allow_nulltype: bool = ...) -> TypeEngine[Any]: ...
-    def dialect_impl(self, dialect: Any) -> Type[Any]: ...
+    def dialect_impl(self, dialect: Dialect) -> Type[Any]: ...
     def adapt(self, __cls: Type[_U], **kw: Any) -> _U: ...
     def coerce_compared_value(
         self, op: Any, value: Any
     ) -> TypeEngine[Any]: ...
-    def compile(self, dialect: Optional[Any] = ...) -> Any: ...
+    def compile(self, dialect: Optional[Dialect] = ...) -> Any: ...
 
 class VisitableCheckKWArg(util.EnsureKWArgType, TraversibleType): ...
 
@@ -130,20 +135,22 @@ class TypeDecorator(SchemaEventTarget, TypeEngine[_T]):
         ) -> ClauseElement: ...
     @property
     def comparator_factory(self) -> Type[Comparator[TypeEngine[_T]]]: ...  # type: ignore[override]
-    def type_engine(self, dialect: Any) -> TypeEngine[Any]: ...
-    def load_dialect_impl(self, dialect: Any) -> TypeEngine[Any]: ...
+    def type_engine(self, dialect: Dialect) -> TypeEngine[Any]: ...
+    def load_dialect_impl(self, dialect: Dialect) -> TypeEngine[Any]: ...
     def __getattr__(self, key: Any) -> Any: ...
     def process_literal_param(
-        self, value: Optional[_T], dialect: Any
+        self, value: Optional[_T], dialect: Dialect
     ) -> str: ...
-    def process_bind_param(self, value: Optional[_T], dialect: Any) -> Any: ...
+    def process_bind_param(
+        self, value: Optional[_T], dialect: Dialect
+    ) -> Any: ...
     def process_result_value(
-        self, value: Any, dialect: Any
+        self, value: Any, dialect: Dialect
     ) -> Optional[_T]: ...
-    def literal_processor(self, dialect: Any) -> _LiteralProcessor[_T]: ...
-    def bind_processor(self, dialect: Any) -> _BindProcessor[_T]: ...
+    def literal_processor(self, dialect: Dialect) -> _LiteralProcessor[_T]: ...
+    def bind_processor(self, dialect: Dialect) -> _BindProcessor[_T]: ...
     def result_processor(
-        self, dialect: Any, coltype: Any
+        self, dialect: Dialect, coltype: Any
     ) -> _ResultProcessor[_T]: ...
     def bind_expression(self, bindparam: Any) -> Any: ...
     def column_expression(self, column: Any) -> Any: ...
@@ -163,7 +170,7 @@ class Variant(TypeDecorator[_T]):
     def coerce_compared_value(
         self: _VT, operator: Any, value: Any
     ) -> Union[_VT, TypeEngine[Any]]: ...
-    def load_dialect_impl(self, dialect: Any) -> TypeEngine[Any]: ...
+    def load_dialect_impl(self, dialect: Dialect) -> TypeEngine[Any]: ...
     def with_variant(
         self, type_: Type[TypeEngine[_U]], dialect_name: str
     ) -> Variant[_U]: ...