0d32638379
This allows exceptions to include the original content of a stanza in the error response by including the parameter clear=False when raising the exception.
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
"""
|
|
SleekXMPP: The Sleek XMPP Library
|
|
Copyright (C) 2010 Nathanael C. Fritz
|
|
This file is part of SleekXMPP.
|
|
|
|
See the file LICENSE for copying permission.
|
|
"""
|
|
|
|
import logging
|
|
import traceback
|
|
import sys
|
|
|
|
from sleekxmpp.exceptions import XMPPError
|
|
from sleekxmpp.stanza import Error
|
|
from sleekxmpp.xmlstream import ET, StanzaBase, register_stanza_plugin
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class RootStanza(StanzaBase):
|
|
|
|
"""
|
|
A top-level XMPP stanza in an XMLStream.
|
|
|
|
The RootStanza class provides a more XMPP specific exception
|
|
handler than provided by the generic StanzaBase class.
|
|
|
|
Methods:
|
|
exception -- Overrides StanzaBase.exception
|
|
"""
|
|
|
|
def exception(self, e):
|
|
"""
|
|
Create and send an error reply.
|
|
|
|
Typically called when an event handler raises an exception.
|
|
The error's type and text content are based on the exception
|
|
object's type and content.
|
|
|
|
Overrides StanzaBase.exception.
|
|
|
|
Arguments:
|
|
e -- Exception object
|
|
"""
|
|
if isinstance(e, XMPPError):
|
|
self.reply(clear=e.clear)
|
|
# We raised this deliberately
|
|
self['error']['condition'] = e.condition
|
|
self['error']['text'] = e.text
|
|
if e.extension is not None:
|
|
# Extended error tag
|
|
extxml = ET.Element("{%s}%s" % (e.extension_ns, e.extension),
|
|
e.extension_args)
|
|
self['error'].append(extxml)
|
|
self['error']['type'] = e.etype
|
|
self.send()
|
|
else:
|
|
self.reply()
|
|
# We probably didn't raise this on purpose, so send an error stanza
|
|
self['error']['condition'] = 'undefined-condition'
|
|
self['error']['text'] = "SleekXMPP got into trouble."
|
|
self.send()
|
|
# log the error
|
|
log.exception('Error handling {%s}%s stanza' %
|
|
(self.namespace, self.name))
|
|
# Finally raise the exception, so it can be handled (or not)
|
|
# at a higher level by using sys.excepthook.
|
|
raise e
|
|
|
|
register_stanza_plugin(RootStanza, Error)
|