mirror of https://github.com/python/cpython
154 lines
3.9 KiB
Python
154 lines
3.9 KiB
Python
# Wrapper module for _elementtree
|
|
|
|
from xml.etree.ElementTree import (ElementTree, dump, iselement, QName,
|
|
fromstringlist,
|
|
tostring, tostringlist, VERSION)
|
|
# These ones are not in ElementTree.__all__
|
|
from xml.etree.ElementTree import ElementPath, register_namespace
|
|
|
|
# Import the C accelerators:
|
|
# Element, SubElement, TreeBuilder, XMLParser, ParseError
|
|
from _elementtree import *
|
|
|
|
|
|
class ElementTree(ElementTree):
|
|
|
|
def parse(self, source, parser=None):
|
|
close_source = False
|
|
if not hasattr(source, 'read'):
|
|
source = open(source, 'rb')
|
|
close_source = True
|
|
try:
|
|
if parser is not None:
|
|
while True:
|
|
data = source.read(65536)
|
|
if not data:
|
|
break
|
|
parser.feed(data)
|
|
self._root = parser.close()
|
|
else:
|
|
parser = XMLParser()
|
|
self._root = parser._parse(source)
|
|
return self._root
|
|
finally:
|
|
if close_source:
|
|
source.close()
|
|
|
|
|
|
class iterparse:
|
|
root = None
|
|
|
|
def __init__(self, file, events=None):
|
|
self._close_file = False
|
|
if not hasattr(file, 'read'):
|
|
file = open(file, 'rb')
|
|
self._close_file = True
|
|
self._file = file
|
|
self._events = []
|
|
self._index = 0
|
|
self._error = None
|
|
self.root = self._root = None
|
|
b = TreeBuilder()
|
|
self._parser = XMLParser(b)
|
|
self._parser._setevents(self._events, events)
|
|
|
|
def __next__(self):
|
|
while True:
|
|
try:
|
|
item = self._events[self._index]
|
|
self._index += 1
|
|
return item
|
|
except IndexError:
|
|
pass
|
|
if self._error:
|
|
e = self._error
|
|
self._error = None
|
|
raise e
|
|
if self._parser is None:
|
|
self.root = self._root
|
|
if self._close_file:
|
|
self._file.close()
|
|
raise StopIteration
|
|
# load event buffer
|
|
del self._events[:]
|
|
self._index = 0
|
|
data = self._file.read(16384)
|
|
if data:
|
|
try:
|
|
self._parser.feed(data)
|
|
except SyntaxError as exc:
|
|
self._error = exc
|
|
else:
|
|
self._root = self._parser.close()
|
|
self._parser = None
|
|
|
|
def __iter__(self):
|
|
return self
|
|
|
|
|
|
# =============================================================================
|
|
#
|
|
# Everything below this line can be removed
|
|
# after cElementTree is folded behind ElementTree.
|
|
#
|
|
# =============================================================================
|
|
|
|
from xml.etree.ElementTree import Comment as _Comment, PI as _PI
|
|
|
|
|
|
def parse(source, parser=None):
|
|
tree = ElementTree()
|
|
tree.parse(source, parser)
|
|
return tree
|
|
|
|
|
|
def XML(text, parser=None):
|
|
if not parser:
|
|
parser = XMLParser()
|
|
parser = XMLParser()
|
|
parser.feed(text)
|
|
return parser.close()
|
|
|
|
|
|
def XMLID(text, parser=None):
|
|
tree = XML(text, parser=parser)
|
|
ids = {}
|
|
for elem in tree.iter():
|
|
id = elem.get('id')
|
|
if id:
|
|
ids[id] = elem
|
|
return tree, ids
|
|
|
|
|
|
class CommentProxy:
|
|
|
|
def __call__(self, text=None):
|
|
element = Element(_Comment)
|
|
element.text = text
|
|
return element
|
|
|
|
def __eq__(self, other):
|
|
return _Comment == other
|
|
|
|
|
|
class PIProxy:
|
|
|
|
def __call__(self, target, text=None):
|
|
element = Element(_PI)
|
|
element.text = target
|
|
if text:
|
|
element.text = element.text + ' ' + text
|
|
return element
|
|
|
|
def __eq__(self, other):
|
|
return _PI == other
|
|
|
|
|
|
Comment = CommentProxy()
|
|
PI = ProcessingInstruction = PIProxy()
|
|
del CommentProxy, PIProxy
|
|
|
|
# Aliases
|
|
fromstring = XML
|
|
XMLTreeBuilder = XMLParser
|