lint: fix mypy

airtai · Nov 25, 2024 · 64c22b4 · 64c22b4
1 parent a67768c
commit 64c22b4
Show file tree

Hide file tree

Showing 26 changed files with 115 additions and 98 deletions.
diff --git a/faststream/broker/core/abc.py b/faststream/broker/core/abc.py
@@ -6,6 +6,7 @@
     Iterable,
     Mapping,
     Optional,
+    Sequence,
 )
 
 from faststream.broker.types import MsgType
@@ -30,7 +31,7 @@ def __init__(
         *,
         prefix: str,
         dependencies: Iterable["Depends"],
-        middlewares: Iterable["BrokerMiddleware[MsgType]"],
+        middlewares: Sequence["BrokerMiddleware[MsgType]"],
         parser: Optional["CustomCallable"],
         decoder: Optional["CustomCallable"],
         include_in_schema: Optional[bool],

diff --git a/faststream/broker/core/usecase.py b/faststream/broker/core/usecase.py
@@ -75,7 +75,7 @@ def __init__(
             Doc("Dependencies to apply to all broker subscribers."),
         ],
         middlewares: Annotated[
-            Iterable["BrokerMiddleware[MsgType]"],
+            Sequence["BrokerMiddleware[MsgType]"],
             Doc("Middlewares to apply to all broker publishers/subscribers."),
         ],
         graceful_timeout: Annotated[

diff --git a/faststream/broker/router.py b/faststream/broker/router.py
@@ -4,6 +4,7 @@
     Callable,
     Iterable,
     Optional,
+    Sequence,
 )
 
 from faststream.broker.core.abc import ABCBroker
@@ -64,7 +65,7 @@ def __init__(
         # base options
         prefix: str,
         dependencies: Iterable["Depends"],
-        middlewares: Iterable["BrokerMiddleware[MsgType]"],
+        middlewares: Sequence["BrokerMiddleware[MsgType]"],
         parser: Optional["CustomCallable"],
         decoder: Optional["CustomCallable"],
         include_in_schema: Optional[bool],

diff --git a/faststream/broker/subscriber/call_item.py b/faststream/broker/subscriber/call_item.py
@@ -9,6 +9,7 @@
     Generic,
     Iterable,
     Optional,
+    Sequence,
     cast,
 )
 
@@ -54,7 +55,7 @@ def __init__(
         filter: "AsyncFilter[StreamMessage[MsgType]]",
         item_parser: Optional["CustomCallable"],
         item_decoder: Optional["CustomCallable"],
-        item_middlewares: Iterable["SubscriberMiddleware[StreamMessage[MsgType]]"],
+        item_middlewares: Sequence["SubscriberMiddleware[StreamMessage[MsgType]]"],
         dependencies: Iterable["Depends"],
     ) -> None:
         self.handler = handler

diff --git a/faststream/broker/subscriber/proto.py b/faststream/broker/subscriber/proto.py
@@ -1,5 +1,14 @@
 from abc import abstractmethod
-from typing import TYPE_CHECKING, Any, Callable, Dict, Iterable, List, Optional
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Callable,
+    Dict,
+    Iterable,
+    List,
+    Optional,
+    Sequence,
+)
 
 from typing_extensions import Self, override
 
