OPT: improve typing 🎨

Author: eigenein

combadge/_helpers/pydantic.py

@@ -1,10 +1,11 @@
 from functools import cache
-from typing import Any
 
 from pydantic import TypeAdapter
 
+from combadge.core.typevars import AnyT
+
 
 @cache
-def get_type_adapter(type_: Any) -> TypeAdapter[Any]:
+def get_type_adapter(type_: AnyT) -> TypeAdapter[AnyT]:
     """Get cached type adapter for the given type."""
     return TypeAdapter(type_)

combadge/_helpers/typing.py

@@ -3,14 +3,14 @@
 from typing_extensions import TypeAliasType
 
 
-def unwrap_type_alias(type_: Any) -> Any:
+def unwrap_type_alias(type_: type[Any] | None) -> type[Any] | None:
     """Extract the inner type, if the given type is a type alias."""
     if isinstance(type_, TypeAliasType):
         type_ = type_.__value__
     return type_
 
 
-def unwrap_annotated(type_: Any) -> Any:
+def unwrap_annotated(type_: type[Any] | None) -> type[Any] | None:
     """Extract the inner type, if the given type is an `Annotated` form."""
     if get_origin(type_) is Annotated:
         type_ = get_args(type_)[0]

combadge/core/backend.py

@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import Any
+from typing import cast
 
 from typing_extensions import Self
 
@@ -20,7 +20,7 @@ class BaseBackend(ABC):
     __slots__ = ("_service_cache",)
 
     def __init__(self) -> None:  # noqa: D107
-        self._service_cache: dict[type, Any] = {}
+        self._service_cache: dict[type, object] = {}
 
     @classmethod
     @abstractmethod
@@ -56,7 +56,7 @@ def __getitem__(self, protocol: type[ServiceProtocolT]) -> ServiceProtocolT:
         service = self._service_cache.get(protocol)
         if service is None:
             service = self._service_cache[protocol] = bind(protocol, self)
-        return service  # noqa: RET504
+        return cast(ServiceProtocolT, service)  # noqa: RET504
 
     def __delitem__(self, protocol: type) -> None:
         """

combadge/core/binder.py

@@ -63,7 +63,7 @@ class BoundService(BaseBoundService, from_protocol):  # type: ignore[misc, valid
     return BoundService
 
 
-def _wrap(method: Callable[..., Any], with_markers: Iterable[MethodMarker]) -> Callable[..., Any]:
+def _wrap(method: FunctionT, with_markers: Iterable[MethodMarker]) -> FunctionT:
     """
     Apply method markers.
 
@@ -78,7 +78,7 @@ def _wrap(method: Callable[..., Any], with_markers: Iterable[MethodMarker]) -> C
     return method
 
 
-def _enumerate_methods(of_protocol: type) -> Iterable[tuple[str, Any]]:
+def _enumerate_methods(of_protocol: type) -> Iterable[tuple[str, Callable[..., Any]]]:
     """Enumerate the service protocol methods."""
 
     for name, method in get_members(of_protocol, callable):

combadge/core/signature.py

@@ -42,7 +42,7 @@ class Signature:
     """A callable that binds the method's arguments, it is cached here to improve performance."""
 
     @classmethod
-    def from_method(cls, method: Any) -> Signature:
+    def from_method(cls, method: Callable[..., Any]) -> Signature:
         """Create a signature from the specified method."""
         annotations_ = get_annotations(method, eval_str=True)
         return_type = cls._extract_return_type(annotations_)

combadge/support/soap/abc.py

@@ -1,5 +1,4 @@
 from dataclasses import dataclass
-from typing import Any
 
 from annotated_types import SLOTS
 
@@ -21,5 +20,5 @@ def get_operation_name(self) -> str:
 class SoapHeader:
     """SOAP request header."""
 
-    soap_header: Any | None = None
+    soap_header: object | None = None
     """SOAP header payload."""