github/SleekXMPP
1
0
Fork 0
mirror of https://github.com/correl/SleekXMPP.git synced 2024-11-24 03:00:15 +00:00
Code Issues Projects Releases Packages Wiki Activity

Updated message stanzas and tests with documentation and PEP8 style.

Browse source
This commit is contained in:
Lance Stout 2010-08-03 12:19:45 -04:00
parent 851e90c572
commit 939ae298c2
2 changed files with 170 additions and 89 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

168
sleekxmpp/stanza/message.py
View file

@ -5,59 +5,139 @@
See the file LICENSE for copying permission. See the file LICENSE for copying permission.
""" """
from .. xmlstream.stanzabase import StanzaBase
from xml.etree import cElementTree as ET from sleekxmpp.stanza import Error
from . error import Error from sleekxmpp.stanza.rootstanza import RootStanza
from . rootstanza import RootStanza from sleekxmpp.xmlstream.stanzabase import StanzaBase, ET
class Message(RootStanza): class Message(RootStanza):
interfaces = set(('type', 'to', 'from', 'id', 'body', 'subject', 'mucroom', 'mucnick'))
types = set((None, 'normal', 'chat', 'headline', 'error', 'groupchat'))
sub_interfaces = set(('body', 'subject'))
name = 'message'
plugin_attrib = name
namespace = 'jabber:client'
def getType(self): """
return self.xml.attrib.get('type', 'normal') XMPP's <message> stanzas are a "push" mechanism to send information
to other XMPP entities without requiring a response.
def chat(self): Chat clients will typically use <message> stanzas that have a type
self['type'] = 'chat' of either "chat" or "groupchat".
return self
def normal(self): When handling a message event, be sure to check if the message is
self['type'] = 'normal' an error response.
return self
def reply(self, body=None): Example <message> stanzas:
StanzaBase.reply(self) <message to="user1@example.com" from="user2@example.com">
if self['type'] == 'groupchat': <body>Hi!</body>
self['to'] = self['to'].bare </message>
del self['id']
if body is not None:
self['body'] = body
return self
def getMucroom(self): <message type="groupchat" to="room@conference.example.com">
if self['type'] == 'groupchat': <body>Hi everyone!</body>
return self['from'].bare </message>
else:
return ''
def setMucroom(self, value): Stanza Interface:
pass body -- The main contents of the message.
subject -- An optional description of the message's contents.
mucroom -- (Read-only) The name of the MUC room that sent the message.
mucnick -- (Read-only) The MUC nickname of message's sender.
def delMucroom(self): Attributes:
pass types -- May be one of: normal, chat, headline, groupchat, or error.
def getMucnick(self): Methods:
if self['type'] == 'groupchat': chat -- Set the message type to 'chat'.
return self['from'].resource normal -- Set the message type to 'normal'.
else: reply -- Overrides StanzaBase.reply
return '' getType -- Overrides StanzaBase interface
getMucroom -- Return the name of the MUC room of the message.
setMucroom -- Dummy method to prevent assignment.
delMucroom -- Dummy method to prevent deletion.
getMucnick -- Return the MUC nickname of the message's sender.
setMucnick -- Dummy method to prevent assignment.
delMucnick -- Dummy method to prevent deletion.
"""
def setMucnick(self, value): namespace = 'jabber:client'
pass name = 'message'
interfaces = set(('type', 'to', 'from', 'id', 'body', 'subject',
'mucroom', 'mucnick'))
sub_interfaces = set(('body', 'subject'))
plugin_attrib = name
types = set((None, 'normal', 'chat', 'headline', 'error', 'groupchat'))
def delMucnick(self): def getType(self):
pass """
Return the message type.
Overrides default stanza interface behavior.
Returns 'normal' if no type attribute is present.
"""
return self._getAttr('type', 'normal')
def chat(self):
"""Set the message type to 'chat'."""
self['type'] = 'chat'
return self
def normal(self):
"""Set the message type to 'chat'."""
self['type'] = 'normal'
return self
def reply(self, body=None):
"""
Create a message reply.
Overrides StanzaBase.reply.
Sets proper 'to' attribute if the message is from a MUC, and
adds a message body if one is given.
Arguments:
body -- Optional text content for the message.
"""
StanzaBase.reply(self)
if self['type'] == 'groupchat':
self['to'] = self['to'].bare
del self['id']
if body is not None:
self['body'] = body
return self
def getMucroom(self):
"""
Return the name of the MUC room where the message originated.
Read-only stanza interface.
"""
if self['type'] == 'groupchat':
return self['from'].bare
else:
return ''
def getMucnick(self):
"""
Return the nickname of the MUC user that sent the message.
Read-only stanza interface.
"""
if self['type'] == 'groupchat':
return self['from'].resource
else:
return ''
def setMucroom(self, value):
"""Dummy method to prevent modification."""
pass
def delMucroom(self):
"""Dummy method to prevent deletion."""
pass
def setMucnick(self, value):
"""Dummy method to prevent modification."""
pass
def delMucnick(self):
"""Dummy method to prevent deletion."""
pass

