summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authormathieui <mathieui@mathieui.net>2015-02-23 19:29:54 +0100
committermathieui <mathieui@mathieui.net>2015-02-24 22:46:05 +0100
commit1e4944d47e8296fdaa792a8b3fc87ea99acc217c (patch)
tree8fa860a18a8b0a5f9a656f4e898faf48bb6202e0
parente68135f59f9a224688679eb91e8063041d6f000b (diff)
downloadslixmpp-1e4944d47e8296fdaa792a8b3fc87ea99acc217c.tar.gz
slixmpp-1e4944d47e8296fdaa792a8b3fc87ea99acc217c.tar.bz2
slixmpp-1e4944d47e8296fdaa792a8b3fc87ea99acc217c.tar.xz
slixmpp-1e4944d47e8296fdaa792a8b3fc87ea99acc217c.zip
XEP-0077: wrap functions with coroutine_wrapper
-rw-r--r--slixmpp/plugins/xep_0077/register.py16
1 files changed, 10 insertions, 6 deletions
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)