* fixed many stanza bugs

* added stanza unhandled (unhandled iqs now reply with feature-not-implemented)
* added stanza exceptions (stanzas may now reply with exceptions when their handler raises an exception)
This commit is contained in:
Nathan Fritz 2009-12-17 01:54:22 +00:00
parent 6897a0b57c
commit 07018c0afa
13 changed files with 262 additions and 125 deletions

View file

@ -30,6 +30,8 @@ from . stanza.message import Message
from . stanza.iq import Iq from . stanza.iq import Iq
from . stanza.presence import Presence from . stanza.presence import Presence
from . stanza.roster import Roster from . stanza.roster import Roster
from . stanza.nick import Nick
from . stanza.htmlim import HTMLIM
import logging import logging
import threading import threading
@ -38,7 +40,6 @@ def stanzaPlugin(stanza, plugin):
stanza.plugin_attrib_map[plugin.plugin_attrib] = plugin stanza.plugin_attrib_map[plugin.plugin_attrib] = plugin
stanza.plugin_tag_map["{%s}%s" % (plugin.namespace, plugin.name)] = plugin stanza.plugin_tag_map["{%s}%s" % (plugin.namespace, plugin.name)] = plugin
stanzaPlugin(Iq, Roster)
class basexmpp(object): class basexmpp(object):
def __init__(self): def __init__(self):
@ -61,9 +62,16 @@ class basexmpp(object):
self.registerStanza(Message) self.registerStanza(Message)
self.registerStanza(Iq) self.registerStanza(Iq)
self.registerStanza(Presence) self.registerStanza(Presence)
self.stanzaPlugin(Iq, Roster)
self.stanzaPlugin(Message, Nick)
self.stanzaPlugin(Message, HTMLIM)
def stanzaPlugin(self, stanza, plugin):
stanza.plugin_attrib_map[plugin.plugin_attrib] = plugin
stanza.plugin_tag_map["{%s}%s" % (plugin.namespace, plugin.name)] = plugin
def Message(self, *args, **kwargs): def Message(self, *args, **kwargs):
return Presence(self, *args, **kwargs) return Message(self, *args, **kwargs)
def Iq(self, *args, **kwargs): def Iq(self, *args, **kwargs):
return Iq(self, *args, **kwargs) return Iq(self, *args, **kwargs)
@ -191,14 +199,13 @@ class basexmpp(object):
message = self.Message(sto=mto, stype=mtype, sfrom=mfrom) message = self.Message(sto=mto, stype=mtype, sfrom=mfrom)
message['body'] = mbody message['body'] = mbody
message['subject'] = msubject message['subject'] = msubject
message['nick'] = mnick if mnick is not None: message['nick'] = mnick
message['html'] = mhtml if mhtml is not None: message['html'] = mhtml
return message return message
def makePresence(self, pshow=None, pstatus=None, ppriority=None, pto=None, ptype=None, pfrom=None): def makePresence(self, pshow=None, pstatus=None, ppriority=None, pto=None, ptype=None, pfrom=None):
presence = self.Presence(stype=ptype, sfrom=pfrom, sto=pto) presence = self.Presence(stype=ptype, sfrom=pfrom, sto=pto)
if pshow is not None: if pshow is not None: presence['type'] = pshow
presence['type'] = pshow
if pfrom is None: #maybe this should be done in stanzabase if pfrom is None: #maybe this should be done in stanzabase
presence['from'] = self.fulljid presence['from'] = self.fulljid
presence['priority'] = ppriority presence['priority'] = ppriority
@ -237,7 +244,10 @@ class basexmpp(object):
def _handlePresence(self, presence): def _handlePresence(self, presence):
"""Update roster items based on presence""" """Update roster items based on presence"""
self.event("presence_%s" % presence['type'], presence) self.event("presence_%s" % presence['type'], presence)
if not presence['type'] in ('available', 'unavailable'): if presence['type'] in ('subscribe', 'subscribed', 'unsubscribe', 'unsubscribed'):
self.event('changed_subscription', presence)
return
elif not presence['type'] in ('available', 'unavailable'):
return return
jid = presence['from'].bare jid = presence['from'].bare
resource = presence['from'].resource resource = presence['from'].resource
@ -260,7 +270,7 @@ class basexmpp(object):
if wasoffline and show in ('available', 'away', 'xa', 'na', 'ffc'): if wasoffline and show in ('available', 'away', 'xa', 'na', 'ffc'):
self.event("got_online", presence) self.event("got_online", presence)
elif not wasoffline and show == 'unavailable': elif not wasoffline and show == 'unavailable':
self.event("got_offline", eventdata) self.event("got_offline", presence)
if len(self.roster[jid]['presence']) > 1: if len(self.roster[jid]['presence']) > 1:
del self.roster[jid]['presence'][resource] del self.roster[jid]['presence'][resource]
else: else:

