summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--slixmpp/stanza/iq.py59
1 files changed, 59 insertions, 0 deletions
diff --git a/slixmpp/stanza/iq.py b/slixmpp/stanza/iq.py
index c9f29f17..e2cef50d 100644
--- a/slixmpp/stanza/iq.py
+++ b/slixmpp/stanza/iq.py
@@ -9,6 +9,7 @@
from slixmpp.stanza.rootstanza import RootStanza
from slixmpp.xmlstream import StanzaBase, ET
from slixmpp.xmlstream.handler import Waiter, Callback
+from slixmpp.xmlstream.asyncio import asyncio
from slixmpp.xmlstream.matcher import MatchIDSender, MatcherId
from slixmpp.exceptions import IqTimeout, IqError
@@ -158,6 +159,64 @@ class Iq(RootStanza):
new_iq['type'] = 'result'
return new_iq
+ @asyncio.coroutine
+ def send_coroutine(self, timeout=None):
+ """Send an <iq> stanza over the XML stream.
+
+ Blocks (with asyncio) until a the reply is received.
+ Use with yield from iq.send_coroutine().
+
+ Overrides StanzaBase.send
+
+ Arguments:
+
+ timeout -- The length of time (in seconds) to wait for a
+ response before an IqTimeout is raised
+ """
+
+ if self.stream.session_bind_event.is_set():
+ matcher = MatchIDSender({
+ 'id': self['id'],
+ 'self': self.stream.boundjid,
+ 'peer': self['to']
+ })
+ else:
+ matcher = MatcherId(self['id'])
+
+ future = asyncio.Future()
+
+ def callback(result):
+ future.set_result(result)
+
+ def callback_timeout():
+ future.set_result(None)
+
+ handler_name = 'IqCallback_%s' % self['id']
+
+ if timeout:
+ self.callback = callback
+ self.stream.schedule('IqTimeout_%s' % self['id'],
+ timeout,
+ callback_timeout,
+ repeat=False)
+ handler = Callback(handler_name,
+ matcher,
+ self._handle_result,
+ once=True)
+ else:
+ handler = Callback(handler_name,
+ matcher,
+ callback,
+ once=True)
+ self.stream.register_handler(handler)
+ StanzaBase.send(self)
+ result = yield from future
+ if result is None:
+ raise IqTimeout(self)
+ if result['type'] == 'error':
+ raise IqError(result)
+ return result
+
def send(self, callback=None, timeout=None, timeout_callback=None):
"""Send an <iq> stanza over the XML stream.