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

Add back stanza-specific exception handlers

Browse source 
(fixes the test suite too)
This commit is contained in:
mathieui 2015-02-23 17:43:35 +01:00
parent 11fbaa4241
commit 74b4ea20bf
No known key found for this signature in database
GPG key ID: C59F84CEEFD616E3
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
slixmpp/xmlstream/xmlstream.py
View file

@ -706,6 +706,7 @@ class XMLStream(asyncio.BaseProtocol):
handlers = self.__event_handlers.get(name, [])
for handler in handlers:
handler_callback, disposable = handler
old_exception = getattr(data, 'exception', None)
# If the callback is a coroutine, schedule it instead of
# running it directly
@ -715,13 +716,19 @@ class XMLStream(asyncio.BaseProtocol):
try:
yield from cb(data)
except Exception as e:
self.exception(e)
if old_exception:
old_exception(e)
else:
self.exception(e)
asyncio.async(handler_callback_routine(handler_callback))
else:
try:
handler_callback(data)
except Exception as e:
self.exception(e)
if old_exception:
old_exception(e)
else:
self.exception(e)
if disposable:
# If the handler is disposable, we will go ahead and
# remove it now instead of waiting for it to be