View file

@ -64,14 +64,7 @@ class ComponentXMPP(basexmpp, XMLStream):
self.server_port = port self.server_port = port
self.set_jid(jid) self.set_jid(jid)
self.secret = secret self.secret = secret
self.registerHandler(Callback('PresenceProbe', MatchXMLMask("<presence xmlns='%s' type='probe'/>" % self.default_ns), self._handlePresenceProbe))
self.registerHandler(Callback('Handshake', MatchXPath('{jabber:component:accept}handshake'), self._handleHandshake)) self.registerHandler(Callback('Handshake', MatchXPath('{jabber:component:accept}handshake'), self._handleHandshake))
self.registerHandler(Callback('PresenceSubscription', MatchMany(\
(MatchXMLMask("<presence xmlns='%s' type='subscribe'/>" % self.default_ns), \
MatchXMLMask("<presence xmlns='%s' type='subscribed'/>" % self.default_ns), \
MatchXMLMask("<presence xmlns='%s' type='unsubscribe'/>" % self.default_ns), \
MatchXMLMask("<presence xmlns='%s' type='unsubscribed'/>" % self.default_ns) \
)), self._handlePresenceSubscription))
def incoming_filter(self, xmlobj): def incoming_filter(self, xmlobj):
if xmlobj.tag.startswith('{jabber:client}'): if xmlobj.tag.startswith('{jabber:client}'):
@ -80,22 +73,6 @@ class ComponentXMPP(basexmpp, XMLStream):
self.incoming_filter(sub) self.incoming_filter(sub)
return xmlobj return xmlobj
def _handlePresenceProbe(self, stanza):
xml = stanza.xml
self.event("got_presence_probe", ({
'from': xml.attrib['from'],
'to': xml.attrib['to']
}))
def _handlePresenceSubscription(self, presence):
xml = presence.xml
self.event("changed_subscription", {
'type' : xml.attrib['type'],
'from': xml.attrib['from'],
'to': xml.attrib['to']
})
def start_stream_handler(self, xml): def start_stream_handler(self, xml):
sid = xml.get('id', '') sid = xml.get('id', '')
handshake = ET.Element('{jabber:component:accept}handshake') handshake = ET.Element('{jabber:component:accept}handshake')

View file

