26 lines
1 KiB
Python
26 lines
1 KiB
Python
|
from .. xmlstream.stanzabase import StanzaBase
|
||
|
|
from xml.etree import cElementTree as ET
|
||
|
|
from . error import Error
|
||
|
|
from .. exceptions import XMPPError
|
||
|
|
import traceback
|
||
|
|
|
||
|
|
class RootStanza(StanzaBase):
|
||
|
|
|
||
|
|
def exception(self, e): #called when a handler raises an exception
|
||
|
|
self.reply()
|
||
|
|
if isinstance(e, XMPPError): # 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'].xml.append(extxml)
|
||
|
|
self['error']['type'] = e.etype
|
||
|
|
else: # we probably didn't raise this on purpose, so send back a traceback
|
||
|
|
self['error']['condition'] = 'undefined-condition'
|
||
|
|
self['error']['text'] = traceback.format_tb(e.__traceback__)
|
||
|
|
self.send()
|
||
|
|
|
||
|
|
# all jabber:client root stanzas should have the error plugin
|
||
|
|
RootStanza.plugin_attrib_map['error'] = Error
|
||
|
|
RootStanza.plugin_tag_map["{%s}%s" % (Error.namespace, Error.name)] = Error
|