Modified the __read_qml() function to specify the item we're searchin…

…g for.
pierre-manchon · Aug 26, 2021 · 2d6778a · 2d6778a
1 parent d218708
commit 2d6778a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/INPMT/__utils/utils.py b/INPMT/__utils/utils.py
@@ -136,9 +136,9 @@ def __set_cfg_val(var, value):
         print("\n")
 
 
-def __read_qml(path_qml: AnyStr) -> List:
+def __read_qml(path_qml: AnyStr, type: AnyStr) -> List:
     xml_data = xml.dom.minidom.parse(path_qml)
     legend = []
-    for item in xml_data.getElementsByTagName("item"):
+    for item in xml_data.getElementsByTagName(type):
         legend.append([item.getAttribute("value"), item.getAttribute("label")])
     return legend