chore: Use types-pika to avoid a typing error

open-contracting · Jan 18, 2022 · bd1e1d8 · bd1e1d8
1 parent aa32949
commit bd1e1d8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -29,8 +29,8 @@
         ],
         "types": [
             "mypy",
-            "pika-stubs",
             "types-orjson",
+            "types-pika",
         ],
         "docs": [
             "furo",

diff --git a/yapw/clients.py b/yapw/clients.py
@@ -52,6 +52,7 @@ class Client(clients.Threaded, clients.Durable, clients.Blocking, clients.Base):
 from typing import Any, Callable, List, Optional, Set, Tuple
 
 import pika
+from pika.exchange_type import ExchangeType
 
 from yapw.decorators import default_decode, halt
 from yapw.ossignal import install_signal_handlers, signal_names
@@ -145,7 +146,7 @@ def __init__(
         super().__init__(**kwargs)
 
         parameters = pika.URLParameters(url)
-        parameters.blocked_connection_timeout = blocked_connection_timeout  # type: ignore # python/mypy#3186
+        parameters.blocked_connection_timeout = blocked_connection_timeout
 
         #: The connection.
         self.connection = pika.BlockingConnection(parameters)  # type: pika.BlockingConnection
@@ -179,7 +180,7 @@ def __init__(
         self,
         *,
         exchange: str = "",
-        exchange_type: str = "direct",
+        exchange_type: ExchangeType = ExchangeType.direct,
         encode: Encode = default_encode,
         content_type: str = "application/json",
         routing_key_template: str = "{exchange}_{routing_key}",