Extract tools.config as config module

nemubot/tools/xmlparser/genericnode.py

@@ -0,0 +1,73 @@
+# Nemubot is a smart and modulable IM bot.
+# Copyright (C) 2012-2015  Mercier Pierre-Olivier
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+class GenericNode:
+
+    def __init__(self, tag, **kwargs):
+        self.tag = tag
+        self.attrs = kwargs
+        self.content = ""
+        self.children = []
+        self._cur = None
+        self._deep_cur = 0
+
+
+    def startElement(self, name, attrs):
+        if self._cur is None:
+            self._cur = GenericNode(name, **attrs)
+            self._deep_cur = 0
+        else:
+            self._deep_cur += 1
+            self._cur.startElement(name, attrs)
+        return True
+
+
+    def characters(self, content):
+        if self._cur is None:
+            self.content += content
+        else:
+            self._cur.characters(content)
+
+
+    def endElement(self, name):
+        if name is None:
+            return
+
+        if self._deep_cur:
+            self._deep_cur -= 1
+            self._cur.endElement(name)
+        else:
+            self.children.append(self._cur)
+            self._cur = None
+        return True
+
+
+    def hasNode(self, nodename):
+        return self.getNode(nodename) is not None
+
+
+    def getNode(self, nodename):
+        for c in self.children:
+            if c is not None and c.tag == nodename:
+                return c
+        return None
+
+
+    def __getitem__(self, item):
+        return self.attrs[item]
+
+    def __contains__(self, item):
+        return item in self.attrs