audk/Tools/Python/XmlRoutines.py

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('&quot;','"').replace('&gt;','>'))
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