blob: 3f9eb2ba3541cbb6a31f55982a38571798df2146 (
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
|
from .. xmlstream.stanzabase import StanzaBase
from xml.etree import cElementTree as ET
class Message(StanzaBase):
interfaces = set(('type', 'to', 'from', 'id', 'body', 'subject'))
types = set((None, 'normal', 'chat', 'headline', 'error'))
sub_interfaces = set(('body', 'subject'))
name = 'message'
namespace = 'jabber:client'
def getType(self):
return self.xml.attrib.get('type', 'normal')
def chat(self):
self['type'] = 'chat'
return self
def normal(self):
self['type'] = 'normal'
return self
def reply(self, body=None):
StanzaBase.reply(self)
if body is not None:
self['body'] = body
return self
if __name__ == '__main__':
m = Message()
m['to'] = 'me'
m['from'] = 'you'
m['type'] = 'chat'
m.reply()
m['body'] = 'Hello there!'
m['subject'] = 'whatever'
m['id'] = 'abc'
print(str(m))
print(m['body'])
print(m['subject'])
print(m['id'])
m['type'] = None
m['body'] = None
m['id'] = None
print(str(m))
print(m['type'])
|
