mirror of
https://github.com/projectacrn/acrn-hypervisor.git
synced 2025-05-16 20:29:53 +00:00
9f13bbc349
Today the script default_populator.py fills in default values by visiting an XML schema and an XML tree simultaneously and add a node to the latter whenever one satisfying the schema does not exist. This visiting logic is not only useful for filling in default values, but also for upgrading XMLs to new schemas. This patch abstracts the flow of the visiting above as a separate class, just like tree visitors or transformers. The current default value populator is then refactored to extend that class by inheritance. Tracked-On: #6690 Signed-off-by: Junjie Mao <junjie.mao@intel.com>
46 lines
1.5 KiB
Python
Executable File
46 lines
1.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
#
|
|
# Copyright (C) 2022 Intel Corporation.
|
|
#
|
|
# SPDX-License-Identifier: BSD-3-Clause
|
|
#
|
|
|
|
import argparse
|
|
import lxml.etree as etree
|
|
from scenario_transformer import ScenarioTransformer
|
|
|
|
class DefaultValuePopulator(ScenarioTransformer):
|
|
def add_missing_nodes(self, xsd_element_node, xml_parent_node, new_node_index):
|
|
element_name = xsd_element_node.get("name")
|
|
default_value = xsd_element_node.get("default")
|
|
|
|
new_node = etree.Element(element_name)
|
|
if default_value is not None:
|
|
new_node.text = default_value
|
|
|
|
if new_node_index is not None:
|
|
xml_parent_node.insert(new_node_index, new_node)
|
|
else:
|
|
xml_parent_node.append(new_node)
|
|
|
|
return [new_node]
|
|
|
|
def main(xsd_file, xml_file, out_file):
|
|
xsd_etree = etree.parse(xsd_file)
|
|
xsd_etree.xinclude()
|
|
populator = DefaultValuePopulator(xsd_etree)
|
|
|
|
xml_etree = etree.parse(xml_file, etree.XMLParser(remove_blank_text=True))
|
|
populator.transform(xml_etree)
|
|
|
|
xml_etree.write(out_file, pretty_print=True)
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Populate a given scenario XML with default values of nonexistent nodes")
|
|
parser.add_argument("xsd", help="Path to the schema of scenario XMLs")
|
|
parser.add_argument("xml", help="Path to the scenario XML file from users")
|
|
parser.add_argument("out", nargs="?", default="out.xml", help="Path where the output is placed")
|
|
args = parser.parse_args()
|
|
|
|
main(args.xsd, args.xml, args.out)
|