@@ -33,7 +42,7 @@ class SubscriberProto(
     running: bool
 
     _broker_dependencies: Iterable["Depends"]
-    _broker_middlewares: Iterable["BrokerMiddleware[MsgType]"]
+    _broker_middlewares: Sequence["BrokerMiddleware[MsgType]"]
     _producer: Optional["ProducerProto"]
 
     @abstractmethod
@@ -98,6 +107,6 @@ def add_call(
         filter_: "Filter[Any]",
         parser_: "CustomCallable",
         decoder_: "CustomCallable",
-        middlewares_: Iterable["SubscriberMiddleware[Any]"],
+        middlewares_: Sequence["SubscriberMiddleware[Any]"],
         dependencies_: Iterable["Depends"],
     ) -> Self: ...
diff --git a/faststream/broker/subscriber/usecase.py b/faststream/broker/subscriber/usecase.py
@@ -10,6 +10,7 @@
     Iterable,
     List,
     Optional,
+    Sequence,
     Tuple,
     Union,
     overload,
@@ -66,7 +67,7 @@ def __init__(
         filter: "Filter[Any]",
         parser: Optional["CustomCallable"],
         decoder: Optional["CustomCallable"],
-        middlewares: Iterable["SubscriberMiddleware[Any]"],
+        middlewares: Sequence["SubscriberMiddleware[Any]"],
         dependencies: Iterable["Depends"],
     ) -> None:
         self.filter = filter
@@ -98,7 +99,7 @@ def __init__(
         no_reply: bool,
         retry: Union[bool, int],
         broker_dependencies: Iterable["Depends"],
-        broker_middlewares: Iterable["BrokerMiddleware[MsgType]"],
+        broker_middlewares: Sequence["BrokerMiddleware[MsgType]"],
         default_parser: "AsyncCallable",
         default_decoder: "AsyncCallable",
         # AsyncAPI information
@@ -211,7 +212,7 @@ def add_call(
         filter_: "Filter[Any]",
         parser_: Optional["CustomCallable"],
         decoder_: Optional["CustomCallable"],
-        middlewares_: Iterable["SubscriberMiddleware[Any]"],
+        middlewares_: Sequence["SubscriberMiddleware[Any]"],
         dependencies_: Iterable["Depends"],
     ) -> Self:
         self._call_options = _CallOptions(
@@ -231,7 +232,7 @@ def __call__(
         filter: Optional["Filter[Any]"] = None,
         parser: Optional["CustomCallable"] = None,
         decoder: Optional["CustomCallable"] = None,
-        middlewares: Iterable["SubscriberMiddleware[Any]"] = (),
+        middlewares: Sequence["SubscriberMiddleware[Any]"] = (),
         dependencies: Iterable["Depends"] = (),
     ) -> Callable[
         [Callable[P_HandlerParams, T_HandlerReturn]],
@@ -246,7 +247,7 @@ def __call__(
         filter: Optional["Filter[Any]"] = None,
         parser: Optional["CustomCallable"] = None,
         decoder: Optional["CustomCallable"] = None,
-        middlewares: Iterable["SubscriberMiddleware[Any]"] = (),
+        middlewares: Sequence["SubscriberMiddleware[Any]"] = (),
         dependencies: Iterable["Depends"] = (),
     ) -> "HandlerCallWrapper[MsgType, P_HandlerParams, T_HandlerReturn]": ...
 
@@ -257,7 +258,7 @@ def __call__(
         filter: Optional["Filter[Any]"] = None,
         parser: Optional["CustomCallable"] = None,
         decoder: Optional["CustomCallable"] = None,
-        middlewares: Iterable["SubscriberMiddleware[Any]"] = (),
+        middlewares: Sequence["SubscriberMiddleware[Any]"] = (),
         dependencies: Iterable["Depends"] = (),
     ) -> Any:
         if (options := self._call_options) is None:

diff --git a/faststream/broker/wrapper/proto.py b/faststream/broker/wrapper/proto.py
@@ -5,6 +5,7 @@
     Iterable,
     Optional,
     Protocol,
+    Sequence,
     Union,
     overload,
 )
@@ -35,7 +36,7 @@ def __call__(
         filter: Optional["Filter[Any]"] = None,
         parser: Optional["CustomCallable"] = None,
         decoder: Optional["CustomCallable"] = None,
-        middlewares: Iterable["SubscriberMiddleware[Any]"] = (),
+        middlewares: Sequence["SubscriberMiddleware[Any]"] = (),
         dependencies: Iterable["Depends"] = (),
     ) -> Callable[
         [Callable[P_HandlerParams, T_HandlerReturn]],
@@ -53,7 +54,7 @@ def __call__(
         filter: Optional["Filter[Any]"] = None,
         parser: Optional["CustomCallable"] = None,
         decoder: Optional["CustomCallable"] = None,
-        middlewares: Iterable["SubscriberMiddleware[Any]"] = (),
+        middlewares: Sequence["SubscriberMiddleware[Any]"] = (),
         dependencies: Iterable["Depends"] = (),
     ) -> "HandlerCallWrapper[MsgType, P_HandlerParams, T_HandlerReturn]": ...
 
@@ -68,7 +69,7 @@ def __call__(
         filter: Optional["Filter[Any]"] = None,
         parser: Optional["CustomCallable"] = None,
         decoder: Optional["CustomCallable"] = None,
-        middlewares: Iterable["SubscriberMiddleware[Any]"] = (),
+        middlewares: Sequence["SubscriberMiddleware[Any]"] = (),
         dependencies: Iterable["Depends"] = (),
     ) -> Union[
         "HandlerCallWrapper[MsgType, P_HandlerParams, T_HandlerReturn]",

diff --git a/faststream/confluent/broker/registrator.py b/faststream/confluent/broker/registrator.py
@@ -295,7 +295,7 @@ def subscriber(
             Doc("Function to decode FastStream msg bytes body to python objects."),
         ] = None,
         middlewares: Annotated[
-            Iterable["SubscriberMiddleware[KafkaMessage]"],
+            Sequence["SubscriberMiddleware[KafkaMessage]"],
             Doc("Subscriber middlewares to wrap incoming message processing."),
         ] = (),
         filter: Annotated[
@@ -577,7 +577,7 @@ def subscriber(
             Doc("Function to decode FastStream msg bytes body to python objects."),
         ] = None,
         middlewares: Annotated[
-            Iterable["SubscriberMiddleware[KafkaMessage]"],
+            Sequence["SubscriberMiddleware[KafkaMessage]"],
             Doc("Subscriber middlewares to wrap incoming message processing."),
         ] = (),
         filter: Annotated[
@@ -859,7 +859,7 @@ def subscriber(
             Doc("Function to decode FastStream msg bytes body to python objects."),
         ] = None,
         middlewares: Annotated[
-            Iterable["SubscriberMiddleware[KafkaMessage]"],
+            Sequence["SubscriberMiddleware[KafkaMessage]"],
             Doc("Subscriber middlewares to wrap incoming message processing."),
         ] = (),
         filter: Annotated[
@@ -1144,7 +1144,7 @@ def subscriber(
             Doc("Function to decode FastStream msg bytes body to python objects."),
         ] = None,
         middlewares: Annotated[
-            Iterable["SubscriberMiddleware[KafkaMessage]"],
+            Sequence["SubscriberMiddleware[KafkaMessage]"],
             Doc("Subscriber middlewares to wrap incoming message processing."),
         ] = (),
         filter: Annotated[

diff --git a/faststream/confluent/router.py b/faststream/confluent/router.py
@@ -88,7 +88,7 @@ def __init__(
         ] = False,
         # basic args
         middlewares: Annotated[
-            Iterable["PublisherMiddleware"],
+            Sequence["PublisherMiddleware"],
             Doc("Publisher middlewares to wrap outgoing messages."),
         ] = (),
         # AsyncAPI args
@@ -378,7 +378,7 @@ def __init__(
             Doc("Function to decode FastStream msg bytes body to python objects."),
         ] = None,
         middlewares: Annotated[
-            Iterable["SubscriberMiddleware[KafkaMessage]"],
+            Sequence["SubscriberMiddleware[KafkaMessage]"],
             Doc("Subscriber middlewares to wrap incoming message processing."),
         ] = (),
         filter: Annotated[
@@ -492,7 +492,7 @@ def __init__(
             ),
         ] = (),
         middlewares: Annotated[
-            Iterable[
+            Sequence[
                 Union[
                     "BrokerMiddleware[Message]",
                     "BrokerMiddleware[Tuple[Message, ...]]",

diff --git a/faststream/confluent/subscriber/factory.py b/faststream/confluent/subscriber/factory.py
@@ -40,7 +40,7 @@ def create_subscriber(
     no_reply: bool,
     retry: bool,
     broker_dependencies: Iterable["Depends"],
-    broker_middlewares: Iterable["BrokerMiddleware[Tuple[ConfluentMsg, ...]]"],
+    broker_middlewares: Sequence["BrokerMiddleware[Tuple[ConfluentMsg, ...]]"],
     # AsyncAPI args
     title_: Optional[str],
     description_: Optional[str],
@@ -64,7 +64,7 @@ def create_subscriber(
     no_reply: bool,
     retry: bool,
     broker_dependencies: Iterable["Depends"],
-    broker_middlewares: Iterable["BrokerMiddleware[ConfluentMsg]"],
+    broker_middlewares: Sequence["BrokerMiddleware[ConfluentMsg]"],
     # AsyncAPI args
     title_: Optional[str],
     description_: Optional[str],
@@ -89,8 +89,8 @@ def create_subscriber(
     retry: bool,
     broker_dependencies: Iterable["Depends"],
     broker_middlewares: Union[
-        Iterable["BrokerMiddleware[Tuple[ConfluentMsg, ...]]"],
-        Iterable["BrokerMiddleware[ConfluentMsg]"],
+        Sequence["BrokerMiddleware[Tuple[ConfluentMsg, ...]]"],
+        Sequence["BrokerMiddleware[ConfluentMsg]"],
     ],
     # AsyncAPI args
     title_: Optional[str],
@@ -118,8 +118,8 @@ def create_subscriber(
     retry: bool,
     broker_dependencies: Iterable["Depends"],
     broker_middlewares: Union[
-        Iterable["BrokerMiddleware[Tuple[ConfluentMsg, ...]]"],
-        Iterable["BrokerMiddleware[ConfluentMsg]"],
+        Sequence["BrokerMiddleware[Tuple[ConfluentMsg, ...]]"],
+        Sequence["BrokerMiddleware[ConfluentMsg]"],
     ],
     # AsyncAPI args
     title_: Optional[str],
@@ -143,7 +143,7 @@ def create_subscriber(
             retry=retry,
             broker_dependencies=broker_dependencies,
             broker_middlewares=cast(
-                Iterable["BrokerMiddleware[Tuple[ConfluentMsg, ...]]"],
+                Sequence["BrokerMiddleware[Tuple[ConfluentMsg, ...]]"],
                 broker_middlewares,
             ),
             title_=title_,
@@ -163,7 +163,8 @@ def create_subscriber(
             retry=retry,
             broker_dependencies=broker_dependencies,
             broker_middlewares=cast(
-                Iterable["BrokerMiddleware[ConfluentMsg]"], broker_middlewares
+                Sequence["BrokerMiddleware[ConfluentMsg]"],
+                broker_middlewares,
             ),
             title_=title_,
             description_=description_,

diff --git a/faststream/confluent/subscriber/usecase.py b/faststream/confluent/subscriber/usecase.py
@@ -65,7 +65,7 @@ def __init__(
         no_reply: bool,
         retry: bool,
         broker_dependencies: Iterable["Depends"],
-        broker_middlewares: Iterable["BrokerMiddleware[MsgType]"],
+        broker_middlewares: Sequence["BrokerMiddleware[MsgType]"],
         # AsyncAPI args
         title_: Optional[str],
         description_: Optional[str],
@@ -286,7 +286,7 @@ def __init__(
         no_reply: bool,
         retry: bool,
         broker_dependencies: Iterable["Depends"],
-        broker_middlewares: Iterable["BrokerMiddleware[Message]"],
+        broker_middlewares: Sequence["BrokerMiddleware[Message]"],
         # AsyncAPI args
         title_: Optional[str],
         description_: Optional[str],
@@ -350,7 +350,7 @@ def __init__(
         no_reply: bool,
         retry: bool,
         broker_dependencies: Iterable["Depends"],
-        broker_middlewares: Iterable["BrokerMiddleware[Tuple[Message, ...]]"],
+        broker_middlewares: Sequence["BrokerMiddleware[Tuple[Message, ...]]"],
         # AsyncAPI args
         title_: Optional[str],
         description_: Optional[str],

diff --git a/faststream/kafka/broker/registrator.py b/faststream/kafka/broker/registrator.py
@@ -404,7 +404,7 @@ def subscriber(
             Doc("Function to decode FastStream msg bytes body to python objects."),
         ] = None,
         middlewares: Annotated[
-            Iterable["SubscriberMiddleware[KafkaMessage]"],
+            Sequence["SubscriberMiddleware[KafkaMessage]"],
             Doc("Subscriber middlewares to wrap incoming message processing."),
         ] = (),
         filter: Annotated[
@@ -785,7 +785,7 @@ def subscriber(
             Doc("Function to decode FastStream msg bytes body to python objects."),
         ] = None,
         middlewares: Annotated[
-            Iterable["SubscriberMiddleware[KafkaMessage]"],
+            Sequence["SubscriberMiddleware[KafkaMessage]"],
             Doc("Subscriber middlewares to wrap incoming message processing."),
         ] = (),
         filter: Annotated[
@@ -1166,7 +1166,7 @@ def subscriber(
             Doc("Function to decode FastStream msg bytes body to python objects."),
         ] = None,
         middlewares: Annotated[
-            Iterable["SubscriberMiddleware[KafkaMessage]"],
+            Sequence["SubscriberMiddleware[KafkaMessage]"],
             Doc("Subscriber middlewares to wrap incoming message processing."),
         ] = (),
         filter: Annotated[
@@ -1550,7 +1550,7 @@ def subscriber(
             Doc("Function to decode FastStream msg bytes body to python objects."),
         ] = None,
         middlewares: Annotated[
-            Iterable["SubscriberMiddleware[KafkaMessage]"],
+            Sequence["SubscriberMiddleware[KafkaMessage]"],
             Doc("Subscriber middlewares to wrap incoming message processing."),
         ] = (),
         max_workers: Annotated[

diff --git a/faststream/kafka/router.py b/faststream/kafka/router.py
@@ -90,7 +90,7 @@ def __init__(
         ] = False,
         # basic args
         middlewares: Annotated[
-            Iterable["PublisherMiddleware"],
+            Sequence["PublisherMiddleware"],
             Doc("Publisher middlewares to wrap outgoing messages."),
         ] = (),
         # AsyncAPI args
@@ -607,7 +607,7 @@ def __init__(
             ),
         ] = (),
         middlewares: Annotated[
-            Iterable[
+            Sequence[
                 Union[
                     "BrokerMiddleware[ConsumerRecord]",
                     "BrokerMiddleware[Tuple[ConsumerRecord, ...]]",