From d5e42ac0e7282500583bf17f21eb2f944600ce76 Mon Sep 17 00:00:00 2001
From: Lance Stout <lancestout@gmail.com>
Date: Mon, 19 Jul 2010 13:58:53 -0400
Subject: Condensed all of the stanzaPlugin functions into a single
 registerStanzaPlugin function.

Updated plugins and tests to use new function.
---
 sleekxmpp/xmlstream/stanzabase.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

(limited to 'sleekxmpp/xmlstream')

diff --git a/sleekxmpp/xmlstream/stanzabase.py b/sleekxmpp/xmlstream/stanzabase.py
index 024fe6cf..7592e1f6 100644
--- a/sleekxmpp/xmlstream/stanzabase.py
+++ b/sleekxmpp/xmlstream/stanzabase.py
@@ -19,6 +19,16 @@ else:
 
 xmltester = type(ET.Element('xml'))
 
+
+def registerStanzaPlugin(stanza, plugin):
+        """
+        Associate a stanza object as a plugin for another stanza.
+        """
+        tag = "{%s}%s" % (plugin.namespace, plugin.name)
+	stanza.plugin_attrib_map[plugin.plugin_attrib] = plugin
+	stanza.plugin_tag_map[tag] = plugin
+
+
 class JID(object):
 	def __init__(self, jid):
 		self.jid = jid
@@ -392,4 +402,4 @@ class StanzaBase(ElementBase):
 
 	def __copy__(self):
 		return self.__class__(xml=copy.deepcopy(self.xml), stream=self.stream)
-	
+                
-- 
cgit v1.2.3