jshiffer/slixmpp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

XEP-0049: wrap functions with coroutine_wrapper

Browse Source
This commit is contained in:
mathieui 2015-02-23 19:29:11 +01:00
parent 6408c5a747
commit e68135f59f
No known key found for this signature in database
GPG Key ID: C59F84CEEFD616E3
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
slixmpp/plugins/xep_0049/private_storage.py
View File

@ -9,6 +9,7 @@
import logging import logging
from slixmpp import Iq from slixmpp import Iq
from slixmpp import coroutine_wrapper
from slixmpp.plugins import BasePlugin from slixmpp.plugins import BasePlugin
from slixmpp.xmlstream.handler import Callback from slixmpp.xmlstream.handler import Callback
from slixmpp.xmlstream.matcher import StanzaPath from slixmpp.xmlstream.matcher import StanzaPath
@ -32,8 +33,9 @@ class XEP_0049(BasePlugin):
def register(self, stanza): def register(self, stanza):
register_stanza_plugin(PrivateXML, stanza, iterable=True) register_stanza_plugin(PrivateXML, stanza, iterable=True)
@coroutine_wrapper
def store(self, data, ifrom=None, timeout=None, callback=None, def store(self, data, ifrom=None, timeout=None, callback=None,
timeout_callback=None): timeout_callback=None, coroutine=False):
iq = self.xmpp.Iq() iq = self.xmpp.Iq()
iq['type'] = 'set' iq['type'] = 'set'
iq['from'] = ifrom iq['from'] = ifrom
@ -44,14 +46,15 @@ class XEP_0049(BasePlugin):
for elem in data: for elem in data:
iq['private'].append(elem) iq['private'].append(elem)
return iq.send(timeout=timeout, callback=callback, return iq.send(timeout=timeout, callback=callback, coroutine=coroutine,
timeout_callback=timeout_callback) timeout_callback=timeout_callback)
@coroutine_wrapper
def retrieve(self, name, ifrom=None, timeout=None, callback=None, def retrieve(self, name, ifrom=None, timeout=None, callback=None,
timeout_callback=None): timeout_callback=None, coroutine=False):
iq = self.xmpp.Iq() iq = self.xmpp.Iq()
iq['type'] = 'get' iq['type'] = 'get'
iq['from'] = ifrom iq['from'] = ifrom
iq['private'].enable(name) iq['private'].enable(name)
return iq.send(timeout=timeout, callback=callback, return iq.send(timeout=timeout, callback=callback, coroutine=coroutine,
timeout_callback=timeout_callback) timeout_callback=timeout_callback)