scripts: add h5py format script

convert output xml to h5py format with below naming convertion 1. file name is the version number. 2. each board is defined as one attribute usage: pip3 install h5py python3 ./h5py_demo.py --input <path to xml file> --output <path to output folder> Signed-off-by: Hake Huang <hake.huang@oss.nxp.com>
hakehuang · Jan 13, 2021 · 8483448 · 8483448 · PerMac · Jan 18, 2021
1 parent 2be7661
commit 8483448
Showing 1 changed file with 63 additions and 0 deletions.
diff --git a/scripts/h5py_demo.py b/scripts/h5py_demo.py
@@ -0,0 +1,63 @@
+"""Create an HDF5 file in memory and retrieve the raw bytes
+require python3.7.5 above
+"""
+import sys
+import os
+import argparse
+import logging
+
+import xml.etree.ElementTree as ET
+import h5py
+
+def file_path(string):
+    '''
+        check whether string is a directory
+    '''
+    if os.path.isfile(string):
+        return string
+    raise NotADirectoryError(string)
+
+def dir_path(string):
+    '''
+        check whether string is a directory
+    '''
+    if os.path.isdir(string):
+        return string
+    raise NotADirectoryError(string)
+
+def parser_args():
+    '''
+    parser arguements
+    '''
+    parser = argparse.ArgumentParser(description='xml to h5py format convert')
+    parser.add_argument('--input', type=file_path, required=True, help='Path to the imput xml file')
+    parser.add_argument('--output', type=dir_path, default=".",
+                        help='Path to the output h5py folder')
+    return parser.parse_args()
+
+def process_xml(opts):
+    '''
+    process xml to hash
+    '''
+    summary = ET.parse(opts.input).getroot()[0]
+    name_in_report = summary.attrib['name']
+    properties = summary[0]
+    version = "unknow"
+    for _p in properties:
+        if _p.attrib['name'] == "version":
+            version = _p.attrib['value']
+    logging.info(version)
+    out_put = os.path.join(opts.output, version + ".h5")
+    logging.info(out_put)
+    _h5f = h5py.File(out_put, 'a')
+    _xmlfh = open(opts.input, 'rb')
+    _h5f.attrs[name_in_report] = _xmlfh.read()
+    _h5f.close()
+    _xmlfh.close()
+
+
+if __name__ == '__main__':
+    sys.stdout.flush()
+    logging.basicConfig(level=logging.DEBUG)
+    m_opts = parser_args()
+    process_xml(m_opts)