45 lines
972 B
Python
45 lines
972 B
Python
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'])
|