From e68135f59f9a224688679eb91e8063041d6f000b Mon Sep 17 00:00:00 2001 From: mathieui Date: Mon, 23 Feb 2015 19:29:11 +0100 Subject: XEP-0049: wrap functions with coroutine_wrapper --- slixmpp/plugins/xep_0049/private_storage.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) (limited to 'slixmpp/plugins/xep_0049') diff --git a/slixmpp/plugins/xep_0049/private_storage.py b/slixmpp/plugins/xep_0049/private_storage.py index a66c05d1..32309069 100644 --- a/slixmpp/plugins/xep_0049/private_storage.py +++ b/slixmpp/plugins/xep_0049/private_storage.py @@ -9,6 +9,7 @@ import logging from slixmpp import Iq +from slixmpp import coroutine_wrapper from slixmpp.plugins import BasePlugin from slixmpp.xmlstream.handler import Callback from slixmpp.xmlstream.matcher import StanzaPath @@ -32,8 +33,9 @@ class XEP_0049(BasePlugin): def register(self, stanza): register_stanza_plugin(PrivateXML, stanza, iterable=True) + @coroutine_wrapper def store(self, data, ifrom=None, timeout=None, callback=None, - timeout_callback=None): + timeout_callback=None, coroutine=False): iq = self.xmpp.Iq() iq['type'] = 'set' iq['from'] = ifrom @@ -44,14 +46,15 @@ class XEP_0049(BasePlugin): for elem in data: iq['private'].append(elem) - return iq.send(timeout=timeout, callback=callback, + return iq.send(timeout=timeout, callback=callback, coroutine=coroutine, timeout_callback=timeout_callback) + @coroutine_wrapper def retrieve(self, name, ifrom=None, timeout=None, callback=None, - timeout_callback=None): + timeout_callback=None, coroutine=False): iq = self.xmpp.Iq() iq['type'] = 'get' iq['from'] = ifrom iq['private'].enable(name) - return iq.send(timeout=timeout, callback=callback, - timeout_callback=timeout_callback) + return iq.send(timeout=timeout, callback=callback, coroutine=coroutine, + timeout_callback=timeout_callback) -- cgit v1.2.3