@ -21,6 +21,98 @@ from __future__ import with_statement
from . import base from . import base
import logging import logging
from xml.etree import cElementTree as ET from xml.etree import cElementTree as ET
from .. xmlstream.stanzabase import ElementBase, JID
from .. stanza.presence import Presence
from .. xmlstream.handler.callback import Callback
from .. xmlstream.matcher.xpath import MatchXPath
from .. xmlstream.matcher.xmlmask import MatchXMLMask
class MUCPresence(ElementBase):
name = 'x'
namespace = 'http://jabber.org/protocol/muc#user'
plugin_attrib = 'muc'
interfaces = set(('affiliation', 'role', 'jid', 'nick', 'room'))
affiliations = set(('', ))
roles = set(('', ))
def getXMLItem(self):
item = self.xml.find('{http://jabber.org/protocol/muc#user}item')
if item is None:
item = ET.Element('{http://jabber.org/protocol/muc#user}item')
self.xml.append(item)
return item
def getAffiliation(self):
#TODO if no affilation, set it to the default and return default
item = self.getXMLItem()
return item.get('affiliation', '')
def setAffiliation(self, value):
item = self.getXMLItem()
#TODO check for valid affiliation
item.attrib['affiliation'] = value
return self
def delAffiliation(self):
item = self.getXMLItem()
#TODO set default affiliation
if 'affiliation' in item.attrib: del item.attrib['affiliation']
return self
def getJid(self):
item = self.getXMLItem()
return JID(item.get('jid', ''))
def setJid(self, value):
item = self.getXMLItem()
if not isinstance(value, str):
value = str(value)
item.attrib['jid'] = value
return self
def delJid(self):
item = self.getXMLItem()
if 'jid' in item.attrib: del item.attrib['jid']
return self
def getRole(self):
item = self.getXMLItem()
#TODO get default role, set default role if none
return item.get('role', '')
def setRole(self, value):
item = self.getXMLItem()
#TODO check for valid role
item.attrib['role'] = value
return self
def delRole(self):
item = self.getXMLItem()
#TODO set default role
if 'role' in item.attrib: del item.attrib['role']
return self
def getNick(self):
return self.parent['from'].resource
def getRoom(self):
return self.parent['from'].bare
def setNick(self, value):
logging.warning("Cannot set nick through mucpresence plugin.")
return self
def setRoom(self, value):
logging.warning("Cannot set room through mucpresence plugin.")
return self
def delNick(self):
logging.warning("Cannot delete nick through mucpresence plugin.")
return self
def delRoom(self):
logging.warning("Cannot delete room through mucpresence plugin.")
return self
class xep_0045(base.base_plugin): class xep_0045(base.base_plugin):
""" """
@ -32,74 +124,40 @@ class xep_0045(base.base_plugin):
self.ourNicks = {} self.ourNicks = {}
self.xep = '0045' self.xep = '0045'
self.description = 'Multi User Chat' self.description = 'Multi User Chat'
self.xmpp.add_handler("<message xmlns='%s' type='groupchat'><body/></message>" % self.xmpp.default_ns, self.handle_groupchat_message) # load MUC support in presence stanzas
self.xmpp.add_handler("<presence xmlns='%s' />" % self.xmpp.default_ns, self.handle_groupchat_presence) self.xmpp.stanzaPlugin(Presence, MUCPresence)
self.xmpp.registerHandler(Callback('MUCPresence', MatchXMLMask("<presence xmlns='%s' />" % self.xmpp.default_ns), self.handle_groupchat_presence))
self.xmpp.registerHandler(Callback('MUCMessage', MatchXMLMask("<message xmlns='%s' type='groupchat'><body/></message>" % self.xmpp.default_ns), self.handle_groupchat_message))
def handle_groupchat_presence(self, xml): def handle_groupchat_presence(self, pr):
""" Handle a presence in a muc. """ Handle a presence in a muc.
""" """
source = xml.attrib['from'] if pr['muc']['room'] not in self.rooms.keys():
room = self.xmpp.getjidbare(source)
if room not in self.rooms.keys():
return return
nick = self.xmpp.getjidresource(source) entry = pr['muc'].getValues()
entry = { if pr['type'] == 'unavailable':
'nick': nick, self.rooms[entry['room']][entry['nick']] = None
'room': room,
}
if 'type' in xml.attrib.keys():
entry['type'] = xml.attrib['type']
else: else:
entry['type'] = 'available' self.rooms[entry['room']][entry['nick']] = entry
for tag in ['status','show','priority']:
if xml.find(('{%s}' % self.xmpp.default_ns) + tag) != None:
entry[tag] = xml.find(('{%s}' % self.xmpp.default_ns) + tag).text
else:
entry[tag] = None
for tag in ['affiliation','role','jid']:
item = xml.find('{http://jabber.org/protocol/muc#user}x/{http://jabber.org/protocol/muc#user}item')
if item != None:
if tag in item.attrib:
entry[tag] = item.attrib[tag]
else:
entry[tag] = None
else:
entry[tag] = None
if entry['status'] == 'unavailable':
self.rooms[room][nick] = None
else:
self.rooms[room][nick] = entry
logging.debug("MUC presence from %s/%s : %s" % (entry['room'],entry['nick'], entry)) logging.debug("MUC presence from %s/%s : %s" % (entry['room'],entry['nick'], entry))
self.xmpp.event("groupchat_presence", entry) self.xmpp.event("groupchat_presence", pr)
def handle_groupchat_message(self, xml): def handle_groupchat_message(self, msg):
""" Handle a message event in a muc. """ Handle a message event in a muc.
""" """
mfrom = xml.attrib['from'] self.xmpp.event('groupchat_message', msg)
message = xml.find('{%s}body' % self.xmpp.default_ns).text
subject = xml.find('{%s}subject' % self.xmpp.default_ns)
if subject:
subject = subject.text
else:
subject = ''
resource = self.xmpp.getjidresource(mfrom)
mfrom = self.xmpp.getjidbare(mfrom)
mtype = xml.attrib.get('type', 'normal')
self.xmpp.event("groupchat_message", {'room': mfrom, 'name': resource, 'type': mtype, 'subject': subject, 'message': message})
def getRoomForm(self, room, ifrom=None): def getRoomForm(self, room, ifrom=None):
iq = self.xmpp.makeIqGet() iq = self.xmpp.makeIqGet()
iq.attrib['to'] = room iq['to'] = room
if ifrom is not None: if ifrom is not None:
iq.attrib['from'] = ifrom iq['from'] = ifrom
query = ET.Element('{http://jabber.org/protocol/muc#owner}query') query = ET.Element('{http://jabber.org/protocol/muc#owner}query')
iq.append(query) iq.append(query)
result = self.xmpp.send(iq, self.xmpp.makeIq(id=iq.get('id'))) result = iq.send()
if result.get('type', 'error') == 'error': if result['type'] == 'error':
return False return False
xform = result.find('{http://jabber.org/protocol/muc#owner}query/{jabber:x:data}x') xform = result.xml.find('{http://jabber.org/protocol/muc#owner}query/{jabber:x:data}x')
if xform is None: return False if xform is None: return False
form = self.xmpp.plugin['xep_0004'].buildForm(xform) form = self.xmpp.plugin['xep_0004'].buildForm(xform)
return form return form
@ -110,15 +168,16 @@ class xep_0045(base.base_plugin):
#form = self.xmpp.plugin['xep_0004'].makeForm(ftype='submit') #form = self.xmpp.plugin['xep_0004'].makeForm(ftype='submit')
#form.addField('FORM_TYPE', value='http://jabber.org/protocol/muc#roomconfig') #form.addField('FORM_TYPE', value='http://jabber.org/protocol/muc#roomconfig')
iq = self.xmpp.makeIqSet() iq = self.xmpp.makeIqSet()
iq.attrib['to'] = room iq['to'] = room
if ifrom is not None: if ifrom is not None:
iq.attrib['from'] = ifrom iq['from'] = ifrom
query = ET.Element('{http://jabber.org/protocol/muc#owner}query') query = ET.Element('{http://jabber.org/protocol/muc#owner}query')
form = form.getXML('submit') form = form.getXML('submit')
query.append(form) query.append(form)
iq.append(query) iq.append(query)
result = self.xmpp.send(iq, self.xmpp.makeIq(iq.get('id'))) #result = self.xmpp.send(iq, self.xmpp.makeIq(iq.get('id')))
if result.get('type', 'error') == 'error': result = iq.send()
if result['type'] == 'error':
return False return False
return True return True
@ -147,8 +206,8 @@ class xep_0045(base.base_plugin):
def destroy(self, room, reason='', altroom = '', ifrom=None): def destroy(self, room, reason='', altroom = '', ifrom=None):
iq = self.xmpp.makeIqSet() iq = self.xmpp.makeIqSet()
if ifrom is not None: if ifrom is not None:
iq.attrib['from'] = ifrom iq['from'] = ifrom
iq.attrib['to'] = room iq['to'] = room
query = ET.Element('{http://jabber.org/protocol/muc#owner}query') query = ET.Element('{http://jabber.org/protocol/muc#owner}query')
destroy = ET.Element('destroy') destroy = ET.Element('destroy')
if altroom: if altroom:
@ -158,8 +217,9 @@ class xep_0045(base.base_plugin):
destroy.append(xreason) destroy.append(xreason)
query.append(destroy) query.append(destroy)
iq.append(query) iq.append(query)
r = self.xmpp.send(iq, self.xmpp.makeIq(iq.get('id'))) #r = self.xmpp.send(iq, self.xmpp.makeIq(iq.get('id')))
if r is None or r.get('type', 'error') == 'error': r = iq.send()
if r is False or r['type'] == 'error':
return False return False
return True return True
@ -171,17 +231,18 @@ class xep_0045(base.base_plugin):
item = ET.Element('item', {'affiliation':affiliation, 'jid':jid}) item = ET.Element('item', {'affiliation':affiliation, 'jid':jid})
query.append(item) query.append(item)
iq = self.xmpp.makeIqSet(query) iq = self.xmpp.makeIqSet(query)
iq.attrib['to'] = room iq['to'] = room
result = self.xmpp.send(iq, "<iq id='%s' />" % iq.get('id')) result = iq.send()
if result is None or result.get('type') != 'result': if result is False or result['type'] != 'result':
raise ValueError raise ValueError
return True return True
def invite(self, room, jid, reason=''): def invite(self, room, jid, reason=''):
""" Invite a jid to a room.""" """ Invite a jid to a room."""
msg = self.xmpp.makeMessage(room, mtype='none') msg = self.xmpp.makeMessage(room)
msg['from'] = self.xmpp.jid
x = ET.Element('{http://jabber.org/protocol/muc#user}x') x = ET.Element('{http://jabber.org/protocol/muc#user}x')
invite = ET.Element('invite', {'to': jid}) invite = ET.Element('{http://jabber.org/protocol/muc#user}invite', {'to': jid})
if reason: if reason:
rxml = ET.Element('reason') rxml = ET.Element('reason')
rxml.text = reason rxml.text = reason
@ -198,11 +259,11 @@ class xep_0045(base.base_plugin):
def getRoomConfig(self, room): def getRoomConfig(self, room):
iq = self.xmpp.makeIqGet('http://jabber.org/protocol/muc#owner') iq = self.xmpp.makeIqGet('http://jabber.org/protocol/muc#owner')
iq.attrib['to'] = room iq['to'] = room
result = self.xmpp.send(iq, "<iq id='%s' />" % iq.get('id')) result = iq.send()
if result is None or result.get('type') != 'result': if result is None or result['type'] != 'result':
raise ValueError raise ValueError
form = result.find('{http://jabber.org/protocol/muc#owner}query/{jabber:x:data}x') form = result.xml.find('{http://jabber.org/protocol/muc#owner}query/{jabber:x:data}x')
if form is None: if form is None:
raise ValueError raise ValueError
return self.xmpp.plugin['xep_0004'].buildForm(form) return self.xmpp.plugin['xep_0004'].buildForm(form)
@ -212,15 +273,15 @@ class xep_0045(base.base_plugin):
x = ET.Element('{jabber:x:data}x', type='cancel') x = ET.Element('{jabber:x:data}x', type='cancel')
query.append(x) query.append(x)
iq = self.xmpp.makeIqSet(query) iq = self.xmpp.makeIqSet(query)
self.xmpp.send(iq, "<iq id='%s' />" % iq.get('id')) iq.send()
def setRoomConfig(self, room, config): def setRoomConfig(self, room, config):
query = ET.Element('{http://jabber.org/protocol/muc#owner}query') query = ET.Element('{http://jabber.org/protocol/muc#owner}query')
x = config.getXML('submit') x = config.getXML('submit')
query.append(x) query.append(x)
iq = self.xmpp.makeIqSet(query) iq = self.xmpp.makeIqSet(query)
iq.attrib['to'] = room iq['to'] = room
self.xmpp.send(iq, "<iq id='%s' />" % iq.get('id')) iq.send()
def getJoinedRooms(self): def getJoinedRooms(self):
return self.rooms.keys() return self.rooms.keys()

