From 1e4944d47e8296fdaa792a8b3fc87ea99acc217c Mon Sep 17 00:00:00 2001 From: mathieui Date: Mon, 23 Feb 2015 19:29:54 +0100 Subject: [PATCH] XEP-0077: wrap functions with coroutine_wrapper --- slixmpp/plugins/xep_0077/register.py | 16 ++++++++++------ 1 file 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)