diff --git a/setup.py b/setup.py index d49be9f..6fc314d 100644 --- 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 index 18caacd..38be320 100644 --- 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}",