View file

@ -5,6 +5,8 @@ class HTMLIM(ElementBase):
name = 'html' name = 'html'
plugin_attrib = 'html' plugin_attrib = 'html'
interfaces = set(('html')) interfaces = set(('html'))
plugin_attrib_map = set()
plugin_xml_map = set()
def setHtml(self, html): def setHtml(self, html):
if issinstance(html, str): if issinstance(html, str):

View file

@ -5,7 +5,7 @@ from .. xmlstream.handler.waiter import Waiter
from .. xmlstream.matcher.id import MatcherId from .. xmlstream.matcher.id import MatcherId
class Iq(StanzaBase): class Iq(StanzaBase):
interfaces = set(('type', 'to', 'from', 'id', 'body', 'subject')) interfaces = set(('type', 'to', 'from', 'id','query'))
types = set(('get', 'result', 'set', 'error')) types = set(('get', 'result', 'set', 'error'))
name = 'iq' name = 'iq'
namespace = 'jabber:client' namespace = 'jabber:client'
@ -13,7 +13,19 @@ class Iq(StanzaBase):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
StanzaBase.__init__(self, *args, **kwargs) StanzaBase.__init__(self, *args, **kwargs)
if self['id'] == '': if self['id'] == '':
self['id'] = self.stream.getId() self['id'] = self.stream.getNewId()
def exception(self, text):
self.reply()
self['error']['condition'] = 'undefined-condition'
self['error']['text'] = text
self.send()
def unhandled(self):
self.reply()
self['error']['condition'] = 'feature-not-implemented'
self['error']['text'] = 'No handlers registered for this request.'
self.send()
def result(self): def result(self):
self['type'] = 'result' self['type'] = 'result'
@ -36,6 +48,29 @@ class Iq(StanzaBase):
self.clear() self.clear()
StanzaBase.setPayload(self, value) StanzaBase.setPayload(self, value)
def setQuery(self, value):
query = self.xml.find("{%s}query" % value)
if query is None:
self.clear()
query = ET.Element("{%s}query" % value)
self.xml.append(query)
return self
def getQuery(self):
for child in self.getchildren():
if child.tag.endswith('query'):
ns =child.tag.split('}')[0]
if '{' in ns:
ns = ns[1:]
return ns
return ''
def delQuery(self):
for child in self.getchildren():
if child.tag.endswith('query'):
self.xml.remove(child)
return self
def unhandled(self): def unhandled(self):
pass pass
# returned unhandled error # returned unhandled error

