diff options
| author | mathieui <mathieui@mathieui.net> | 2015-02-23 19:29:54 +0100 |
|---|---|---|
| committer | mathieui <mathieui@mathieui.net> | 2015-02-24 22:46:05 +0100 |
| commit | 1e4944d47e8296fdaa792a8b3fc87ea99acc217c (patch) | |
| tree | 8fa860a18a8b0a5f9a656f4e898faf48bb6202e0 | |
| parent | e68135f59f9a224688679eb91e8063041d6f000b (diff) | |
| download | slixmpp-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.py | 16 |
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) |