初始提交

build/script/sdk_generator/xml_parser.py

@@ -0,0 +1,75 @@
+#!/usr/bin/env python3
+# coding=utf-8
+# Copyright (c) HiSilicon (Shanghai) Technologies Co., Ltd. 2022-2022. All rights reserved.
+
+import os
+from xml.etree import cElementTree as ElementTree
+
+class XmlParser:
+    def __init__(self):
+        self.d2xml_level = 0
+        self.indent = '  '
+
+    def _dict_to_xml(self, data, root='object'):
+        if isinstance(data, dict):
+            xml_ret = '  ' * self.d2xml_level + f'<{root}>\n'
+            self.d2xml_level += 1
+            for key, value in data.items():
+                xml_ret += self._dict_to_xml(value, key)
+            self.d2xml_level -= 1
+            xml_ret += '  ' * self.d2xml_level + f'</{root}>\n'
+            return xml_ret
+
+        if isinstance(data, (list, tuple, set)):
+            xml_ret = ''
+            for item in data:
+                xml_ret += self._dict_to_xml(item, root)
+            return xml_ret
+        if data is None:
+            data = ''
+        xml_ret = '  ' * self.d2xml_level + f'<{root}>'
+        xml_ret += str(data)
+        xml_ret += f'</{root}>\n'
+        return xml_ret
+
+    def dict_to_xml(self, data, root='object'):
+        self.d2xml_level = 0
+        xml_str = '<?xml version="1.0" encoding="UTF-8"?>\n'
+        xml_str += self._dict_to_xml(data, root)
+        return xml_str
+
+    def et2dict(self, root):
+        xml_dict = {}
+        if root.items():
+            xml_dict.update(dict(root.items()))
+        for element in root:
+            if element:
+                key = element.tag
+                val = self.et2dict(element)
+            else:
+                key = element.tag
+                val = element.text
+
+            if key not in xml_dict:
+                xml_dict[key] = val
+                continue
+
+            if not isinstance(xml_dict[key], list):
+                xml_dict[key] = [xml_dict[key]]
+            temp = xml_dict[key]
+            temp.append(val)
+            xml_dict[key] = temp
+        return xml_dict
+
+    def xml2dict(self, xml_file):
+        tree = ElementTree.parse(xml_file)
+        root = tree.getroot()
+        return self.et2dict(root)
+
+if __name__ == "__main__":
+    script_dir = os.path.split(os.path.realpath(__file__))[0]
+    file = os.path.join(script_dir, 'sdk_template', 'template.ewp')
+    test = XmlParser()
+    dict_ = test.xml2dict(file)
+    xml = test.dict_to_xml(dict_, root='project')
+    print(xml)