summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-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)