pep/slixmpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix custom stanza examples

Browse source
This commit is contained in:
mathieui 2017-02-14 01:00:41 +01:00
parent 5bf69dca76
commit cc4522d9cd
No known key found for this signature in database
GPG key ID: C59F84CEEFD616E3
2 changed files with 17 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
examples/custom_stanzas/custom_stanza_provider.py
View file

@ -50,7 +50,7 @@ class ActionBot(slixmpp.ClientXMPP):
register_stanza_plugin(Iq, Action)
def start(self, event):
async def start(self, event):
"""
Process the session_start event.
@ -73,7 +73,7 @@ class ActionBot(slixmpp.ClientXMPP):
"""
self.event('custom_action', iq)
def _handle_action_event(self, iq):
async def _handle_action_event(self, iq):
"""
Respond to the custom action event.
"""
@ -82,17 +82,20 @@ class ActionBot(slixmpp.ClientXMPP):
if method == 'is_prime' and param == '2':
print("got message: %s" % iq)
iq.reply()
iq['action']['status'] = 'done'
iq.send()
rep = iq.reply()
rep['action']['status'] = 'done'
await rep.send()
elif method == 'bye':
print("got message: %s" % iq)
rep = iq.reply()
rep['action']['status'] = 'done'
await rep.send()
self.disconnect()
else:
print("got message: %s" % iq)
iq.reply()
iq['action']['status'] = 'error'
iq.send()
rep = iq.reply()
rep['action']['status'] = 'error'
await rep.send()
if __name__ == '__main__':
# Setup the command line arguments.

12
examples/custom_stanzas/custom_stanza_user.py
View file

@ -43,7 +43,7 @@ class ActionUserBot(slixmpp.ClientXMPP):
register_stanza_plugin(Iq, Action)
def start(self, event):
async def start(self, event):
"""
Process the session_start event.
@ -57,11 +57,11 @@ class ActionUserBot(slixmpp.ClientXMPP):
data.
"""
self.send_presence()
self.get_roster()
await self.get_roster()
self.send_custom_iq()
await self.send_custom_iq()
def send_custom_iq(self):
async def send_custom_iq(self):
"""Create and send two custom actions.
If the first action was successful, then send
@ -74,14 +74,14 @@ class ActionUserBot(slixmpp.ClientXMPP):
iq['action']['param'] = '2'
try:
resp = iq.send()
resp = await iq.send()
if resp['action']['status'] == 'done':
#sending bye
iq2 = self.Iq()
iq2['to'] = self.action_provider
iq2['type'] = 'set'
iq2['action']['method'] = 'bye'
iq2.send(block=False)
await iq2.send()
self.disconnect()
except XMPPError: