You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
2.5 KiB
79 lines
2.5 KiB
# Copyright © 2011-2024 Splunk, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"): you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
# File for utility functions
|
|
|
|
|
|
def xml_compare(expected, found):
|
|
"""Checks equality of two ``ElementTree`` objects.
|
|
|
|
:param expected: An ``ElementTree`` object.
|
|
:param found: An ``ElementTree`` object.
|
|
:return: ``Boolean``, whether the two objects are equal.
|
|
"""
|
|
|
|
# if comparing the same ET object
|
|
if expected == found:
|
|
return True
|
|
|
|
# compare element attributes, ignoring order
|
|
if set(expected.items()) != set(found.items()):
|
|
return False
|
|
|
|
# check for equal number of children
|
|
expected_children = list(expected)
|
|
found_children = list(found)
|
|
if len(expected_children) != len(found_children):
|
|
return False
|
|
|
|
# compare children
|
|
if not all(xml_compare(a, b) for a, b in zip(expected_children, found_children)):
|
|
return False
|
|
|
|
# compare elements, if there is no text node, return True
|
|
if (expected.text is None or expected.text.strip() == "") and (
|
|
found.text is None or found.text.strip() == ""
|
|
):
|
|
return True
|
|
return (
|
|
expected.tag == found.tag
|
|
and expected.text == found.text
|
|
and expected.attrib == found.attrib
|
|
)
|
|
|
|
|
|
def parse_parameters(param_node):
|
|
if param_node.tag == "param":
|
|
return param_node.text
|
|
if param_node.tag == "param_list":
|
|
parameters = []
|
|
for mvp in param_node:
|
|
parameters.append(mvp.text)
|
|
return parameters
|
|
raise ValueError(f"Invalid configuration scheme, {param_node.tag} tag unexpected.")
|
|
|
|
|
|
def parse_xml_data(parent_node, child_node_tag):
|
|
data = {}
|
|
for child in parent_node:
|
|
child_name = child.get("name")
|
|
if child.tag == child_node_tag:
|
|
if child_node_tag == "stanza":
|
|
data[child_name] = {"__app": child.get("app", None)}
|
|
for param in child:
|
|
data[child_name][param.get("name")] = parse_parameters(param)
|
|
elif "item" == parent_node.tag:
|
|
data[child_name] = parse_parameters(child)
|
|
return data
|