mirror of https://github.com/acidanthera/audk.git
142 lines
4.4 KiB
Python
Executable File
142 lines
4.4 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
# Copyright (c) 2007, Intel Corporation
|
|
# All rights reserved. This program and the accompanying materials
|
|
# are licensed and made available under the terms and conditions of the BSD License
|
|
# which accompanies this distribution. The full text of the license may be found at
|
|
# http://opensource.org/licenses/bsd-license.php
|
|
#
|
|
# THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
|
|
|
|
"""This is an XML API that uses a syntax similar to XPath, but it is written in
|
|
standard python so that no extra python packages are required to use it."""
|
|
|
|
import xml.dom.minidom
|
|
|
|
def XmlList(Dom, String):
|
|
"""Get a list of XML Elements using XPath style syntax."""
|
|
if Dom.nodeType==Dom.DOCUMENT_NODE:
|
|
return XmlList(Dom.documentElement, String)
|
|
if String[0] == "/":
|
|
return XmlList(Dom, String[1:])
|
|
if String == "" :
|
|
return []
|
|
TagList = String.split('/')
|
|
nodes = []
|
|
if Dom.nodeType == Dom.ELEMENT_NODE and Dom.tagName.strip() == TagList[0]:
|
|
if len(TagList) == 1:
|
|
nodes = [Dom]
|
|
else:
|
|
restOfPath = "/".join(TagList[1:])
|
|
for child in Dom.childNodes:
|
|
nodes = nodes + XmlList(child, restOfPath)
|
|
return nodes
|
|
|
|
def XmlElement (Dom, String):
|
|
"""Return a single element that matches the String which is XPath style syntax."""
|
|
try:
|
|
return XmlList (Dom, String)[0].firstChild.data.strip()
|
|
except:
|
|
return ''
|
|
|
|
def XmlElementData (Dom):
|
|
"""Get the text for this element."""
|
|
return Dom.firstChild.data.strip()
|
|
|
|
def XmlAttribute (Dom, AttName):
|
|
"""Return a single attribute named AttName."""
|
|
try:
|
|
return Dom.getAttribute(AttName)
|
|
except:
|
|
return ''
|
|
|
|
def XmlTopTag(Dom):
|
|
"""Return the name of the Root or top tag in the XML tree."""
|
|
return Dom.firstChild.nodeName
|
|
|
|
def XmlParseFile (FileName):
|
|
"""Parse an XML file into a DOM and return the DOM."""
|
|
try:
|
|
f = open(FileName, 'r')
|
|
Dom = xml.dom.minidom.parse(f)
|
|
f.close()
|
|
return Dom
|
|
except:
|
|
return xml.dom.minidom.parseString('<empty/>')
|
|
|
|
def XmlParseFileSection (FileName, Tag):
|
|
"""Parse a section of an XML file into a DOM(Document Object Model) and return the DOM."""
|
|
try:
|
|
f = open(FileName, 'r')
|
|
except:
|
|
return xml.dom.minidom.parseString('<empty/>')
|
|
Start = '<' + Tag
|
|
End = '</' + Tag + '>'
|
|
File = ''
|
|
while File.find(Start) < 0 or File.find(End) < 0:
|
|
Section = f.read(0x1000)
|
|
if Section == '':
|
|
break
|
|
File += Section
|
|
f.close()
|
|
if File.find(Start) < 0 or File.find(End) < 0:
|
|
return xml.dom.minidom.parseString('<empty/>')
|
|
File = File[File.find(Start):File.find(End)+len(End)]
|
|
try:
|
|
return xml.dom.minidom.parseString(File)
|
|
except:
|
|
return xml.dom.minidom.parseString('<empty/>')
|
|
|
|
def XmlSaveFile (Dom, FileName, Encoding='UTF-8'):
|
|
"""Save a DOM(Document Object Model) into an XML file."""
|
|
try:
|
|
f = open(FileName, 'w')
|
|
f.write(Dom.toxml(Encoding).replace('"','"').replace('>','>'))
|
|
f.close()
|
|
return True
|
|
except:
|
|
return False
|
|
|
|
def XmlRemoveElement(Node):
|
|
"""Remove an element node from DOM(Document Object Model) tree."""
|
|
ParentNode = Node.parentNode
|
|
if ParentNode == None:
|
|
return False
|
|
PreviousSibling = Node.previousSibling
|
|
while PreviousSibling != None and PreviousSibling.nodeType == PreviousSibling.TEXT_NODE and PreviousSibling.data.strip() == '':
|
|
Temp = PreviousSibling
|
|
PreviousSibling = PreviousSibling.previousSibling
|
|
ParentNode.removeChild(Temp)
|
|
ParentNode.removeChild(Node)
|
|
return True
|
|
|
|
def XmlAppendChildElement(ParentNode, TagName, ElementText='', AttributeDictionary = {}):
|
|
"""Add a child element to a DOM(Document Object Model) tree with optional Attributes."""
|
|
TagName = TagName.strip()
|
|
if TagName == '':
|
|
return False
|
|
Depth = 0
|
|
Dom = ParentNode
|
|
while Dom != None and Dom.nodeType != Dom.DOCUMENT_NODE:
|
|
Dom = Dom.parentNode
|
|
Depth += 1
|
|
if Dom == None:
|
|
return False
|
|
ParentNode.appendChild(Dom.createTextNode('\n%*s' % (Depth * 2, '')))
|
|
ElementNode = Dom.createElement(TagName)
|
|
if ElementText != '':
|
|
ElementNode.appendChild(Dom.createTextNode(ElementText))
|
|
for Item in AttributeDictionary:
|
|
ElementNode.setAttribute(Item, AttributeDictionary[Item])
|
|
ParentNode.appendChild(ElementNode)
|
|
return True
|
|
|
|
|
|
# This acts like the main() function for the script, unless it is 'import'ed into another
|
|
# script.
|
|
if __name__ == '__main__':
|
|
|
|
# Nothing to do here. Could do some unit tests.
|
|
pass
|