mirror of
https://github.com/open62541/open62541.git
synced 2025-06-03 04:00:21 +00:00
60 lines
2.7 KiB
Python
Executable File
60 lines
2.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
### This Source Code Form is subject to the terms of the Mozilla Public
|
|
### License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
### file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
### Copyright 2023 (c) Fraunhofer IOSB (Author: Noel Graf)
|
|
|
|
import xml.dom.minidom as dom
|
|
import codecs
|
|
import re
|
|
import base64
|
|
import argparse
|
|
|
|
###############################
|
|
# Parse the Command Line Input#
|
|
###############################
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('-x', '--xml',
|
|
metavar="<nodeSetXML>",
|
|
dest="xmlfile",
|
|
help='NodeSet XML file with nodes that shall be generated.')
|
|
|
|
parser.add_argument('outputFile',
|
|
metavar='<outputFile>',
|
|
help='output file w/o extension')
|
|
|
|
args = parser.parse_args()
|
|
|
|
# Extract the BSD Blob from the XML file.
|
|
nodeset_base = open(args.xmlfile, "rb")
|
|
fileContent = nodeset_base.read()
|
|
# Remove BOM since the dom parser cannot handle it on python 3 windows
|
|
if fileContent.startswith(codecs.BOM_UTF8):
|
|
fileContent = fileContent.lstrip(codecs.BOM_UTF8)
|
|
fileContent = fileContent.decode("utf-8")
|
|
|
|
# Remove the uax namespace from tags. UaModeler adds this namespace to some elements
|
|
fileContent = re.sub(r"<([/]?)uax:(.+?)([/]?)>", "<\\g<1>\\g<2>\\g<3>>", fileContent)
|
|
|
|
nodesets = dom.parseString(fileContent).getElementsByTagName("UANodeSet")
|
|
if len(nodesets) == 0 or len(nodesets) > 1:
|
|
raise Exception("contains no or more then 1 nodeset")
|
|
nodeset = nodesets[0]
|
|
variableNodes = nodeset.getElementsByTagName("UAVariable")
|
|
for nd in variableNodes:
|
|
if (nd.hasAttribute("SymbolicName") and (re.match(r".*_BinarySchema", nd.attributes["SymbolicName"].nodeValue) or nd.attributes["SymbolicName"].nodeValue == "TypeDictionary_BinarySchema")) or (nd.hasAttribute("ParentNodeId") and not nd.hasAttribute("SymbolicName") and re.fullmatch(r"i=93", nd.attributes["ParentNodeId"].nodeValue)):
|
|
type_content = nd.getElementsByTagName("Value")[0].getElementsByTagName("ByteString")[0]
|
|
with open(args.outputFile, 'w') as f:
|
|
f.write(base64.b64decode(type_content.firstChild.nodeValue).decode("utf-8"))
|
|
elif nd.hasAttribute("BrowseName") and nd.getAttribute("BrowseName").endswith("TypeDictionary"):
|
|
references = nd.getElementsByTagName("Reference")
|
|
for ref in references:
|
|
if ref.getAttribute("ReferenceType") == "HasComponent" and ref.firstChild.nodeValue == "i=93":
|
|
type_content = nd.getElementsByTagName("Value")[0].getElementsByTagName("ByteString")[0]
|
|
with open(args.outputFile, 'w') as f:
|
|
f.write(base64.b64decode(type_content.firstChild.nodeValue).decode("utf-8"))
|
|
break
|