From 1e4944d47e8296fdaa792a8b3fc87ea99acc217c Mon Sep 17 00:00:00 2001
From: mathieui <mathieui@mathieui.net>
Date: Mon, 23 Feb 2015 19:29:54 +0100
Subject: XEP-0077: wrap functions with coroutine_wrapper

---
 slixmpp/plugins/xep_0077/register.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

(limited to 'slixmpp')

diff --git a/slixmpp/plugins/xep_0077/register.py b/slixmpp/plugins/xep_0077/register.py
index eb2e7443..b4beac84 100644
--- a/slixmpp/plugins/xep_0077/register.py
+++ b/slixmpp/plugins/xep_0077/register.py
@@ -9,6 +9,7 @@
 import logging
 import ssl
 
+from slixmpp import coroutine_wrapper
 from slixmpp.stanza import StreamFeatures, Iq
 from slixmpp.xmlstream import register_stanza_plugin, JID
 from slixmpp.plugins import BasePlugin
@@ -81,26 +82,29 @@ class XEP_0077(BasePlugin):
             return True
         return False
 
+    @coroutine_wrapper
     def get_registration(self, jid=None, ifrom=None,
-                         timeout=None, callback=None):
+                         timeout=None, callback=None, coroutine=False):
         iq = self.xmpp.Iq()
         iq['type'] = 'get'
         iq['to'] = jid
         iq['from'] = ifrom
         iq.enable('register')
-        return iq.send(timeout=timeout, callback=callback)
+        return iq.send(timeout=timeout, callback=callback, coroutine=coroutine)
 
+    @coroutine_wrapper
     def cancel_registration(self, jid=None, ifrom=None,
-                            timeout=None, callback=None):
+                            timeout=None, callback=None, coroutine=False):
         iq = self.xmpp.Iq()
         iq['type'] = 'set'
         iq['to'] = jid
         iq['from'] = ifrom
         iq['register']['remove'] = True
-        return iq.send(timeout=timeout, callback=callback)
+        return iq.send(timeout=timeout, callback=callback, coroutine=coroutine)
 
+    @coroutine_wrapper
     def change_password(self, password, jid=None, ifrom=None,
-                        timeout=None, callback=None):
+                        timeout=None, callback=None, coroutine=False):
         iq = self.xmpp.Iq()
         iq['type'] = 'set'
         iq['to'] = jid
@@ -111,4 +115,4 @@ class XEP_0077(BasePlugin):
         else:
             iq['register']['username'] = self.xmpp.boundjid.user
         iq['register']['password'] = password
-        return iq.send(timeout=timeout, callback=callback)
+        return iq.send(timeout=timeout, callback=callback, coroutine=coroutine)
-- 
cgit v1.2.3