u"""The ``lxml.etree`` module implements the extended ElementTree API for XML. """ __docformat__ = u"restructuredtext en" __all__ = [ 'AttributeBasedElementClassLookup', 'C14NError', 'CDATA', 'Comment', 'CommentBase', 'CustomElementClassLookup', 'DEBUG', 'DTD', 'DTDError', 'DTDParseError', 'DTDValidateError', 'DocumentInvalid', 'ETCompatXMLParser', 'ETXPath', 'Element', 'ElementBase', 'ElementClassLookup', 'ElementDefaultClassLookup', 'ElementNamespaceClassLookup', 'ElementTree', 'Entity', 'EntityBase', 'Error', 'ErrorDomains', 'ErrorLevels', 'ErrorTypes', 'Extension', 'FallbackElementClassLookup', 'FunctionNamespace', 'HTML', 'HTMLParser', 'LIBXML_COMPILED_VERSION', 'LIBXML_VERSION', 'LIBXSLT_COMPILED_VERSION', 'LIBXSLT_VERSION', 'LXML_VERSION', 'LxmlError', 'LxmlRegistryError', 'LxmlSyntaxError', 'NamespaceRegistryError', 'PI', 'PIBase', 'ParseError', 'ParserBasedElementClassLookup', 'ParserError', 'ProcessingInstruction', 'PyErrorLog', 'PythonElementClassLookup', 'QName', 'RelaxNG', 'RelaxNGError', 'RelaxNGErrorTypes', 'RelaxNGParseError', 'RelaxNGValidateError', 'Resolver', 'Schematron', 'SchematronError', 'SchematronParseError', 'SchematronValidateError', 'SerialisationError', 'SubElement', 'TreeBuilder', 'XInclude', 'XIncludeError', 'XML', 'XMLDTDID', 'XMLID', 'XMLParser', 'XMLSchema', 'XMLSchemaError', 'XMLSchemaParseError', 'XMLSchemaValidateError', 'XMLSyntaxError', 'XMLTreeBuilder', 'XPath', 'XPathDocumentEvaluator', 'XPathError', 'XPathEvalError', 'XPathEvaluator', 'XPathFunctionError', 'XPathResultError', 'XPathSyntaxError', 'XSLT', 'XSLTAccessControl', 'XSLTApplyError', 'XSLTError', 'XSLTExtension', 'XSLTExtensionError', 'XSLTParseError', 'XSLTSaveError', 'cleanup_namespaces', 'clear_error_log', 'dump', 'fromstring', 'fromstringlist', 'get_default_parser', 'iselement', 'iterparse', 'iterwalk', 'parse', 'parseid', 'register_namespace', 'set_default_parser', 'set_element_class_lookup', 'strip_attributes', 'strip_elements', 'strip_tags', 'tostring', 'tostringlist', 'tounicode', 'use_global_python_log' ] cimport tree, python, config from tree cimport xmlDoc, xmlNode, xmlAttr, xmlNs, _isElement, _getNs from python cimport callable, _cstr, _isString cimport xpath cimport c14n cimport cstd try: import __builtin__ except ImportError: # Python 3 import builtins as __builtin__ cdef object _unicode try: _unicode = __builtin__.unicode except AttributeError: # Python 3 _unicode = __builtin__.str del __builtin__ cdef object os_path_abspath from os.path import abspath as os_path_abspath cdef object BytesIO, StringIO try: from io import BytesIO, StringIO except (ImportError, AttributeError): from StringIO import StringIO, StringIO as BytesIO cdef object _elementpath import _elementpath cdef object sys import sys cdef object re import re cdef object gzip import gzip cdef object ITER_EMPTY ITER_EMPTY = iter(()) cdef object EMPTY_READ_ONLY_DICT EMPTY_READ_ONLY_DICT = python.PyDictProxy_New({}) # the rules # --------- # any libxml C argument/variable is prefixed with c_ # any non-public function/class is prefixed with an underscore # instance creation is always through factories # what to do with libxml2/libxslt error messages? # 0 : drop # 1 : use log DEF __DEBUG = 1 # maximum number of lines in the libxml2/xslt log if __DEBUG == 1 DEF __MAX_LOG_SIZE = 100 # make the compiled-in debug state publicly available DEBUG = __DEBUG # global per-thread setup tree.xmlThrDefIndentTreeOutput(1) tree.xmlThrDefLineNumbersDefaultValue(1) _initThreadLogging() # initialize parser (and threading) xmlparser.xmlInitParser() # filename encoding cdef object _FILENAME_ENCODING _FILENAME_ENCODING = sys.getfilesystemencoding() if _FILENAME_ENCODING is None: _FILENAME_ENCODING = sys.getdefaultencoding() if _FILENAME_ENCODING is None: _FILENAME_ENCODING = b'ascii' else: _FILENAME_ENCODING = _FILENAME_ENCODING.encode(u"UTF-8") cdef char* _C_FILENAME_ENCODING _C_FILENAME_ENCODING = _cstr(_FILENAME_ENCODING) # set up some default namespace prefixes cdef object _DEFAULT_NAMESPACE_PREFIXES _DEFAULT_NAMESPACE_PREFIXES = { b"http://www.w3.org/XML/1998/namespace": b'xml', b"http://www.w3.org/1999/xhtml": b"html", b"http://www.w3.org/1999/XSL/Transform": b"xsl", b"http://www.w3.org/1999/02/22-rdf-syntax-ns#": b"rdf", b"http://schemas.xmlsoap.org/wsdl/": b"wsdl", # xml schema b"http://www.w3.org/2001/XMLSchema": b"xs", b"http://www.w3.org/2001/XMLSchema-instance": b"xsi", # dublin core b"http://purl.org/dc/elements/1.1/": b"dc", # objectify b"http://codespeak.net/lxml/objectify/pytype" : b"py", } cdef object _check_internal_prefix = re.compile(b"ns\d+$").match def register_namespace(prefix, uri): u"""Registers a namespace prefix that newly created Elements in that namespace will use. The registry is global, and any existing mapping for either the given prefix or the namespace URI will be removed. """ prefix_utf, uri_utf = _utf8(prefix), _utf8(uri) if _check_internal_prefix(prefix_utf): raise ValueError("Prefix format reserved for internal use") _tagValidOrRaise(prefix_utf) _uriValidOrRaise(uri_utf) for k, v in _DEFAULT_NAMESPACE_PREFIXES.items(): if k == uri_utf or v == prefix_utf: del _DEFAULT_NAMESPACE_PREFIXES[k] _DEFAULT_NAMESPACE_PREFIXES[uri_utf] = prefix_utf # Error superclass for ElementTree compatibility class Error(Exception): pass # module level superclass for all exceptions class LxmlError(Error): u"""Main exception base class for lxml. All other exceptions inherit from this one. """ def __init__(self, message, error_log=None): if python.PY_VERSION_HEX >= 0x02050000: # Python >= 2.5 uses new style class exceptions super(_Error, self).__init__(message) else: error_super_init(self, message) if error_log is None: self.error_log = __copyGlobalErrorLog() else: self.error_log = error_log.copy() cdef object _Error = Error if python.PY_VERSION_HEX >= 0x02050000 else None cdef object error_super_init = Error.__init__ if python.PY_VERSION_HEX < 0x02050000 else None # superclass for all syntax errors class LxmlSyntaxError(LxmlError, SyntaxError): u"""Base class for all syntax errors. """ pass class C14NError(LxmlError): u"""Error during C14N serialisation. """ pass # version information cdef __unpackDottedVersion(version): cdef list version_list = [] l = (version.decode(u"ASCII").replace(u'-', u'.').split(u'.') + [0]*4)[:4] for item in l: try: item = int(item) except ValueError: if item.startswith(u'dev'): count = item[3:] item = -300 elif item.startswith(u'alpha'): count = item[5:] item = -200 elif item.startswith(u'beta'): count = item[4:] item = -100 else: count = 0 if count: item = item + int(count) version_list.append(item) return tuple(version_list) cdef __unpackIntVersion(int c_version): return ( ((c_version / (100*100)) % 100), ((c_version / 100) % 100), (c_version % 100) ) cdef int _LIBXML_VERSION_INT try: _LIBXML_VERSION_INT = int( re.match(u'[0-9]+', (tree.xmlParserVersion).decode(u"ASCII")).group(0)) except Exception: print u"Unknown libxml2 version: %s" % (tree.xmlParserVersion).decode(u"ASCII") _LIBXML_VERSION_INT = 0 LIBXML_VERSION = __unpackIntVersion(_LIBXML_VERSION_INT) LIBXML_COMPILED_VERSION = __unpackIntVersion(tree.LIBXML_VERSION) LXML_VERSION = __unpackDottedVersion(tree.LXML_VERSION_STRING) __version__ = (tree.LXML_VERSION_STRING).decode(u"ASCII") # class for temporary storage of Python references, # used e.g. for XPath results cdef class _TempStore: cdef list _storage def __init__(self): self._storage = [] cdef int add(self, obj) except -1: self._storage.append(obj) return 0 cdef int clear(self) except -1: del self._storage[:] return 0 # class for temporarily storing exceptions raised in extensions cdef class _ExceptionContext: cdef object _exc_info cdef void clear(self): self._exc_info = None cdef void _store_raised(self): self._exc_info = sys.exc_info() cdef void _store_exception(self, exception): self._exc_info = (exception, None, None) cdef bint _has_raised(self): return self._exc_info is not None cdef int _raise_if_stored(self) except -1: if self._exc_info is None: return 0 type, value, traceback = self._exc_info self._exc_info = None if value is None and traceback is None: raise type else: raise type, value, traceback # forward declarations cdef public class _Document [ type LxmlDocumentType, object LxmlDocument ] cdef public class _Element [ type LxmlElementType, object LxmlElement ] cdef class _BaseParser cdef class QName ctypedef public xmlNode* (*_node_to_node_function)(xmlNode*) ################################################################################ # Include submodules include "proxy.pxi" # Proxy handling (element backpointers/memory/etc.) include "apihelpers.pxi" # Private helper functions include "xmlerror.pxi" # Error and log handling ################################################################################ # Public Python API cdef public class _Document [ type LxmlDocumentType, object LxmlDocument ]: u"""Internal base class to reference a libxml document. When instances of this class are garbage collected, the libxml document is cleaned up. """ cdef int _ns_counter cdef bytes _prefix_tail cdef xmlDoc* _c_doc cdef _BaseParser _parser def __dealloc__(self): # if there are no more references to the document, it is safe # to clean the whole thing up, as all nodes have a reference to # the document tree.xmlFreeDoc(self._c_doc) cdef getroot(self): # return an element proxy for the document root cdef xmlNode* c_node c_node = tree.xmlDocGetRootElement(self._c_doc) if c_node is NULL: return None return _elementFactory(self, c_node) cdef bint hasdoctype(self): # DOCTYPE gets parsed into internal subset (xmlDTD*) return self._c_doc is not NULL and self._c_doc.intSubset is not NULL cdef getdoctype(self): # get doctype info: root tag, public/system ID (or None if not known) cdef tree.xmlDtd* c_dtd cdef xmlNode* c_root_node public_id = None sys_url = None c_dtd = self._c_doc.intSubset if c_dtd is not NULL: if c_dtd.ExternalID is not NULL: public_id = funicode(c_dtd.ExternalID) if c_dtd.SystemID is not NULL: sys_url = funicode(c_dtd.SystemID) c_dtd = self._c_doc.extSubset if c_dtd is not NULL: if not public_id and c_dtd.ExternalID is not NULL: public_id = funicode(c_dtd.ExternalID) if not sys_url and c_dtd.SystemID is not NULL: sys_url = funicode(c_dtd.SystemID) c_root_node = tree.xmlDocGetRootElement(self._c_doc) if c_root_node is NULL: root_name = None else: root_name = funicode(c_root_node.name) return (root_name, public_id, sys_url) cdef getxmlinfo(self): # return XML version and encoding (or None if not known) cdef xmlDoc* c_doc = self._c_doc if c_doc.version is NULL: version = None else: version = funicode(c_doc.version) if c_doc.encoding is NULL: encoding = None else: encoding = funicode(c_doc.encoding) return (version, encoding) cdef isstandalone(self): # returns True for "standalone=true", # False for "standalone=false", None if not provided if self._c_doc.standalone == -1: return None else: return (self._c_doc.standalone == 1) cdef bytes buildNewPrefix(self): # get a new unique prefix ("nsX") for this document cdef bytes ns if self._ns_counter < len(_PREFIX_CACHE): ns = _PREFIX_CACHE[self._ns_counter] else: ns = python.PyBytes_FromFormat("ns%d", self._ns_counter) if self._prefix_tail is not None: ns += self._prefix_tail self._ns_counter += 1 if self._ns_counter < 0: # overflow! self._ns_counter = 0 if self._prefix_tail is None: self._prefix_tail = b"A" else: self._prefix_tail += b"A" return ns cdef xmlNs* _findOrBuildNodeNs(self, xmlNode* c_node, char* c_href, char* c_prefix, bint is_attribute) except NULL: u"""Get or create namespace structure for a node. Reuses the prefix if possible. """ cdef xmlNs* c_ns cdef xmlNs* c_doc_ns cdef python.PyObject* dict_result if c_node.type != tree.XML_ELEMENT_NODE: assert c_node.type == tree.XML_ELEMENT_NODE, \ u"invalid node type %d, expected %d" % ( c_node.type, tree.XML_ELEMENT_NODE) # look for existing ns declaration c_ns = _searchNsByHref(c_node, c_href, is_attribute) if c_ns is not NULL: if is_attribute and c_ns.prefix is NULL: # do not put namespaced attributes into the default # namespace as this would break serialisation pass else: return c_ns # none found => determine a suitable new prefix if c_prefix is NULL: dict_result = python.PyDict_GetItem( _DEFAULT_NAMESPACE_PREFIXES, c_href) if dict_result is not NULL: prefix = dict_result else: prefix = self.buildNewPrefix() c_prefix = _cstr(prefix) # make sure the prefix is not in use already while tree.xmlSearchNs(self._c_doc, c_node, c_prefix) is not NULL: prefix = self.buildNewPrefix() c_prefix = _cstr(prefix) # declare the namespace and return it c_ns = tree.xmlNewNs(c_node, c_href, c_prefix) if c_ns is NULL: python.PyErr_NoMemory() return c_ns cdef int _setNodeNs(self, xmlNode* c_node, char* href) except -1: u"Lookup namespace structure and set it for the node." cdef xmlNs* c_ns c_ns = self._findOrBuildNodeNs(c_node, href, NULL, 0) tree.xmlSetNs(c_node, c_ns) cdef tuple __initPrefixCache(): cdef int i return tuple([ python.PyBytes_FromFormat("ns%d", i) for i in range(30) ]) cdef tuple _PREFIX_CACHE _PREFIX_CACHE = __initPrefixCache() cdef extern from "etree_defs.h": # macro call to 't->tp_new()' for fast instantiation cdef _Document NEW_DOCUMENT "PY_NEW" (object t) cdef _Document _documentFactory(xmlDoc* c_doc, _BaseParser parser): cdef _Document result result = NEW_DOCUMENT(_Document) result._c_doc = c_doc result._ns_counter = 0 result._prefix_tail = None if parser is None: parser = __GLOBAL_PARSER_CONTEXT.getDefaultParser() result._parser = parser return result cdef class DocInfo: u"Document information provided by parser and DTD." cdef _Document _doc def __cinit__(self, tree): u"Create a DocInfo object for an ElementTree object or root Element." self._doc = _documentOrRaise(tree) root_name, public_id, system_url = self._doc.getdoctype() if not root_name and (public_id or system_url): raise ValueError, u"Could not find root node" property root_name: u"Returns the name of the root node as defined by the DOCTYPE." def __get__(self): root_name, public_id, system_url = self._doc.getdoctype() return root_name property public_id: u"Returns the public ID of the DOCTYPE." def __get__(self): root_name, public_id, system_url = self._doc.getdoctype() return public_id property system_url: u"Returns the system ID of the DOCTYPE." def __get__(self): root_name, public_id, system_url = self._doc.getdoctype() return system_url property xml_version: u"Returns the XML version as declared by the document." def __get__(self): xml_version, encoding = self._doc.getxmlinfo() return xml_version property encoding: u"Returns the encoding name as declared by the document." def __get__(self): xml_version, encoding = self._doc.getxmlinfo() return encoding property standalone: u"""Returns the standalone flag as declared by the document. The possible values are True (``standalone='yes'``), False (``standalone='no'`` or flag not provided in the declaration), and None (unknown or no declaration found). Note that a normal truth test on this value will always tell if the ``standalone`` flag was set to ``'yes'`` or not. """ def __get__(self): return self._doc.isstandalone() property URL: u"The source URL of the document (or None if unknown)." def __get__(self): if self._doc._c_doc.URL is NULL: return None return _decodeFilename(self._doc._c_doc.URL) def __set__(self, url): cdef char* c_oldurl url = _encodeFilename(url) c_oldurl = self._doc._c_doc.URL if url is None: self._doc._c_doc.URL = NULL else: self._doc._c_doc.URL = tree.xmlStrdup(_cstr(url)) if c_oldurl is not NULL: tree.xmlFree(c_oldurl) property doctype: u"Returns a DOCTYPE declaration string for the document." def __get__(self): root_name, public_id, system_url = self._doc.getdoctype() if public_id: if system_url: return u'' % ( root_name, public_id, system_url) else: return u'' % ( root_name, public_id) elif system_url: return u'' % ( root_name, system_url) elif self._doc.hasdoctype(): return u'' % root_name else: return u"" property internalDTD: u"Returns a DTD validator based on the internal subset of the document." def __get__(self): return _dtdFactory(self._doc._c_doc.intSubset) property externalDTD: u"Returns a DTD validator based on the external subset of the document." def __get__(self): return _dtdFactory(self._doc._c_doc.extSubset) cdef public class _Element [ type LxmlElementType, object LxmlElement ]: u"""Element class. References a document object and a libxml node. By pointing to a Document instance, a reference is kept to _Document as long as there is some pointer to a node in it. """ cdef python.PyObject* _gc_doc cdef _Document _doc cdef xmlNode* _c_node cdef object _tag def _init(self): u"""_init(self) Called after object initialisation. Custom subclasses may override this if they recursively call _init() in the superclasses. """ def __dealloc__(self): #print "trying to free node:", self._c_node #displayNode(self._c_node, 0) if self._c_node is not NULL: _unregisterProxy(self) attemptDeallocation(self._c_node) _releaseProxy(self) # MANIPULATORS def __setitem__(self, x, value): u"""__setitem__(self, x, value) Replaces the given subelement index or slice. """ cdef xmlNode* c_node cdef xmlNode* c_next cdef xmlDoc* c_source_doc cdef _Element element cdef bint left_to_right cdef Py_ssize_t slicelength, step _assertValidNode(self) if value is None: raise ValueError, u"cannot assign None" if python.PySlice_Check(x): # slice assignment _findChildSlice(x, self._c_node, &c_node, &step, &slicelength) if step > 0: left_to_right = 1 else: left_to_right = 0 step = -step _replaceSlice(self, c_node, slicelength, step, left_to_right, value) return else: # otherwise: normal item assignment element = value _assertValidNode(element) c_node = _findChild(self._c_node, x) if c_node is NULL: raise IndexError, u"list index out of range" c_source_doc = element._c_node.doc c_next = element._c_node.next _removeText(c_node.next) tree.xmlReplaceNode(c_node, element._c_node) _moveTail(c_next, element._c_node) moveNodeToDocument(self._doc, c_source_doc, element._c_node) if not attemptDeallocation(c_node): moveNodeToDocument(self._doc, c_node.doc, c_node) def __delitem__(self, x): u"""__delitem__(self, x) Deletes the given subelement or a slice. """ cdef xmlNode* c_node cdef xmlNode* c_next cdef Py_ssize_t step, slicelength _assertValidNode(self) if python.PySlice_Check(x): # slice deletion if _isFullSlice(x): c_node = self._c_node.children if c_node is not NULL: if not _isElement(c_node): c_node = _nextElement(c_node) while c_node is not NULL: c_next = _nextElement(c_node) _removeNode(self._doc, c_node) c_node = c_next else: _findChildSlice(x, self._c_node, &c_node, &step, &slicelength) _deleteSlice(self._doc, c_node, slicelength, step) else: # item deletion c_node = _findChild(self._c_node, x) if c_node is NULL: raise IndexError, u"index out of range: %d" % x _removeText(c_node.next) _removeNode(self._doc, c_node) def __deepcopy__(self, memo): u"__deepcopy__(self, memo)" return self.__copy__() def __copy__(self): u"__copy__(self)" cdef xmlDoc* c_doc cdef xmlNode* c_node cdef _Document new_doc _assertValidNode(self) c_doc = _copyDocRoot(self._doc._c_doc, self._c_node) # recursive new_doc = _documentFactory(c_doc, self._doc._parser) root = new_doc.getroot() if root is not None: return root # Comment/PI c_node = c_doc.children while c_node is not NULL and c_node.type != self._c_node.type: c_node = c_node.next if c_node is NULL: return None return _elementFactory(new_doc, c_node) def set(self, key, value): u"""set(self, key, value) Sets an element attribute. """ _assertValidNode(self) _setAttributeValue(self, key, value) def append(self, _Element element not None): u"""append(self, element) Adds a subelement to the end of this element. """ _assertValidNode(self) _assertValidNode(element) _appendChild(self, element) def addnext(self, _Element element not None): u"""addnext(self, element) Adds the element as a following sibling directly after this element. This is normally used to set a processing instruction or comment after the root node of a document. Note that tail text is automatically discarded when adding at the root level. """ _assertValidNode(self) _assertValidNode(element) if self._c_node.parent != NULL and not _isElement(self._c_node.parent): if element._c_node.type != tree.XML_PI_NODE: if element._c_node.type != tree.XML_COMMENT_NODE: raise TypeError, u"Only processing instructions and comments can be siblings of the root element" element.tail = None _appendSibling(self, element) def addprevious(self, _Element element not None): u"""addprevious(self, element) Adds the element as a preceding sibling directly before this element. This is normally used to set a processing instruction or comment before the root node of a document. Note that tail text is automatically discarded when adding at the root level. """ _assertValidNode(self) _assertValidNode(element) if self._c_node.parent != NULL and not _isElement(self._c_node.parent): if element._c_node.type != tree.XML_PI_NODE: if element._c_node.type != tree.XML_COMMENT_NODE: raise TypeError, u"Only processing instructions and comments can be siblings of the root element" element.tail = None _prependSibling(self, element) def extend(self, elements): u"""extend(self, elements) Extends the current children by the elements in the iterable. """ cdef _Element element _assertValidNode(self) for element in elements: if element is None: raise TypeError, u"Node must not be None" _assertValidNode(element) _appendChild(self, element) def clear(self): u"""clear(self) Resets an element. This function removes all subelements, clears all attributes and sets the text and tail properties to None. """ cdef xmlAttr* c_attr cdef xmlAttr* c_attr_next cdef xmlNode* c_node cdef xmlNode* c_node_next _assertValidNode(self) c_node = self._c_node # remove self.text and self.tail _removeText(c_node.children) _removeText(c_node.next) # remove all attributes c_attr = c_node.properties while c_attr is not NULL: c_attr_next = c_attr.next tree.xmlRemoveProp(c_attr) c_attr = c_attr_next # remove all subelements c_node = c_node.children if c_node is not NULL: if not _isElement(c_node): c_node = _nextElement(c_node) while c_node is not NULL: c_node_next = _nextElement(c_node) _removeNode(self._doc, c_node) c_node = c_node_next def insert(self, index, _Element element not None): u"""insert(self, index, element) Inserts a subelement at the given position in this element """ cdef xmlNode* c_node cdef xmlNode* c_next cdef xmlDoc* c_source_doc _assertValidNode(self) _assertValidNode(element) c_node = _findChild(self._c_node, index) if c_node is NULL: _appendChild(self, element) return c_source_doc = c_node.doc c_next = element._c_node.next tree.xmlAddPrevSibling(c_node, element._c_node) _moveTail(c_next, element._c_node) moveNodeToDocument(self._doc, c_source_doc, element._c_node) def remove(self, _Element element not None): u"""remove(self, element) Removes a matching subelement. Unlike the find methods, this method compares elements based on identity, not on tag value or contents. """ cdef xmlNode* c_node cdef xmlNode* c_next _assertValidNode(self) _assertValidNode(element) c_node = element._c_node if c_node.parent is not self._c_node: raise ValueError, u"Element is not a child of this node." c_next = element._c_node.next tree.xmlUnlinkNode(c_node) _moveTail(c_next, c_node) # fix namespace declarations moveNodeToDocument(self._doc, c_node.doc, c_node) def replace(self, _Element old_element not None, _Element new_element not None): u"""replace(self, old_element, new_element) Replaces a subelement with the element passed as second argument. """ cdef xmlNode* c_old_node cdef xmlNode* c_old_next cdef xmlNode* c_new_node cdef xmlNode* c_new_next cdef xmlDoc* c_source_doc _assertValidNode(self) _assertValidNode(old_element) _assertValidNode(new_element) c_old_node = old_element._c_node if c_old_node.parent is not self._c_node: raise ValueError, u"Element is not a child of this node." c_old_next = c_old_node.next c_new_node = new_element._c_node c_new_next = c_new_node.next c_source_doc = c_new_node.doc tree.xmlReplaceNode(c_old_node, c_new_node) _moveTail(c_new_next, c_new_node) _moveTail(c_old_next, c_old_node) moveNodeToDocument(self._doc, c_source_doc, c_new_node) # fix namespace declarations moveNodeToDocument(self._doc, c_old_node.doc, c_old_node) # PROPERTIES property tag: u"""Element tag """ def __get__(self): if self._tag is not None: return self._tag _assertValidNode(self) self._tag = _namespacedName(self._c_node) return self._tag def __set__(self, value): cdef _BaseParser parser _assertValidNode(self) ns, name = _getNsTag(value) parser = self._doc._parser if parser is not None and parser._for_html: _htmlTagValidOrRaise(name) else: _tagValidOrRaise(name) self._tag = value tree.xmlNodeSetName(self._c_node, _cstr(name)) if ns is None: self._c_node.ns = NULL else: self._doc._setNodeNs(self._c_node, _cstr(ns)) property attrib: u"""Element attribute dictionary. Where possible, use get(), set(), keys(), values() and items() to access element attributes. """ def __get__(self): _assertValidNode(self) return _Attrib(self) ## cdef python.PyObject* ref ## if self._attrib is not None: ## ref = python.PyWeakref_GET_OBJECT(self._attrib) ## if ref is not None: ## return ref ## attrib = _Attrib(self) ## self._attrib = python.PyWeakref_NewRef(attrib, NULL) ## return attrib property text: u"""Text before the first subelement. This is either a string or the value None, if there was no text. """ def __get__(self): _assertValidNode(self) return _collectText(self._c_node.children) def __set__(self, value): _assertValidNode(self) if isinstance(value, QName): value = python.PyUnicode_FromEncodedObject( _resolveQNameText(self, value), 'UTF-8', 'strict') _setNodeText(self._c_node, value) # using 'del el.text' is the wrong thing to do #def __del__(self): # _setNodeText(self._c_node, None) property tail: u"""Text after this element's end tag, but before the next sibling element's start tag. This is either a string or the value None, if there was no text. """ def __get__(self): _assertValidNode(self) return _collectText(self._c_node.next) def __set__(self, value): _assertValidNode(self) _setTailText(self._c_node, value) # using 'del el.tail' is the wrong thing to do #def __del__(self): # _setTailText(self._c_node, None) # not in ElementTree, read-only property prefix: u"""Namespace prefix or None. """ def __get__(self): if self._c_node.ns is not NULL: if self._c_node.ns.prefix is not NULL: return funicode(self._c_node.ns.prefix) return None # not in ElementTree, read-only property sourceline: u"""Original line number as found by the parser or None if unknown. """ def __get__(self): cdef long line _assertValidNode(self) line = tree.xmlGetLineNo(self._c_node) if line > 0: return line else: return None def __set__(self, line): _assertValidNode(self) if line < 0: self._c_node.line = 0 else: self._c_node.line = line # not in ElementTree, read-only property nsmap: u"""Namespace prefix->URI mapping known in the context of this Element. This includes all namespace declarations of the parents. Note that changing the returned dict has no effect on the Element. """ def __get__(self): cdef xmlNode* c_node cdef xmlNs* c_ns cdef dict nsmap = {} _assertValidNode(self) c_node = self._c_node while c_node is not NULL and c_node.type == tree.XML_ELEMENT_NODE: c_ns = c_node.nsDef while c_ns is not NULL: if c_ns.prefix is NULL: prefix = None else: prefix = funicode(c_ns.prefix) if prefix not in nsmap: nsmap[prefix] = funicode(c_ns.href) c_ns = c_ns.next c_node = c_node.parent return nsmap # not in ElementTree, read-only property base: u"""The base URI of the Element (xml:base or HTML base URL). None if the base URI is unknown. Note that the value depends on the URL of the document that holds the Element if there is no xml:base attribute on the Element or its ancestors. Setting this property will set an xml:base attribute on the Element, regardless of the document type (XML or HTML). """ def __get__(self): cdef char* c_base _assertValidNode(self) c_base = tree.xmlNodeGetBase(self._doc._c_doc, self._c_node) if c_base is NULL: if self._doc._c_doc.URL is NULL: return None return _decodeFilename(self._doc._c_doc.URL) base = _decodeFilename(c_base) tree.xmlFree(c_base) return base def __set__(self, url): cdef char* c_base _assertValidNode(self) if url is None: c_base = NULL else: url = _encodeFilename(url) c_base = _cstr(url) tree.xmlNodeSetBase(self._c_node, c_base) # ACCESSORS def __repr__(self): u"__repr__(self)" return u"" % (self.tag, id(self)) def __getitem__(self, x): u"""Returns the subelement at the given position or the requested slice. """ cdef xmlNode* c_node cdef Py_ssize_t step, slicelength cdef Py_ssize_t c, i cdef _node_to_node_function next_element cdef list result _assertValidNode(self) if python.PySlice_Check(x): # slicing if _isFullSlice(x): return _collectChildren(self) _findChildSlice(x, self._c_node, &c_node, &step, &slicelength) if c_node is NULL: return [] if step > 0: next_element = _nextElement else: step = -step next_element = _previousElement result = [] c = 0 while c_node is not NULL and c < slicelength: result.append(_elementFactory(self._doc, c_node)) c = c + 1 for i from 0 <= i < step: c_node = next_element(c_node) return result else: # indexing c_node = _findChild(self._c_node, x) if c_node is NULL: raise IndexError, u"list index out of range" return _elementFactory(self._doc, c_node) def __len__(self): u"""__len__(self) Returns the number of subelements. """ _assertValidNode(self) return _countElements(self._c_node.children) def __nonzero__(self): #u"__nonzero__(self)" # currently fails in Py3.1 import warnings warnings.warn( u"The behavior of this method will change in future versions. " u"Use specific 'len(elem)' or 'elem is not None' test instead.", FutureWarning ) # emulate old behaviour _assertValidNode(self) return _hasChild(self._c_node) def __contains__(self, element): u"__contains__(self, element)" cdef xmlNode* c_node _assertValidNode(self) if not isinstance(element, _Element): return 0 c_node = (<_Element>element)._c_node return c_node is not NULL and c_node.parent is self._c_node def __iter__(self): u"__iter__(self)" return ElementChildIterator(self) def __reversed__(self): u"__reversed__(self)" return ElementChildIterator(self, reversed=True) def index(self, _Element child not None, start=None, stop=None): u"""index(self, child, start=None, stop=None) Find the position of the child within the parent. This method is not part of the original ElementTree API. """ cdef Py_ssize_t k, l cdef Py_ssize_t c_start, c_stop cdef xmlNode* c_child cdef xmlNode* c_start_node _assertValidNode(self) _assertValidNode(child) c_child = child._c_node if c_child.parent is not self._c_node: raise ValueError, u"Element is not a child of this node." # handle the unbounded search straight away (normal case) if stop is None and (start is None or start == 0): k = 0 c_child = c_child.prev while c_child is not NULL: if _isElement(c_child): k = k + 1 c_child = c_child.prev return k # check indices if start is None: c_start = 0 else: c_start = start if stop is None: c_stop = 0 else: c_stop = stop if c_stop == 0 or \ c_start >= c_stop and (c_stop > 0 or c_start < 0): raise ValueError, u"list.index(x): x not in slice" # for negative slice indices, check slice before searching index if c_start < 0 or c_stop < 0: # start from right, at most up to leftmost(c_start, c_stop) if c_start < c_stop: k = -c_start else: k = -c_stop c_start_node = self._c_node.last l = 1 while c_start_node != c_child and l < k: if _isElement(c_start_node): l = l + 1 c_start_node = c_start_node.prev if c_start_node == c_child: # found! before slice end? if c_stop < 0 and l <= -c_stop: raise ValueError, u"list.index(x): x not in slice" elif c_start < 0: raise ValueError, u"list.index(x): x not in slice" # now determine the index backwards from child c_child = c_child.prev k = 0 if c_stop > 0: # we can optimize: stop after c_stop elements if not found while c_child != NULL and k < c_stop: if _isElement(c_child): k = k + 1 c_child = c_child.prev if k < c_stop: return k else: # traverse all while c_child != NULL: if _isElement(c_child): k = k + 1 c_child = c_child.prev if c_start > 0: if k >= c_start: return k else: return k if c_start != 0 or c_stop != 0: raise ValueError, u"list.index(x): x not in slice" else: raise ValueError, u"list.index(x): x not in list" def get(self, key, default=None): u"""get(self, key, default=None) Gets an element attribute. """ _assertValidNode(self) return _getAttributeValue(self, key, default) def keys(self): u"""keys(self) Gets a list of attribute names. The names are returned in an arbitrary order (just like for an ordinary Python dictionary). """ _assertValidNode(self) return _collectAttributes(self._c_node, 1) def values(self): u"""values(self) Gets element attribute values as a sequence of strings. The attributes are returned in an arbitrary order. """ _assertValidNode(self) return _collectAttributes(self._c_node, 2) def items(self): u"""items(self) Gets element attributes, as a sequence. The attributes are returned in an arbitrary order. """ _assertValidNode(self) return _collectAttributes(self._c_node, 3) def getchildren(self): u"""getchildren(self) Returns all direct children. The elements are returned in document order. :deprecated: Note that this method has been deprecated as of ElementTree 1.3 and lxml 2.0. New code should use ``list(element)`` or simply iterate over elements. """ _assertValidNode(self) return _collectChildren(self) def getparent(self): u"""getparent(self) Returns the parent of this element or None for the root element. """ cdef xmlNode* c_node #_assertValidNode(self) # not needed c_node = _parentElement(self._c_node) if c_node is NULL: return None return _elementFactory(self._doc, c_node) def getnext(self): u"""getnext(self) Returns the following sibling of this element or None. """ cdef xmlNode* c_node #_assertValidNode(self) # not needed c_node = _nextElement(self._c_node) if c_node is NULL: return None return _elementFactory(self._doc, c_node) def getprevious(self): u"""getprevious(self) Returns the preceding sibling of this element or None. """ cdef xmlNode* c_node #_assertValidNode(self) # not needed c_node = _previousElement(self._c_node) if c_node is NULL: return None return _elementFactory(self._doc, c_node) def itersiblings(self, tag=None, *, preceding=False): u"""itersiblings(self, tag=None, preceding=False) Iterate over the following or preceding siblings of this element. The direction is determined by the 'preceding' keyword which defaults to False, i.e. forward iteration over the following siblings. When True, the iterator yields the preceding siblings in reverse document order, i.e. starting right before the current element and going left. The generated elements can be restricted to a specific tag name with the 'tag' keyword. """ return SiblingsIterator(self, tag, preceding=preceding) def iterancestors(self, tag=None): u"""iterancestors(self, tag=None) Iterate over the ancestors of this element (from parent to parent). The generated elements can be restricted to a specific tag name with the 'tag' keyword. """ return AncestorsIterator(self, tag) def iterdescendants(self, tag=None): u"""iterdescendants(self, tag=None) Iterate over the descendants of this element in document order. As opposed to ``el.iter()``, this iterator does not yield the element itself. The generated elements can be restricted to a specific tag name with the 'tag' keyword. """ return ElementDepthFirstIterator(self, tag, inclusive=False) def iterchildren(self, tag=None, *, reversed=False): u"""iterchildren(self, tag=None, reversed=False) Iterate over the children of this element. As opposed to using normal iteration on this element, the generated elements can be restricted to a specific tag name with the 'tag' keyword and reversed with the 'reversed' keyword. """ return ElementChildIterator(self, tag, reversed=reversed) def getroottree(self): u"""getroottree(self) Return an ElementTree for the root node of the document that contains this element. This is the same as following element.getparent() up the tree until it returns None (for the root element) and then build an ElementTree for the last parent that was returned.""" _assertValidDoc(self._doc) return _elementTreeFactory(self._doc, None) def getiterator(self, tag=None): u"""getiterator(self, tag=None) Returns a sequence or iterator of all elements in the subtree in document order (depth first pre-order), starting with this element. Can be restricted to find only elements with a specific tag (pass ``tag="xyz"``) or from a namespace (pass ``tag="{ns}*"``). You can also pass the Element, Comment, ProcessingInstruction and Entity factory functions to look only for the specific element type. :deprecated: Note that this method is deprecated as of ElementTree 1.3 and lxml 2.0. It returns an iterator in lxml, which diverges from the original ElementTree behaviour. If you want an efficient iterator, use the ``element.iter()`` method instead. You should only use this method in new code if you require backwards compatibility with older versions of lxml or ElementTree. """ return ElementDepthFirstIterator(self, tag) def iter(self, tag=None): u"""iter(self, tag=None) Iterate over all elements in the subtree in document order (depth first pre-order), starting with this element. Can be restricted to find only elements with a specific tag (pass ``tag="xyz"``) or from a namespace (pass ``tag="{ns}*"``). You can also pass the Element, Comment, ProcessingInstruction and Entity factory functions to look only for the specific element type. """ return ElementDepthFirstIterator(self, tag) def itertext(self, tag=None, *, with_tail=True): u"""itertext(self, tag=None, with_tail=True) Iterates over the text content of a subtree. You can pass the ``tag`` keyword argument to restrict text content to a specific tag name. You can set the ``with_tail`` keyword argument to ``False`` to skip over tail text. """ return ElementTextIterator(self, tag, with_tail=with_tail) def makeelement(self, _tag, attrib=None, nsmap=None, **_extra): u"""makeelement(self, _tag, attrib=None, nsmap=None, **_extra) Creates a new element associated with the same document. """ _assertValidDoc(self._doc) return _makeElement(_tag, NULL, self._doc, None, None, None, attrib, nsmap, _extra) def find(self, path, namespaces=None): u"""find(self, path, namespaces=None) Finds the first matching subelement, by tag name or path. The optional ``namespaces`` argument accepts a prefix-to-namespace mapping that allows the usage of XPath prefixes in the path expression. """ if isinstance(path, QName): path = (path).text return _elementpath.find(self, path, namespaces) def findtext(self, path, default=None, namespaces=None): u"""findtext(self, path, default=None, namespaces=None) Finds text for the first matching subelement, by tag name or path. The optional ``namespaces`` argument accepts a prefix-to-namespace mapping that allows the usage of XPath prefixes in the path expression. """ if isinstance(path, QName): path = (path).text return _elementpath.findtext(self, path, default, namespaces) def findall(self, path, namespaces=None): u"""findall(self, path, namespaces=None) Finds all matching subelements, by tag name or path. The optional ``namespaces`` argument accepts a prefix-to-namespace mapping that allows the usage of XPath prefixes in the path expression. """ if isinstance(path, QName): path = (path).text return _elementpath.findall(self, path, namespaces) def iterfind(self, path, namespaces=None): u"""iterfind(self, path, namespaces=None) Iterates over all matching subelements, by tag name or path. The optional ``namespaces`` argument accepts a prefix-to-namespace mapping that allows the usage of XPath prefixes in the path expression. """ if isinstance(path, QName): path = (path).text return _elementpath.iterfind(self, path, namespaces) def xpath(self, _path, *, namespaces=None, extensions=None, smart_strings=True, **_variables): u"""xpath(self, _path, namespaces=None, extensions=None, smart_strings=True, **_variables) Evaluate an xpath expression using the element as context node. """ evaluator = XPathElementEvaluator(self, namespaces=namespaces, extensions=extensions, smart_strings=smart_strings) return evaluator(_path, **_variables) cdef extern from "etree_defs.h": # macro call to 't->tp_new()' for fast instantiation cdef object NEW_ELEMENT "PY_NEW" (object t) cdef _Element _elementFactory(_Document doc, xmlNode* c_node): cdef _Element result result = getProxy(c_node) if result is not None: return result if c_node is NULL: return None element_class = LOOKUP_ELEMENT_CLASS( ELEMENT_CLASS_LOOKUP_STATE, doc, c_node) if hasProxy(c_node): # prevent re-entry race condition - we just called into Python return getProxy(c_node) result = NEW_ELEMENT(element_class) if hasProxy(c_node): # prevent re-entry race condition - we just called into Python result._c_node = NULL return getProxy(c_node) _registerProxy(result, doc, c_node) if element_class is not _Element: result._init() return result cdef class __ContentOnlyElement(_Element): cdef int _raiseImmutable(self) except -1: raise TypeError, u"this element does not have children or attributes" def set(self, key, value): u"set(self, key, value)" self._raiseImmutable() def append(self, value): u"append(self, value)" self._raiseImmutable() def insert(self, index, value): u"insert(self, index, value)" self._raiseImmutable() def __setitem__(self, index, value): u"__setitem__(self, index, value)" self._raiseImmutable() property attrib: def __get__(self): return {} property text: def __get__(self): _assertValidNode(self) if self._c_node.content is NULL: return '' else: return funicode(self._c_node.content) def __set__(self, value): cdef tree.xmlDict* c_dict cdef char* c_text _assertValidNode(self) if value is None: c_text = NULL else: value = _utf8(value) c_text = _cstr(value) tree.xmlNodeSetContent(self._c_node, c_text) # ACCESSORS def __getitem__(self, x): u"__getitem__(self, x)" if python.PySlice_Check(x): return [] else: raise IndexError, u"list index out of range" def __len__(self): u"__len__(self)" return 0 def get(self, key, default=None): u"get(self, key, default=None)" return None def keys(self): u"keys(self)" return [] def items(self): u"items(self)" return [] def values(self): u"values(self)" return [] cdef class _Comment(__ContentOnlyElement): property tag: def __get__(self): return Comment def __repr__(self): return u"" % self.text cdef class _ProcessingInstruction(__ContentOnlyElement): property tag: def __get__(self): return ProcessingInstruction property target: # not in ElementTree def __get__(self): _assertValidNode(self) return funicode(self._c_node.name) def __set__(self, value): _assertValidNode(self) value = _utf8(value) c_text = _cstr(value) tree.xmlNodeSetName(self._c_node, c_text) def __repr__(self): text = self.text if text: return u"" % (self.target, text) else: return u"" % self.target cdef class _Entity(__ContentOnlyElement): property tag: def __get__(self): return Entity property name: # not in ElementTree def __get__(self): _assertValidNode(self) return funicode(self._c_node.name) def __set__(self, value): _assertValidNode(self) value_utf = _utf8(value) assert u'&' not in value and u';' not in value, \ u"Invalid entity name '%s'" % value tree.xmlNodeSetName(self._c_node, _cstr(value_utf)) property text: # FIXME: should this be None or '&[VALUE];' or the resolved # entity value ? def __get__(self): _assertValidNode(self) return u'&%s;' % funicode(self._c_node.name) def __repr__(self): return u"&%s;" % self.name cdef class QName: u"""QName(text_or_uri, tag=None) QName wrapper for qualified XML names. Pass a tag name by itself or a namespace URI and a tag name to create a qualified name. The ``text`` property holds the qualified name in ``{namespace}tagname`` notation. The ``namespace`` and ``localname`` properties hold the respective parts of the tag name. You can pass QName objects wherever a tag name is expected. Also, setting Element text from a QName will resolve the namespace prefix and set a qualified text value. """ cdef readonly object text cdef readonly object localname cdef readonly object namespace def __init__(self, text_or_uri_or_element, tag=None): if not _isString(text_or_uri_or_element): if isinstance(text_or_uri_or_element, _Element): text_or_uri_or_element = (<_Element>text_or_uri_or_element).tag if not _isString(text_or_uri_or_element): raise ValueError, (u"Invalid input tag of type %r" % type(text_or_uri_or_element)) elif isinstance(text_or_uri_or_element, QName): text_or_uri_or_element = (text_or_uri_or_element).text else: text_or_uri_or_element = unicode(text_or_uri_or_element) ns_utf, tag_utf = _getNsTag(text_or_uri_or_element) if tag is not None: # either ('ns', 'tag') or ('{ns}oldtag', 'newtag') if ns_utf is None: ns_utf = tag_utf # case 1: namespace ended up as tag name tag_utf = _utf8(tag) _tagValidOrRaise(tag_utf) self.localname = python.PyUnicode_FromEncodedObject( tag_utf, 'UTF-8', NULL) if ns_utf is None: self.namespace = None self.text = self.localname else: self.namespace = python.PyUnicode_FromEncodedObject( ns_utf, 'UTF-8', NULL) self.text = u"{%s}%s" % (self.namespace, self.localname) def __str__(self): return self.text def __hash__(self): return self.text.__hash__() def __richcmp__(one, other, int op): if not _isString(one): one = unicode(one) if not _isString(other): other = unicode(other) return python.PyObject_RichCompare(one, other, op) cdef public class _ElementTree [ type LxmlElementTreeType, object LxmlElementTree ]: cdef _Document _doc cdef _Element _context_node # Note that _doc is only used to store the original document if we do not # have a _context_node. All methods should prefer self._context_node._doc # to honour tree restructuring. _doc can happily be None! cdef _assertHasRoot(self): u"""We have to take care here: the document may not have a root node! This can happen if ElementTree() is called without any argument and the caller 'forgets' to call parse() afterwards, so this is a bug in the caller program. """ assert self._context_node is not None, \ u"ElementTree not initialized, missing root" def parse(self, source, _BaseParser parser=None, *, base_url=None): u"""parse(self, source, parser=None, base_url=None) Updates self with the content of source and returns its root """ cdef _Document doc = None try: doc = _parseDocument(source, parser, base_url) self._context_node = doc.getroot() if self._context_node is None: self._doc = doc except _TargetParserResult, result_container: # raises a TypeError if we don't get an _Element self._context_node = result_container.result return self._context_node def _setroot(self, _Element root not None): u"""_setroot(self, root) Relocate the ElementTree to a new root node. """ _assertValidNode(root) if root._c_node.type != tree.XML_ELEMENT_NODE: raise TypeError, u"Only elements can be the root of an ElementTree" self._context_node = root self._doc = None def getroot(self): u"""getroot(self) Gets the root element for this tree. """ return self._context_node def __copy__(self): return _elementTreeFactory(self._doc, self._context_node) def __deepcopy__(self, memo): cdef _Element root cdef _Document doc cdef xmlDoc* c_doc if self._context_node is not None: root = self._context_node.__copy__() _copyNonElementSiblings(self._context_node._c_node, root._c_node) doc = root._doc c_doc = self._context_node._doc._c_doc if c_doc.intSubset is not NULL and doc._c_doc.intSubset is NULL: doc._c_doc.intSubset = tree.xmlCopyDtd(c_doc.intSubset) if doc._c_doc.intSubset is NULL: python.PyErr_NoMemory() if c_doc.extSubset is not NULL and not doc._c_doc.extSubset is NULL: doc._c_doc.extSubset = tree.xmlCopyDtd(c_doc.extSubset) if doc._c_doc.extSubset is NULL: python.PyErr_NoMemory() return _elementTreeFactory(None, root) elif self._doc is not None: _assertValidDoc(self._doc) c_doc = tree.xmlCopyDoc(self._doc._c_doc, 1) if c_doc is NULL: python.PyErr_NoMemory() doc = _documentFactory(c_doc, self._doc._parser) return _elementTreeFactory(doc, None) else: # so what ... return self # not in ElementTree, read-only property docinfo: u"""Information about the document provided by parser and DTD. This value is only defined for ElementTree objects based on the root node of a parsed document (e.g. those returned by the parse functions), not for trees that were built manually. """ def __get__(self): self._assertHasRoot() return DocInfo(self._context_node._doc) # not in ElementTree, read-only property parser: u"""The parser that was used to parse the document in this ElementTree. """ def __get__(self): if self._context_node is not None and \ self._context_node._doc is not None: return self._context_node._doc._parser if self._doc is not None: return self._doc._parser return None def write(self, file, *, encoding=None, method=u"xml", pretty_print=False, xml_declaration=None, with_tail=True, standalone=None, docstring=None, compression=0, exclusive=False, with_comments=True): u"""write(self, file, encoding=None, method="xml", pretty_print=False, xml_declaration=None, with_tail=True, standalone=None, compression=0, exclusive=False, with_comments=True) Write the tree to a filename, file or file-like object. Defaults to ASCII encoding and writing a declaration as needed. The keyword argument 'method' selects the output method: 'xml', 'html', 'text' or 'c14n'. Default is 'xml'. The ``exclusive`` and ``with_comments`` arguments are only used with C14N output, where they request exclusive and uncommented C14N serialisation respectively. Passing a boolean value to the ``standalone`` option will output an XML declaration with the corresponding ``standalone`` flag. The ``compression`` option enables GZip compression level 1-9. """ cdef bint write_declaration cdef int is_standalone self._assertHasRoot() _assertValidNode(self._context_node) if compression is None or compression < 0: compression = 0 # C14N serialisation if method == 'c14n': if encoding is not None: raise ValueError("Cannot specify encoding with C14N") if xml_declaration: raise ValueError("Cannot enable XML declaration in C14N") _tofilelikeC14N(file, self._context_node, exclusive, with_comments, compression) return if not with_comments: raise ValueError("Can only discard comments in C14N serialisation") # suppress decl. in default case (purely for ElementTree compatibility) if xml_declaration is not None: write_declaration = xml_declaration if encoding is None: encoding = u'ASCII' elif encoding is None: encoding = u'ASCII' write_declaration = 0 else: encoding = encoding.upper() write_declaration = encoding not in \ (u'US-ASCII', u'ASCII', u'UTF8', u'UTF-8') if standalone is None: is_standalone = -1 elif standalone: write_declaration = 1 is_standalone = 1 else: write_declaration = 1 is_standalone = 0 _tofilelike(file, self._context_node, encoding, docstring, method, write_declaration, 1, pretty_print, with_tail, is_standalone, compression) def getpath(self, _Element element not None): u"""getpath(self, element) Returns a structural, absolute XPath expression to find that element. """ cdef _Document doc cdef _Element root cdef xmlDoc* c_doc cdef char* c_path _assertValidNode(element) if self._context_node is not None: root = self._context_node doc = root._doc elif self._doc is not None: doc = self._doc root = doc.getroot() else: raise ValueError, u"Element is not in this tree." _assertValidDoc(doc) _assertValidNode(root) if element._doc is not doc: raise ValueError, u"Element is not in this tree." c_doc = _fakeRootDoc(doc._c_doc, root._c_node) c_path = tree.xmlGetNodePath(element._c_node) _destroyFakeDoc(doc._c_doc, c_doc) if c_path is NULL: python.PyErr_NoMemory() path = funicode(c_path) tree.xmlFree(c_path) return path def getiterator(self, tag=None): u"""getiterator(self, tag=None) Returns a sequence or iterator of all elements in document order (depth first pre-order), starting with the root element. Can be restricted to find only elements with a specific tag (pass ``tag="xyz"`` or ``tag="{ns}xyz"``) or from a namespace (pass ``tag="{ns}*"``). You can also pass the Element, Comment, ProcessingInstruction and Entity factory functions to look only for the specific element type. :deprecated: Note that this method is deprecated as of ElementTree 1.3 and lxml 2.0. It returns an iterator in lxml, which diverges from the original ElementTree behaviour. If you want an efficient iterator, use the ``tree.iter()`` method instead. You should only use this method in new code if you require backwards compatibility with older versions of lxml or ElementTree. """ root = self.getroot() if root is None: return () return root.getiterator(tag) def iter(self, tag=None): u"""iter(self, tag=None) Creates an iterator for the root element. The iterator loops over all elements in this tree, in document order. """ root = self.getroot() if root is None: return () return root.iter(tag) def find(self, path, namespaces=None): u"""find(self, path, namespaces=None) Finds the first toplevel element with given tag. Same as ``tree.getroot().find(path)``. The optional ``namespaces`` argument accepts a prefix-to-namespace mapping that allows the usage of XPath prefixes in the path expression. """ self._assertHasRoot() root = self.getroot() if _isString(path): start = path[:1] if start == u"/": path = u"." + path elif start == b"/": path = b"." + path return root.find(path, namespaces) def findtext(self, path, default=None, namespaces=None): u"""findtext(self, path, default=None, namespaces=None) Finds the text for the first element matching the ElementPath expression. Same as getroot().findtext(path) The optional ``namespaces`` argument accepts a prefix-to-namespace mapping that allows the usage of XPath prefixes in the path expression. """ self._assertHasRoot() root = self.getroot() if _isString(path): start = path[:1] if start == u"/": path = u"." + path elif start == b"/": path = b"." + path return root.findtext(path, default, namespaces) def findall(self, path, namespaces=None): u"""findall(self, path, namespaces=None) Finds all elements matching the ElementPath expression. Same as getroot().findall(path). The optional ``namespaces`` argument accepts a prefix-to-namespace mapping that allows the usage of XPath prefixes in the path expression. """ self._assertHasRoot() root = self.getroot() if _isString(path): start = path[:1] if start == u"/": path = u"." + path elif start == b"/": path = b"." + path return root.findall(path, namespaces) def iterfind(self, path, namespaces=None): u"""iterfind(self, path, namespaces=None) Iterates over all elements matching the ElementPath expression. Same as getroot().iterfind(path). The optional ``namespaces`` argument accepts a prefix-to-namespace mapping that allows the usage of XPath prefixes in the path expression. """ self._assertHasRoot() root = self.getroot() if _isString(path): start = path[:1] if start == u"/": path = u"." + path elif start == b"/": path = b"." + path return root.iterfind(path, namespaces) def xpath(self, _path, *, namespaces=None, extensions=None, smart_strings=True, **_variables): u"""xpath(self, _path, namespaces=None, extensions=None, smart_strings=True, **_variables) XPath evaluate in context of document. ``namespaces`` is an optional dictionary with prefix to namespace URI mappings, used by XPath. ``extensions`` defines additional extension functions. Returns a list (nodeset), or bool, float or string. In case of a list result, return Element for element nodes, string for text and attribute values. Note: if you are going to apply multiple XPath expressions against the same document, it is more efficient to use XPathEvaluator directly. """ self._assertHasRoot() evaluator = XPathDocumentEvaluator(self, namespaces=namespaces, extensions=extensions, smart_strings=smart_strings) return evaluator(_path, **_variables) def xslt(self, _xslt, extensions=None, access_control=None, **_kw): u"""xslt(self, _xslt, extensions=None, access_control=None, **_kw) Transform this document using other document. xslt is a tree that should be XSLT keyword parameters are XSLT transformation parameters. Returns the transformed tree. Note: if you are going to apply the same XSLT stylesheet against multiple documents, it is more efficient to use the XSLT class directly. """ self._assertHasRoot() style = XSLT(_xslt, extensions=extensions, access_control=access_control) return style(self, **_kw) def relaxng(self, relaxng): u"""relaxng(self, relaxng) Validate this document using other document. The relaxng argument is a tree that should contain a Relax NG schema. Returns True or False, depending on whether validation succeeded. Note: if you are going to apply the same Relax NG schema against multiple documents, it is more efficient to use the RelaxNG class directly. """ self._assertHasRoot() schema = RelaxNG(relaxng) return schema.validate(self) def xmlschema(self, xmlschema): u"""xmlschema(self, xmlschema) Validate this document using other document. The xmlschema argument is a tree that should contain an XML Schema. Returns True or False, depending on whether validation succeeded. Note: If you are going to apply the same XML Schema against multiple documents, it is more efficient to use the XMLSchema class directly. """ self._assertHasRoot() schema = XMLSchema(xmlschema) return schema.validate(self) def xinclude(self): u"""xinclude(self) Process the XInclude nodes in this document and include the referenced XML fragments. There is support for loading files through the file system, HTTP and FTP. Note that XInclude does not support custom resolvers in Python space due to restrictions of libxml2 <= 2.6.29. """ self._assertHasRoot() XInclude()(self._context_node) def write_c14n(self, file, *, exclusive=False, with_comments=True, compression=0): u"""write_c14n(self, file, exclusive=False, with_comments=True, compression=0) C14N write of document. Always writes UTF-8. The ``compression`` option enables GZip compression level 1-9. """ self._assertHasRoot() _assertValidNode(self._context_node) if compression is None or compression < 0: compression = 0 _tofilelikeC14N(file, self._context_node, exclusive, with_comments, compression) cdef _ElementTree _elementTreeFactory(_Document doc, _Element context_node): return _newElementTree(doc, context_node, _ElementTree) cdef _ElementTree _newElementTree(_Document doc, _Element context_node, object baseclass): cdef _ElementTree result result = baseclass() if context_node is None and doc is not None: context_node = doc.getroot() if context_node is None: _assertValidDoc(doc) result._doc = doc else: _assertValidNode(context_node) result._context_node = context_node return result cdef class _Attrib: u"""A dict-like proxy for the ``Element.attrib`` property. """ cdef _Element _element def __cinit__(self, _Element element not None): _assertValidNode(element) self._element = element # MANIPULATORS def __setitem__(self, key, value): _setAttributeValue(self._element, key, value) def __delitem__(self, key): _delAttribute(self._element, key) def update(self, sequence_or_dict): if isinstance(sequence_or_dict, dict): sequence_or_dict = sequence_or_dict.items() for key, value in sequence_or_dict: _setAttributeValue(self._element, key, value) def pop(self, key, *default): if python.PyTuple_GET_SIZE(default) > 1: raise TypeError, u"pop expected at most 2 arguments, got %d" % ( python.PyTuple_GET_SIZE(default)+1) result = _getAttributeValue(self._element, key, None) if result is None: if python.PyTuple_GET_SIZE(default) == 0: raise KeyError, key else: result = python.PyTuple_GET_ITEM(default, 0) python.Py_INCREF(result) else: _delAttribute(self._element, key) return result def clear(self): cdef xmlNode* c_node c_node = self._element._c_node while c_node.properties is not NULL: tree.xmlRemoveProp(c_node.properties) # ACCESSORS def __repr__(self): return repr(dict( _attributeIteratorFactory(self._element, 3) )) def __getitem__(self, key): result = _getAttributeValue(self._element, key, None) if result is None: raise KeyError, key else: return result def __nonzero__(self): cdef xmlAttr* c_attr c_attr = self._element._c_node.properties while c_attr is not NULL: if c_attr.type == tree.XML_ATTRIBUTE_NODE: return 1 c_attr = c_attr.next return 0 def __len__(self): cdef xmlAttr* c_attr cdef Py_ssize_t c c = 0 c_attr = self._element._c_node.properties while c_attr is not NULL: if c_attr.type == tree.XML_ATTRIBUTE_NODE: c = c + 1 c_attr = c_attr.next return c def get(self, key, default=None): return _getAttributeValue(self._element, key, default) def keys(self): return _collectAttributes(self._element._c_node, 1) def __iter__(self): return iter(_collectAttributes(self._element._c_node, 1)) def iterkeys(self): return iter(_collectAttributes(self._element._c_node, 1)) def values(self): return _collectAttributes(self._element._c_node, 2) def itervalues(self): return iter(_collectAttributes(self._element._c_node, 2)) def items(self): return _collectAttributes(self._element._c_node, 3) def iteritems(self): return iter(_collectAttributes(self._element._c_node, 3)) def has_key(self, key): if key in self: return True else: return False def __contains__(self, key): cdef xmlNode* c_node cdef char* c_href ns, tag = _getNsTag(key) c_node = self._element._c_node c_href = NULL if ns is None else _cstr(ns) if tree.xmlHasNsProp(c_node, _cstr(tag), c_href): return 1 else: return 0 def __richcmp__(one, other, int op): if not python.PyDict_Check(one): one = dict(one) if not python.PyDict_Check(other): other = dict(other) return python.PyObject_RichCompare(one, other, op) cdef extern from "etree_defs.h": # macro call to 't->tp_new()' for fast instantiation cdef _Attrib NEW_ATTRIB "PY_NEW" (object t) cdef class _AttribIterator: u"""Attribute iterator - for internal use only! """ # XML attributes must not be removed while running! cdef _Element _node cdef xmlAttr* _c_attr cdef int _keysvalues # 1 - keys, 2 - values, 3 - items (key, value) def __iter__(self): return self def __next__(self): cdef xmlAttr* c_attr if self._node is None: raise StopIteration c_attr = self._c_attr while c_attr is not NULL and c_attr.type != tree.XML_ATTRIBUTE_NODE: c_attr = c_attr.next if c_attr is NULL: self._node = None raise StopIteration self._c_attr = c_attr.next if self._keysvalues == 1: return _namespacedName(c_attr) elif self._keysvalues == 2: return _attributeValue(self._node._c_node, c_attr) else: return (_namespacedName(c_attr), _attributeValue(self._node._c_node, c_attr)) cdef object _attributeIteratorFactory(_Element element, int keysvalues): cdef _AttribIterator attribs if element._c_node.properties is NULL: return ITER_EMPTY attribs = _AttribIterator() attribs._node = element attribs._c_attr = element._c_node.properties attribs._keysvalues = keysvalues return attribs cdef public class _ElementTagMatcher [ object LxmlElementTagMatcher, type LxmlElementTagMatcherType ]: cdef object _pystrings cdef int _node_type cdef char* _href cdef char* _name cdef _initTagMatch(self, tag): self._href = NULL self._name = NULL if tag is None: self._node_type = 0 elif tag is Comment: self._node_type = tree.XML_COMMENT_NODE elif tag is ProcessingInstruction: self._node_type = tree.XML_PI_NODE elif tag is Entity: self._node_type = tree.XML_ENTITY_REF_NODE elif tag is Element: self._node_type = tree.XML_ELEMENT_NODE else: self._node_type = tree.XML_ELEMENT_NODE self._pystrings = _getNsTag(tag) if self._pystrings[0] is not None: self._href = _cstr(self._pystrings[0]) self._name = _cstr(self._pystrings[1]) if self._name[0] == c'*' and self._name[1] == c'\0': self._name = NULL cdef public class _ElementIterator(_ElementTagMatcher) [ object LxmlElementIterator, type LxmlElementIteratorType ]: # we keep Python references here to control GC cdef _Element _node cdef _node_to_node_function _next_element def __iter__(self): return self cdef void _storeNext(self, _Element node): cdef xmlNode* c_node c_node = self._next_element(node._c_node) while c_node is not NULL and \ self._node_type != 0 and \ (self._node_type != c_node.type or not _tagMatches(c_node, self._href, self._name)): c_node = self._next_element(c_node) if c_node is NULL: self._node = None else: # Python ref: self._node = _elementFactory(node._doc, c_node) def __next__(self): cdef xmlNode* c_node cdef _Element current_node if self._node is None: raise StopIteration # Python ref: current_node = self._node self._storeNext(current_node) return current_node cdef class ElementChildIterator(_ElementIterator): u"""ElementChildIterator(self, node, tag=None, reversed=False) Iterates over the children of an element. """ def __cinit__(self, _Element node not None, tag=None, *, reversed=False): cdef xmlNode* c_node _assertValidNode(node) self._initTagMatch(tag) if reversed: c_node = _findChildBackwards(node._c_node, 0) self._next_element = _previousElement else: c_node = _findChildForwards(node._c_node, 0) self._next_element = _nextElement if tag is not None: while c_node is not NULL and \ self._node_type != 0 and \ (self._node_type != c_node.type or not _tagMatches(c_node, self._href, self._name)): c_node = self._next_element(c_node) if c_node is not NULL: # store Python ref: self._node = _elementFactory(node._doc, c_node) cdef class SiblingsIterator(_ElementIterator): u"""SiblingsIterator(self, node, tag=None, preceding=False) Iterates over the siblings of an element. You can pass the boolean keyword ``preceding`` to specify the direction. """ def __cinit__(self, _Element node not None, tag=None, *, preceding=False): _assertValidNode(node) self._initTagMatch(tag) if preceding: self._next_element = _previousElement else: self._next_element = _nextElement self._storeNext(node) cdef class AncestorsIterator(_ElementIterator): u"""AncestorsIterator(self, node, tag=None) Iterates over the ancestors of an element (from parent to parent). """ def __cinit__(self, _Element node not None, tag=None): _assertValidNode(node) self._initTagMatch(tag) self._next_element = _parentElement self._storeNext(node) cdef class ElementDepthFirstIterator(_ElementTagMatcher): u"""ElementDepthFirstIterator(self, node, tag=None, inclusive=True) Iterates over an element and its sub-elements in document order (depth first pre-order). Note that this also includes comments, entities and processing instructions. To filter them out, check if the ``tag`` property of the returned element is a string (i.e. not None and not a factory function), or pass the ``Element`` factory for the ``tag`` keyword. If the optional ``tag`` argument is not None, the iterator returns only the elements that match the respective name and namespace. The optional boolean argument 'inclusive' defaults to True and can be set to False to exclude the start element itself. Note that the behaviour of this iterator is completely undefined if the tree it traverses is modified during iteration. """ # we keep Python references here to control GC # keep next node to return and the (s)top node cdef _Element _next_node cdef _Element _top_node def __cinit__(self, _Element node not None, tag=None, *, inclusive=True): _assertValidNode(node) self._top_node = node self._next_node = node self._initTagMatch(tag) if not inclusive or \ tag is not None and \ self._node_type != 0 and \ (self._node_type != node._c_node.type or not _tagMatches(node._c_node, self._href, self._name)): # this cannot raise StopIteration, self._next_node != None self.__next__() def __iter__(self): return self def __next__(self): cdef xmlNode* c_node cdef _Element current_node if self._next_node is None: raise StopIteration current_node = self._next_node c_node = self._next_node._c_node if self._name is NULL and self._href is NULL: c_node = self._nextNodeAnyTag(c_node) else: c_node = self._nextNodeMatchTag(c_node) if c_node is NULL: self._next_node = None else: self._next_node = _elementFactory(current_node._doc, c_node) return current_node cdef xmlNode* _nextNodeAnyTag(self, xmlNode* c_node): tree.BEGIN_FOR_EACH_ELEMENT_FROM(self._top_node._c_node, c_node, 0) if self._node_type == 0 or self._node_type == c_node.type: return c_node tree.END_FOR_EACH_ELEMENT_FROM(c_node) return NULL cdef xmlNode* _nextNodeMatchTag(self, xmlNode* c_node): cdef char* c_name = NULL if self._name is not NULL: c_name = tree.xmlDictExists(c_node.doc.dict, self._name, -1) if c_name is NULL: # not found in dict => not in document at all return NULL tree.BEGIN_FOR_EACH_ELEMENT_FROM(self._top_node._c_node, c_node, 0) if c_node.type == tree.XML_ELEMENT_NODE: if (c_name is NULL or c_name is c_node.name) and \ _tagMatches(c_node, self._href, self._name): return c_node tree.END_FOR_EACH_ELEMENT_FROM(c_node) return NULL cdef class ElementTextIterator: u"""ElementTextIterator(self, element, tag=None, with_tail=True) Iterates over the text content of a subtree. You can pass the ``tag`` keyword argument to restrict text content to a specific tag name. You can set the ``with_tail`` keyword argument to ``False`` to skip over tail text. """ cdef object _nextEvent cdef _Element _start_element def __cinit__(self, _Element element not None, tag=None, *, with_tail=True): _assertValidNode(element) if with_tail: events = (u"start", u"end") else: events = (u"start",) self._start_element = element self._nextEvent = iterwalk(element, events=events, tag=tag).__next__ def __iter__(self): return self def __next__(self): cdef _Element element while result is None: event, element = self._nextEvent() # raises StopIteration if event == u"start": result = element.text elif element is not self._start_element: result = element.tail return result cdef xmlNode* _createElement(xmlDoc* c_doc, object name_utf) except NULL: cdef xmlNode* c_node c_node = tree.xmlNewDocNode(c_doc, NULL, _cstr(name_utf), NULL) return c_node cdef xmlNode* _createComment(xmlDoc* c_doc, char* text): cdef xmlNode* c_node c_node = tree.xmlNewDocComment(c_doc, text) return c_node cdef xmlNode* _createPI(xmlDoc* c_doc, char* target, char* text): cdef xmlNode* c_node c_node = tree.xmlNewDocPI(c_doc, target, text) return c_node cdef xmlNode* _createEntity(xmlDoc* c_doc, char* name): cdef xmlNode* c_node c_node = tree.xmlNewReference(c_doc, name) return c_node # module-level API for ElementTree def Element(_tag, attrib=None, nsmap=None, **_extra): u"""Element(_tag, attrib=None, nsmap=None, **_extra) Element factory. This function returns an object implementing the Element interface. Also look at the `_Element.makeelement()` and `_BaseParser.makeelement()` methods, which provide a faster way to create an Element within a specific document or parser context. """ return _makeElement(_tag, NULL, None, None, None, None, attrib, nsmap, _extra) def Comment(text=None): u"""Comment(text=None) Comment element factory. This factory function creates a special element that will be serialized as an XML comment. """ cdef _Document doc cdef xmlNode* c_node cdef xmlDoc* c_doc if text is None: text = b'' else: text = _utf8(text) c_doc = _newXMLDoc() doc = _documentFactory(c_doc, None) c_node = _createComment(c_doc, _cstr(text)) tree.xmlAddChild(c_doc, c_node) return _elementFactory(doc, c_node) def ProcessingInstruction(target, text=None): u"""ProcessingInstruction(target, text=None) ProcessingInstruction element factory. This factory function creates a special element that will be serialized as an XML processing instruction. """ cdef _Document doc cdef xmlNode* c_node cdef xmlDoc* c_doc target = _utf8(target) if text is None: text = b'' else: text = _utf8(text) c_doc = _newXMLDoc() doc = _documentFactory(c_doc, None) c_node = _createPI(c_doc, _cstr(target), _cstr(text)) tree.xmlAddChild(c_doc, c_node) return _elementFactory(doc, c_node) PI = ProcessingInstruction cdef class CDATA: u"""CDATA(data) CDATA factory. This factory creates an opaque data object that can be used to set Element text. The usual way to use it is:: >>> from lxml import etree >>> el = etree.Element('content') >>> el.text = etree.CDATA('a string') """ cdef object _utf8_data def __init__(self, data): self._utf8_data = _utf8(data) def Entity(name): u"""Entity(name) Entity factory. This factory function creates a special element that will be serialized as an XML entity reference or character reference. Note, however, that entities will not be automatically declared in the document. A document that uses entity references requires a DTD to define the entities. """ cdef _Document doc cdef xmlNode* c_node cdef xmlDoc* c_doc cdef char* c_name name_utf = _utf8(name) c_name = _cstr(name_utf) if c_name[0] == c'#': if not _characterReferenceIsValid(c_name + 1): raise ValueError, u"Invalid character reference: '%s'" % name elif not _xmlNameIsValid(c_name): raise ValueError, u"Invalid entity reference: '%s'" % name c_doc = _newXMLDoc() doc = _documentFactory(c_doc, None) c_node = _createEntity(c_doc, c_name) tree.xmlAddChild(c_doc, c_node) return _elementFactory(doc, c_node) def SubElement(_Element _parent not None, _tag, attrib=None, nsmap=None, **_extra): u"""SubElement(_parent, _tag, attrib=None, nsmap=None, **_extra) Subelement factory. This function creates an element instance, and appends it to an existing element. """ return _makeSubElement(_parent, _tag, None, None, attrib, nsmap, _extra) def ElementTree(_Element element=None, *, file=None, _BaseParser parser=None): u"""ElementTree(element=None, file=None, parser=None) ElementTree wrapper class. """ cdef xmlNode* c_next cdef xmlNode* c_node cdef xmlNode* c_node_copy cdef xmlDoc* c_doc cdef _ElementTree etree cdef _Document doc if element is not None: doc = element._doc elif file is not None: try: doc = _parseDocument(file, parser, None) except _TargetParserResult, result_container: return result_container.result else: c_doc = _newXMLDoc() doc = _documentFactory(c_doc, parser) return _elementTreeFactory(doc, element) def HTML(text, _BaseParser parser=None, *, base_url=None): u"""HTML(text, parser=None, base_url=None) Parses an HTML document from a string constant. Returns the root node (or the result returned by a parser target). This function can be used to embed "HTML literals" in Python code. To override the parser with a different ``HTMLParser`` you can pass it to the ``parser`` keyword argument. The ``base_url`` keyword argument allows to set the original base URL of the document to support relative Paths when looking up external entities (DTD, XInclude, ...). """ cdef _Document doc if parser is None: parser = __GLOBAL_PARSER_CONTEXT.getDefaultParser() if not isinstance(parser, HTMLParser): parser = __DEFAULT_HTML_PARSER try: doc = _parseMemoryDocument(text, base_url, parser) return doc.getroot() except _TargetParserResult, result_container: return result_container.result def XML(text, _BaseParser parser=None, *, base_url=None): u"""XML(text, parser=None, base_url=None) Parses an XML document or fragment from a string constant. Returns the root node (or the result returned by a parser target). This function can be used to embed "XML literals" in Python code, like in >>> root = etree.XML("") To override the parser with a different ``XMLParser`` you can pass it to the ``parser`` keyword argument. The ``base_url`` keyword argument allows to set the original base URL of the document to support relative Paths when looking up external entities (DTD, XInclude, ...). """ cdef _Document doc if parser is None: parser = __GLOBAL_PARSER_CONTEXT.getDefaultParser() if not isinstance(parser, XMLParser): parser = __DEFAULT_XML_PARSER try: doc = _parseMemoryDocument(text, base_url, parser) return doc.getroot() except _TargetParserResult, result_container: return result_container.result def fromstring(text, _BaseParser parser=None, *, base_url=None): u"""fromstring(text, parser=None, base_url=None) Parses an XML document or fragment from a string. Returns the root node (or the result returned by a parser target). To override the default parser with a different parser you can pass it to the ``parser`` keyword argument. The ``base_url`` keyword argument allows to set the original base URL of the document to support relative Paths when looking up external entities (DTD, XInclude, ...). """ cdef _Document doc try: doc = _parseMemoryDocument(text, base_url, parser) return doc.getroot() except _TargetParserResult, result_container: return result_container.result def fromstringlist(strings, _BaseParser parser=None): u"""fromstringlist(strings, parser=None) Parses an XML document from a sequence of strings. Returns the root node (or the result returned by a parser target). To override the default parser with a different parser you can pass it to the ``parser`` keyword argument. """ cdef _Document doc if parser is None: parser = __GLOBAL_PARSER_CONTEXT.getDefaultParser() feed = parser.feed for data in strings: feed(data) return parser.close() def iselement(element): u"""iselement(element) Checks if an object appears to be a valid element object. """ return isinstance(element, _Element) and (<_Element>element)._c_node is not NULL def dump(_Element elem not None, *, pretty_print=True, with_tail=True): u"""dump(elem, pretty_print=True, with_tail=True) Writes an element tree or element structure to sys.stdout. This function should be used for debugging only. """ _assertValidNode(elem) _dumpToFile(sys.stdout, elem._c_node, pretty_print, with_tail) def tostring(element_or_tree, *, encoding=None, method=u"xml", xml_declaration=None, pretty_print=False, with_tail=True, standalone=None, doctype=None, exclusive=False, with_comments=True): u"""tostring(element_or_tree, encoding=None, method="xml", xml_declaration=None, pretty_print=False, with_tail=True, standalone=None, doctype=None, exclusive=False, with_comments=True) Serialize an element to an encoded string representation of its XML tree. Defaults to ASCII encoding without XML declaration. This behaviour can be configured with the keyword arguments 'encoding' (string) and 'xml_declaration' (bool). Note that changing the encoding to a non UTF-8 compatible encoding will enable a declaration by default. You can also serialise to a Unicode string without declaration by passing the ``unicode`` function as encoding (or ``str`` in Py3), or the name 'unicode'. This changes the return value from a byte string to an unencoded unicode string. The keyword argument 'pretty_print' (bool) enables formatted XML. The keyword argument 'method' selects the output method: 'xml', 'html', plain 'text' (text content without tags) or 'c14n'. Default is 'xml'. The ``exclusive`` and ``with_comments`` arguments are only used with C14N output, where they request exclusive and uncommented C14N serialisation respectively. Passing a boolean value to the ``standalone`` option will output an XML declaration with the corresponding ``standalone`` flag. The ``doctype`` option allows passing in a plain string that will be serialised before the XML tree. Note that passing in non well-formed content here will make the XML output non well-formed. You can prevent the tail text of the element from being serialised by passing the boolean ``with_tail`` option. This has no impact on the tail text of children, which will always be serialised. """ cdef bint write_declaration cdef int is_standalone # C14N serialisation if method == 'c14n': if encoding is not None: raise ValueError("Cannot specify encoding with C14N") if xml_declaration: raise ValueError("Cannot enable XML declaration in C14N") return _tostringC14N(element_or_tree, exclusive, with_comments) if not with_comments: raise ValueError("Can only discard comments in C14N serialisation") if encoding is _unicode or (encoding is not None and encoding.upper() == 'UNICODE'): if xml_declaration: raise ValueError, \ u"Serialisation to unicode must not request an XML declaration" write_declaration = 0 encoding = _unicode elif xml_declaration is None: # by default, write an XML declaration only for non-standard encodings write_declaration = encoding is not None and encoding.upper() not in \ (u'ASCII', u'UTF-8', u'UTF8', u'US-ASCII') else: write_declaration = xml_declaration if encoding is None: encoding = u'ASCII' if standalone is None: is_standalone = -1 elif standalone: write_declaration = 1 is_standalone = 1 else: write_declaration = 1 is_standalone = 0 if isinstance(element_or_tree, _Element): return _tostring(<_Element>element_or_tree, encoding, doctype, method, write_declaration, 0, pretty_print, with_tail, is_standalone) elif isinstance(element_or_tree, _ElementTree): return _tostring((<_ElementTree>element_or_tree)._context_node, encoding, doctype, method, write_declaration, 1, pretty_print, with_tail, is_standalone) else: raise TypeError, u"Type '%s' cannot be serialized." % \ python._fqtypename(element_or_tree) def tostringlist(element_or_tree, *args, **kwargs): u"""tostringlist(element_or_tree, *args, **kwargs) Serialize an element to an encoded string representation of its XML tree, stored in a list of partial strings. This is purely for ElementTree 1.3 compatibility. The result is a single string wrapped in a list. """ return [tostring(element_or_tree, *args, **kwargs)] def tounicode(element_or_tree, *, method=u"xml", pretty_print=False, with_tail=True, doctype=None): u"""tounicode(element_or_tree, method="xml", pretty_print=False, with_tail=True, doctype=None) Serialize an element to the Python unicode representation of its XML tree. :deprecated: use ``tostring(el, encoding=unicode)`` instead. Note that the result does not carry an XML encoding declaration and is therefore not necessarily suited for serialization to byte streams without further treatment. The boolean keyword argument 'pretty_print' enables formatted XML. The keyword argument 'method' selects the output method: 'xml', 'html' or plain 'text'. You can prevent the tail text of the element from being serialised by passing the boolean ``with_tail`` option. This has no impact on the tail text of children, which will always be serialised. """ if isinstance(element_or_tree, _Element): return _tostring(<_Element>element_or_tree, _unicode, doctype, method, 0, 0, pretty_print, with_tail, -1) elif isinstance(element_or_tree, _ElementTree): return _tostring((<_ElementTree>element_or_tree)._context_node, _unicode, doctype, method, 0, 1, pretty_print, with_tail, -1) else: raise TypeError, u"Type '%s' cannot be serialized." % \ type(element_or_tree) def parse(source, _BaseParser parser=None, *, base_url=None): u"""parse(source, parser=None, base_url=None) Return an ElementTree object loaded with source elements. If no parser is provided as second argument, the default parser is used. The ``source`` can be any of the following: - a file name/path - a file object - a file-like object - a URL using the HTTP or FTP protocol To parse from a string, use the ``fromstring()`` function instead. Note that it is generally faster to parse from a file path or URL than from an open file object or file-like object. Transparent decompression from gzip compressed sources is supported (unless explicitly disabled in libxml2). The ``base_url`` keyword allows setting a URL for the document when parsing from a file-like object. This is needed when looking up external entities (DTD, XInclude, ...) with relative paths. """ cdef _Document doc try: doc = _parseDocument(source, parser, base_url) return _elementTreeFactory(doc, None) except _TargetParserResult, result_container: return result_container.result ################################################################################ # Include submodules include "readonlytree.pxi" # Read-only implementation of Element proxies include "classlookup.pxi" # Element class lookup mechanisms include "nsclasses.pxi" # Namespace implementation and registry include "docloader.pxi" # Support for custom document loaders include "parser.pxi" # XML Parser include "saxparser.pxi" # SAX-like Parser interface and tree builder include "parsertarget.pxi" # ET Parser target include "serializer.pxi" # XML output functions include "iterparse.pxi" # incremental XML parsing include "xmlid.pxi" # XMLID and IDDict include "xinclude.pxi" # XInclude include "cleanup.pxi" # Cleanup and recursive element removal functions ################################################################################ # Include submodules for XPath and XSLT include "extensions.pxi" # XPath/XSLT extension functions include "xpath.pxi" # XPath evaluation include "xslt.pxi" # XSL transformations include "xsltext.pxi" # XSL extension elements ################################################################################ # Validation class DocumentInvalid(LxmlError): u"""Validation error. Raised by all document validators when their ``assertValid(tree)`` method fails. """ pass cdef class _Validator: u"Base class for XML validators." cdef _ErrorLog _error_log def __cinit__(self): self._error_log = _ErrorLog() def validate(self, etree): u"""validate(self, etree) Validate the document using this schema. Returns true if document is valid, false if not. """ return self(etree) def assertValid(self, etree): u"""assertValid(self, etree) Raises `DocumentInvalid` if the document does not comply with the schema. """ if not self(etree): raise DocumentInvalid(self._error_log._buildExceptionMessage( u"Document does not comply with schema"), self._error_log) def assert_(self, etree): u"""assert_(self, etree) Raises `AssertionError` if the document does not comply with the schema. """ if not self(etree): raise AssertionError, self._error_log._buildExceptionMessage( u"Document does not comply with schema") cpdef _append_log_message(self, int domain, int type, int level, int line, message, filename): self._error_log._receiveGeneric(domain, type, level, line, message, filename) cpdef _clear_error_log(self): self._error_log.clear() property error_log: u"The log of validation errors and warnings." def __get__(self): assert self._error_log is not None, "XPath evaluator not initialised" return self._error_log.copy() include "dtd.pxi" # DTD include "relaxng.pxi" # RelaxNG include "xmlschema.pxi" # XMLSchema include "schematron.pxi" # Schematron (requires libxml2 2.6.21+) ################################################################################ # Public C API include "public-api.pxi"