summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--kombu/abstract.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/kombu/abstract.py b/kombu/abstract.py
index 9bafba99..48a917c9 100644
--- a/kombu/abstract.py
+++ b/kombu/abstract.py
@@ -10,6 +10,7 @@ from .exceptions import NotBoundError
from .utils.functional import ChannelPromise
if TYPE_CHECKING:
+ from kombu.connection import Connection
from kombu.transport.virtual import Channel
@@ -80,19 +81,19 @@ class MaybeChannelBound(Object):
can_cache_declaration = False
def __call__(
- self: _MaybeChannelBoundType, channel: Channel
+ self: _MaybeChannelBoundType, channel: (Channel | Connection)
) -> _MaybeChannelBoundType:
"""`self(channel) -> self.bind(channel)`."""
return self.bind(channel)
def bind(
- self: _MaybeChannelBoundType, channel: Channel
+ self: _MaybeChannelBoundType, channel: (Channel | Connection)
) -> _MaybeChannelBoundType:
"""Create copy of the instance that is bound to a channel."""
return copy(self).maybe_bind(channel)
def maybe_bind(
- self: _MaybeChannelBoundType, channel: Channel
+ self: _MaybeChannelBoundType, channel: (Channel | Connection)
) -> _MaybeChannelBoundType:
"""Bind instance to channel if not already bound."""
if not self.is_bound and channel: