summaryrefslogtreecommitdiff
path: root/sleekxmpp/stanza/presence.py
blob: 6a8247c569ac08a492455cbc08d388cf4646ba1d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
from .. xmlstream.stanzabase import StanzaBase
from xml.etree import cElementTree as ET
from . error import Error
from . rootstanza import RootStanza

class Presence(RootStanza):
	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)