diff options
| -rw-r--r-- | slixmpp/stanza/iq.py | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/slixmpp/stanza/iq.py b/slixmpp/stanza/iq.py index af7e3109..385bbbd9 100644 --- a/slixmpp/stanza/iq.py +++ b/slixmpp/stanza/iq.py @@ -188,10 +188,19 @@ class Iq(RootStanza): future = asyncio.Future() def callback_success(result): - if result['type'] == 'error': + type_ = result['type'] + if type_ == 'result': + future.set_result(result) + elif type_ == 'error': future.set_exception(IqError(result)) else: - future.set_result(result) + # Most likely an iq addressed to ourself, rearm the callback. + handler = constr(handler_name, + matcher, + callback_success, + once=True) + self.stream.register_handler(handler) + return if timeout is not None: self.stream.cancel_schedule('IqTimeout_%s' % self['id']) |