View file

@ -4,7 +4,7 @@ from . error import Error
class Message(StanzaBase): class Message(StanzaBase):
interfaces = set(('type', 'to', 'from', 'id', 'body', 'subject')) interfaces = set(('type', 'to', 'from', 'id', 'body', 'subject'))
types = set((None, 'normal', 'chat', 'headline', 'error')) types = set((None, 'normal', 'chat', 'headline', 'error', 'groupchat'))
sub_interfaces = set(('body', 'subject')) sub_interfaces = set(('body', 'subject'))
name = 'message' name = 'message'
namespace = 'jabber:client' namespace = 'jabber:client'
@ -22,9 +22,16 @@ class Message(StanzaBase):
def reply(self, body=None): def reply(self, body=None):
StanzaBase.reply(self) StanzaBase.reply(self)
del self['id']
if body is not None: if body is not None:
self['body'] = body self['body'] = body
return self return self
def exception(self, text):
self.reply()
self['error']['condition'] = 'undefined-condition'
self['error']['text'] = text
self.send()
Message.plugin_attrib_map['error'] = Error Message.plugin_attrib_map['error'] = Error
Message.plugin_tag_map["{%s}%s" % (Error.namespace, Error.name)] = Error Message.plugin_tag_map["{%s}%s" % (Error.namespace, Error.name)] = Error

