summaryrefslogtreecommitdiff
path: root/sleekxmpp/xmlstream/stanzabase.py
diff options
context:
space:
mode:
authorLance Stout <lancestout@gmail.com>2012-05-05 11:34:29 -0700
committerLance Stout <lancestout@gmail.com>2012-05-05 11:34:29 -0700
commit6a32417957a290c29b3e833bebf69fbcd8d9a8d3 (patch)
treeafbcc20a34fccaec01310823c8f026d42b38bd64 /sleekxmpp/xmlstream/stanzabase.py
parent97a7be7dfaa013d039886d01a0d81d1e614b17d1 (diff)
parentfa86f956efadd006a6d54f1b5f3b83c23b8eeebf (diff)
downloadslixmpp-6a32417957a290c29b3e833bebf69fbcd8d9a8d3.tar.gz
slixmpp-6a32417957a290c29b3e833bebf69fbcd8d9a8d3.tar.bz2
slixmpp-6a32417957a290c29b3e833bebf69fbcd8d9a8d3.tar.xz
slixmpp-6a32417957a290c29b3e833bebf69fbcd8d9a8d3.zip
Merge pull request #163 from whooo/master
factory for recurring substanzas
Diffstat (limited to 'sleekxmpp/xmlstream/stanzabase.py')
-rw-r--r--sleekxmpp/xmlstream/stanzabase.py42
1 files changed, 42 insertions, 0 deletions
diff --git a/sleekxmpp/xmlstream/stanzabase.py b/sleekxmpp/xmlstream/stanzabase.py
index 8a4f07ae..3b148446 100644
--- a/sleekxmpp/xmlstream/stanzabase.py
+++ b/sleekxmpp/xmlstream/stanzabase.py
@@ -67,6 +67,9 @@ def register_stanza_plugin(stanza, plugin, iterable=False, overrides=False):
if iterable:
stanza.plugin_iterables.add(plugin)
+ if iterable and hasattr(plugin, 'multi_attrib'):
+ multiplugin = multifactory(plugin, plugin.multi_attrib)
+ register_stanza_plugin(stanza, multiplugin)
if overrides:
for interface in plugin.overrides:
stanza.plugin_overrides[interface] = plugin.plugin_attrib
@@ -76,6 +79,45 @@ def register_stanza_plugin(stanza, plugin, iterable=False, overrides=False):
registerStanzaPlugin = register_stanza_plugin
+def multifactory(stanza, plugin_attrib):
+ """
+ Returns a ElementBase class for handling reoccuring child stanzas
+ """
+ class Multi(ElementBase):
+ """
+ Template class for multifactory
+ """
+ def setup(self, xml=None):
+ self.xml = ET.Element('')
+
+ def get_multi(self):
+ parent = self.parent()
+ res = filter(lambda sub: isinstance(sub, self._multistanza), parent)
+ return tuple(res)
+
+ def set_multi(self, val):
+ parent = self.parent()
+ del parent[self.plugin_attrib]
+ for sub in val:
+ parent.append(sub)
+
+ def del_multi(self):
+ parent = self.parent()
+ res = filter(lambda sub: isinstance(sub, self._multistanza), parent)
+ for stanza in res:
+ parent.iterables.remove(stanza)
+ parent.xml.remove(stanza.xml)
+
+ Multi.is_extension = True
+ Multi.plugin_attrib = plugin_attrib
+ Multi._multistanza = stanza
+ Multi.interfaces = (plugin_attrib,)
+ setattr(Multi, "get_%s" % plugin_attrib, get_multi)
+ setattr(Multi, "set_%s" % plugin_attrib, set_multi)
+ setattr(Multi, "del_%s" % plugin_attrib, del_multi)
+ return Multi
+
+
def fix_ns(xpath, split=False, propagate_ns=True, default_ns=''):
"""Apply the stanza's namespace to elements in an XPath expression.