Modified the return values for several methods so that they can be chained.

For example:

    iq.reply().error().setPayload(something.xml).send()
This commit is contained in:
Lance Stout 2010-05-22 22:40:30 +08:00 committed by Nathan Fritz
parent 828cba875f
commit 35f4ef3452
2 changed files with 6 additions and 1 deletions

View file

@ -37,6 +37,7 @@ class Iq(RootStanza):
def setPayload(self, value):
self.clear()
StanzaBase.setPayload(self, value)
return self
def setQuery(self, value):
query = self.xml.find("{%s}query" % value)

View file

@ -332,7 +332,7 @@ class StanzaBase(ElementBase):
def setType(self, value):
if value in self.types:
self.xml.attrib['type'] = value
self.xml.attrib['type'] = value
return self
def getPayload(self):
@ -340,15 +340,18 @@ class StanzaBase(ElementBase):
def setPayload(self, value):
self.xml.append(value)
return self
def delPayload(self):
self.clear()
return self
def clear(self):
for child in self.xml.getchildren():
self.xml.remove(child)
for plugin in list(self.plugins.keys()):
del self.plugins[plugin]
return self
def reply(self):
self['from'], self['to'] = self['to'], self['from']
@ -357,6 +360,7 @@ class StanzaBase(ElementBase):
def error(self):
self['type'] = 'error'
return self
def getTo(self):
return JID(self._getAttr('to'))