diff options
author | Ask Solem <ask@celeryproject.org> | 2013-10-04 14:31:58 +0100 |
---|---|---|
committer | Ask Solem <ask@celeryproject.org> | 2013-10-04 14:31:58 +0100 |
commit | 2f3d9918eaba2cf3068d0ed5a5dfdf8088b81c9e (patch) | |
tree | 12ce338b1d1aa465819af6f3dd787136eda0a120 | |
parent | bf25876b95880e2ebd870367e9ab335da5a46578 (diff) | |
download | py-amqp-2f3d9918eaba2cf3068d0ed5a5dfdf8088b81c9e.tar.gz |
Sets Message.channel on delivery. Closes #12
-rw-r--r-- | amqp/method_framing.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/amqp/method_framing.py b/amqp/method_framing.py index a272213..cee05ba 100644 --- a/amqp/method_framing.py +++ b/amqp/method_framing.py @@ -46,10 +46,10 @@ _CONTENT_METHODS = [ class _PartialMessage(object): """Helper class to build up a multi-frame method.""" - def __init__(self, method_sig, args): + def __init__(self, method_sig, args, channel): self.method_sig = method_sig self.args = args - self.msg = Message() + self.msg = Message(channel=channel) self.body_parts = [] self.body_received = 0 self.body_size = None @@ -147,7 +147,9 @@ class MethodReader(object): # # Save what we've got so far and wait for the content-header # - self.partial_messages[channel] = _PartialMessage(method_sig, args) + self.partial_messages[channel] = _PartialMessage( + method_sig, args, channel, + ) self.expected_types[channel] = 2 else: self._quick_put((channel, method_sig, args, None)) |