07018c0afa
* 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)
63 lines
1.7 KiB
Python
63 lines
1.7 KiB
Python
from .. xmlstream.stanzabase import StanzaBase
|
|
from xml.etree import cElementTree as ET
|
|
from . error import Error
|
|
|
|
class Presence(StanzaBase):
|
|
interfaces = set(('type', 'to', 'from', 'id', 'status', 'priority'))
|
|
types = set(('available', 'unavailable', 'error', 'probe', 'subscribe', 'subscribed', 'unsubscribe', 'unsubscribed'))
|
|
showtypes = set(('dnd', 'ffc', 'xa', 'away'))
|
|
sub_interfaces = set(('status', 'priority'))
|
|
name = 'presence'
|
|
namespace = 'jabber:client'
|
|
|
|
def getShowElement(self):
|
|
return self.xml.find("{%s}show" % self.namespace)
|
|
|
|
def setType(self, value):
|
|
if value in self.types:
|
|
show = self.getShowElement()
|
|
if value in self.types:
|
|
if show is not None:
|
|
self.xml.remove(show)
|
|
if value == 'available':
|
|
value = ''
|
|
self._setAttr('type', value)
|
|
elif value in self.showtypes:
|
|
if show is None:
|
|
show = ET.Element("{%s}show" % self.namespace)
|
|
show.text = value
|
|
return self
|
|
|
|
def setPriority(self, value):
|
|
self._setSubText('priority', str(value))
|
|
|
|
def getPriority(self):
|
|
p = self._getSubText('priority')
|
|
if not p: p = 0
|
|
return int(p)
|
|
|
|
def getType(self):
|
|
out = self._getAttr('type')
|
|
if not out:
|
|
show = self.getShowElement()
|
|
if show is not None:
|
|
out = show.text
|
|
if not out or out is None:
|
|
out = 'available'
|
|
return out
|
|
|
|
def reply(self):
|
|
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_tag_map["{%s}%s" % (Error.namespace, Error.name)] = Error
|