jshiffer/slixmpp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rearm an iq callback if it was addressed to ourself.

Browse Source
This commit is contained in:
Emmanuel Gil Peyrot 2018-03-14 17:37:55 +01:00
parent 2f1225bad3
commit 5184713356
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
slixmpp/stanza/iq.py
View File

@ -188,10 +188,19 @@ class Iq(RootStanza):
future = asyncio.Future() future = asyncio.Future()
def callback_success(result): 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)) future.set_exception(IqError(result))
else: 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: if timeout is not None:
self.stream.cancel_schedule('IqTimeout_%s' % self['id']) self.stream.cancel_schedule('IqTimeout_%s' % self['id'])