From 92e4bc752a7da814b5b7c19b0f5d2ee95f715f7e Mon Sep 17 00:00:00 2001
From: mathieui <mathieui@mathieui.net>
Date: Sat, 21 Feb 2015 23:45:30 +0100
Subject: =?UTF-8?q?Add=20a=20=E2=80=9Cblocking=E2=80=9D=20send=5Fcoroutine?=
 =?UTF-8?q?=20method=20to=20the=20Iq=20class?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 slixmpp/stanza/iq.py | 59 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)

(limited to 'slixmpp/stanza')

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.
 
-- 
cgit v1.2.3