View file

@ -1,10 +1,12 @@
from .. xmlstream.stanzabase import ElementBase, ET from .. xmlstream.stanzabase import ElementBase, ET
class HTMLIM(ElementBase): class Nick(ElementBase):
namespace = 'http://jabber.org/nick/nick' namespace = 'http://jabber.org/nick/nick'
name = 'nick' name = 'nick'
plugin_attrib = 'nick' plugin_attrib = 'nick'
interfaces = set(('nick')) interfaces = set(('nick'))
plugin_attrib_map = set()
plugin_xml_map = set()
def setNick(self, nick): def setNick(self, nick):
self.xml.text = nick self.xml.text = nick

View file

@ -19,6 +19,8 @@ class Presence(StanzaBase):
if value in self.types: if value in self.types:
if show is not None: if show is not None:
self.xml.remove(show) self.xml.remove(show)
if value == 'available':
value = ''
self._setAttr('type', value) self._setAttr('type', value)
elif value in self.showtypes: elif value in self.showtypes:
if show is None: if show is None:
@ -44,8 +46,18 @@ class Presence(StanzaBase):
out = 'available' out = 'available'
return out return out
def delType(self): def reply(self):
self.setType('available') if self['type'] == 'unsubscribe':
self['type'] = 'unsubscribed'
elif self['type'] == 'subscribe':
self['type'] = 'subscribed'
return StanzaBase.reply(self)
def exception(self, text):
self.reply()
self['error']['condition'] = 'undefined-condition'
self['error']['text'] = text
self.send()
Presence.plugin_attrib_map['error'] = Error Presence.plugin_attrib_map['error'] = Error
Presence.plugin_tag_map["{%s}%s" % (Error.namespace, Error.name)] = Error Presence.plugin_tag_map["{%s}%s" % (Error.namespace, Error.name)] = Error

