From c1e29783fbf9c79eb621f9a17ef90afba82d0d43 Mon Sep 17 00:00:00 2001 From: Alex Willmer Date: Wed, 14 Feb 2018 23:27:54 +0000 Subject: [PATCH] Fix AttributeError in mitogen.core.Context.send_await() As of adc8fe3aed470de55a6e5ea3d7c6a31f7c6d1ff1 Receiver objects do not have a get_data() method and Receiver.get() does not unpickle the message. --- mitogen/core.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/mitogen/core.py b/mitogen/core.py index ca796ab2..700a2fb1 100644 --- a/mitogen/core.py +++ b/mitogen/core.py @@ -772,9 +772,10 @@ class Context(object): def send_await(self, msg, deadline=None): """Send `msg` and wait for a response with an optional timeout.""" receiver = self.send_async(msg) - response = receiver.get_data(deadline) - IOLOG.debug('%r._send_await() -> %r', self, response) - return response + response = receiver.get(deadline) + data = response.unpickle() + IOLOG.debug('%r._send_await() -> %r', self, data) + return data def __repr__(self): return 'Context(%s, %r)' % (self.context_id, self.name)