69
tests/test_messagestanzas.py
View file

@ -2,44 +2,45 @@ from sleektest import *
from sleekxmpp.stanza.message import Message from sleekxmpp.stanza.message import Message
from sleekxmpp.stanza.htmlim import HTMLIM from sleekxmpp.stanza.htmlim import HTMLIM
class TestMessageStanzas(SleekTest): class TestMessageStanzas(SleekTest):
def setUp(self): def setUp(self):
registerStanzaPlugin(Message, HTMLIM) registerStanzaPlugin(Message, HTMLIM)
def testGroupchatReplyRegression(self): def testGroupchatReplyRegression(self):
"Regression groupchat reply should be to barejid" "Regression groupchat reply should be to barejid"
msg = self.Message() msg = self.Message()
msg['to'] = 'me@myserver.tld' msg['to'] = 'me@myserver.tld'
msg['from'] = 'room@someservice.someserver.tld/somenick' msg['from'] = 'room@someservice.someserver.tld/somenick'
msg['type'] = 'groupchat' msg['type'] = 'groupchat'
msg['body'] = "this is a message" msg['body'] = "this is a message"
msg.reply() msg.reply()
self.failUnless(str(msg['to']) == 'room@someservice.someserver.tld') self.failUnless(str(msg['to']) == 'room@someservice.someserver.tld')
def testAttribProperty(self): def testAttribProperty(self):
"Test attrib property returning self" "Test attrib property returning self"
msg = self.Message() msg = self.Message()
msg.attrib.attrib.attrib['to'] = 'usr@server.tld' msg.attrib.attrib.attrib['to'] = 'usr@server.tld'
self.failUnless(str(msg['to']) == 'usr@server.tld') self.failUnless(str(msg['to']) == 'usr@server.tld')
def testHTMLPlugin(self): def testHTMLPlugin(self):
"Test message/html/html stanza" "Test message/html/html stanza"
msg = self.Message() msg = self.Message()
msg['to'] = "fritzy@netflint.net/sleekxmpp" msg['to'] = "fritzy@netflint.net/sleekxmpp"
msg['body'] = "this is the plaintext message" msg['body'] = "this is the plaintext message"
msg['type'] = 'chat' msg['type'] = 'chat'
p = ET.Element('{http://www.w3.org/1999/xhtml}p') p = ET.Element('{http://www.w3.org/1999/xhtml}p')
p.text = "This is the htmlim message" p.text = "This is the htmlim message"
msg['html']['html'] = p msg['html']['html'] = p
self.checkMessage(msg, """ self.checkMessage(msg, """
<message to="fritzy@netflint.net/sleekxmpp" type="chat"> <message to="fritzy@netflint.net/sleekxmpp" type="chat">
<body>this is the plaintext message</body> <body>this is the plaintext message</body>
<html xmlns="http://jabber.org/protocol/xhtml-im"> <html xmlns="http://jabber.org/protocol/xhtml-im">
<body xmlns="http://www.w3.org/1999/xhtml"> <body xmlns="http://www.w3.org/1999/xhtml">
<p>This is the htmlim message</p> <p>This is the htmlim message</p>
</body> </body>
</html> </html>
</message>""") </message>""")
suite = unittest.TestLoader().loadTestsFromTestCase(TestMessageStanzas) suite = unittest.TestLoader().loadTestsFromTestCase(TestMessageStanzas)