View file

@ -20,7 +20,7 @@ class Waiter(base.BaseHandler):
return self._payload.get(True, timeout) return self._payload.get(True, timeout)
except queue.Empty: except queue.Empty:
logging.warning("Timed out waiting for %s" % self.name) logging.warning("Timed out waiting for %s" % self.name)
return StanzaBase(stype='error') return False
def checkDelete(self): def checkDelete(self):
return True return True

View file

@ -0,0 +1,6 @@
from . import base
class MatcherId(base.MatcherBase):
def match(self, xml):
return xml.get('id') == self._criteria

View file

@ -1,4 +1,5 @@
from xml.etree import cElementTree as ET from xml.etree import cElementTree as ET
import logging
class JID(object): class JID(object):
def __init__(self, jid): def __init__(self, jid):
@ -45,7 +46,12 @@ class ElementBase(object):
if self.xml is None: if self.xml is None:
self.xml = xml self.xml = xml
if self.xml is None: if self.xml is None:
self.xml = ET.Element("{%(namespace)s}%(name)s" % {'name': self.name, 'namespace': self.namespace}) for ename in self.name.split('/'):
new = ET.Element("{%(namespace)s}%(name)s" % {'name': self.name, 'namespace': self.namespace})
if self.xml is None:
self.xml = new
else:
self.xml.append(new)
if self.parent is not None: if self.parent is not None:
self.parent.xml.append(self.xml) self.parent.xml.append(self.xml)
return True #had to generate XML return True #had to generate XML
@ -70,7 +76,7 @@ class ElementBase(object):
else: else:
return self._getAttr(attrib) return self._getAttr(attrib)
elif attrib in self.plugin_attrib_map: elif attrib in self.plugin_attrib_map:
self.initPlugin(attrib) if attrib not in self.plugins: self.initPlugin(attrib)
return self.plugins[attrib] return self.plugins[attrib]
else: else:
return '' return ''
@ -87,9 +93,10 @@ class ElementBase(object):
self._setAttr(attrib, value) self._setAttr(attrib, value)
else: else:
self.__delitem__(attrib) self.__delitem__(attrib)
elif attrib in self.plugin_map: elif attrib in self.plugin_attrib_map:
if attrib not in self.plugins: self.initPlugin(attrib)
self.initPlugin(attrib) self.initPlugin(attrib)
self.plugins[attrib].setValues(value) self.plugins[attrib][attrib] = value
return self return self
def __delitem__(self, attrib): def __delitem__(self, attrib):
@ -101,7 +108,7 @@ class ElementBase(object):
return self._delSub(attrib) return self._delSub(attrib)
else: else:
self._delAttr(attrib) self._delAttr(attrib)
elif attrib in self.plugin_map: elif attrib in self.plugin_attrib_map:
if attrib in self.plugins: if attrib in self.plugins:
del self.plugins[attrib] del self.plugins[attrib]
return self return self
@ -114,6 +121,9 @@ class ElementBase(object):
return True return True
def _setAttr(self, name, value): def _setAttr(self, name, value):
if value is None or value == '':
self.__delitem__(name)
else:
self.xml.attrib[name] = value self.xml.attrib[name] = value
def _delAttr(self, name): def _delAttr(self, name):
@ -131,11 +141,13 @@ class ElementBase(object):
return stanza.text return stanza.text
def _setSubText(self, name, attrib={}, text=None): def _setSubText(self, name, attrib={}, text=None):
if text is None or text == '':
return self.__delitem__(name)
stanza = self.xml.find("{%s}%s" % (self.namespace, name)) stanza = self.xml.find("{%s}%s" % (self.namespace, name))
if stanza is None: if stanza is None:
self.xml.append(ET.Element("{%s}%s" % (self.namespace, name), attrib)) #self.xml.append(ET.Element("{%s}%s" % (self.namespace, name), attrib))
stanza = self.xml.find("{%s}%s" % (self.namespace, name)) stanza = ET.Element("{%s}%s" % (self.namespace, name))
if text is not None: self.xml.append(stanza)
stanza.text = text stanza.text = text
return stanza return stanza
@ -228,6 +240,9 @@ class StanzaBase(ElementBase):
def unhandled(self): def unhandled(self):
pass pass
def exception(self, text):
logging.error(text)
def send(self): def send(self):
self.stream.sendRaw(str(self)) self.stream.sendRaw(str(self))
@ -285,7 +300,7 @@ class StanzaBase(ElementBase):
text[cc] = '&gt;' text[cc] = '&gt;'
elif c == "'": elif c == "'":
text[cc] = '&apos;' text[cc] = '&apos;'
elif self.escape_quotes: else:
text[cc] = '&quot;' text[cc] = '&quot;'
cc += 1 cc += 1
return ''.join(text) return ''.join(text)

View file

@ -262,6 +262,8 @@ class XMLStream(object):
handler.prerun(stanza) handler.prerun(stanza)
self.eventqueue.put(('stanza', handler, stanza)) self.eventqueue.put(('stanza', handler, stanza))
if handler.checkDelete(): self.__handlers.pop(self.__handlers.index(handler)) if handler.checkDelete(): self.__handlers.pop(self.__handlers.index(handler))
else:
stanza.unhandled()
#loop through handlers and test match #loop through handlers and test match
#spawn threads as necessary, call handlers, sending Stanza #spawn threads as necessary, call handlers, sending Stanza
@ -274,10 +276,18 @@ class XMLStream(object):
except queue.Empty: except queue.Empty:
event = None event = None
if event is not None: if event is not None:
etype, handler, stanza = event etype, handler, *args = event
if etype == 'stanza': if etype == 'stanza':
handler.run(stanza) try:
if etype == 'quit': handler.run(args[0])
except:
args[0].exception(traceback.format_exc())
elif etype == 'sched':
try:
handler.run(*args)
except:
logging.error(traceback.format_exc())
elif etype == 'quit':
logging.debug("Quitting eventRunner thread") logging.debug("Quitting eventRunner thread")
return False return False