Daniel Veillard | 5e5c2d0 | 2002-02-09 18:03:01 +0000 | [diff] [blame] | 1 | import libxml2mod |
Daniel Veillard | 85bb5b0 | 2003-12-04 14:12:05 +0000 | [diff] [blame] | 2 | import types |
Daniel Veillard | d2897fd | 2002-01-30 16:37:32 +0000 | [diff] [blame] | 3 | |
Daniel Veillard | a81355e | 2004-09-28 11:08:27 +0000 | [diff] [blame^] | 4 | # The root of all libxml2 errors. |
| 5 | class libxmlError(Exception): pass |
| 6 | |
Daniel Veillard | 1971ee2 | 2002-01-31 20:29:19 +0000 | [diff] [blame] | 7 | # |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 8 | # Errors raised by the wrappers when some tree handling failed. |
| 9 | # |
Daniel Veillard | a81355e | 2004-09-28 11:08:27 +0000 | [diff] [blame^] | 10 | class treeError(libxmlError): |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 11 | def __init__(self, msg): |
| 12 | self.msg = msg |
| 13 | def __str__(self): |
| 14 | return self.msg |
| 15 | |
Daniel Veillard | a81355e | 2004-09-28 11:08:27 +0000 | [diff] [blame^] | 16 | class parserError(libxmlError): |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 17 | def __init__(self, msg): |
| 18 | self.msg = msg |
| 19 | def __str__(self): |
| 20 | return self.msg |
| 21 | |
Daniel Veillard | a81355e | 2004-09-28 11:08:27 +0000 | [diff] [blame^] | 22 | class uriError(libxmlError): |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 23 | def __init__(self, msg): |
| 24 | self.msg = msg |
| 25 | def __str__(self): |
| 26 | return self.msg |
| 27 | |
Daniel Veillard | a81355e | 2004-09-28 11:08:27 +0000 | [diff] [blame^] | 28 | class xpathError(libxmlError): |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 29 | def __init__(self, msg): |
| 30 | self.msg = msg |
| 31 | def __str__(self): |
| 32 | return self.msg |
| 33 | |
Daniel Veillard | c6d4a93 | 2002-09-12 15:00:57 +0000 | [diff] [blame] | 34 | class ioWrapper: |
| 35 | def __init__(self, _obj): |
| 36 | self.__io = _obj |
| 37 | self._o = None |
| 38 | |
| 39 | def io_close(self): |
| 40 | if self.__io == None: |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 41 | return(-1) |
| 42 | self.__io.close() |
| 43 | self.__io = None |
| 44 | return(0) |
Daniel Veillard | c6d4a93 | 2002-09-12 15:00:57 +0000 | [diff] [blame] | 45 | |
| 46 | def io_flush(self): |
| 47 | if self.__io == None: |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 48 | return(-1) |
| 49 | self.__io.flush() |
| 50 | return(0) |
Daniel Veillard | c6d4a93 | 2002-09-12 15:00:57 +0000 | [diff] [blame] | 51 | |
| 52 | def io_read(self, len = -1): |
| 53 | if self.__io == None: |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 54 | return(-1) |
Daniel Veillard | c6d4a93 | 2002-09-12 15:00:57 +0000 | [diff] [blame] | 55 | if len < 0: |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 56 | return(self.__io.read()) |
| 57 | return(self.__io.read(len)) |
Daniel Veillard | c6d4a93 | 2002-09-12 15:00:57 +0000 | [diff] [blame] | 58 | |
| 59 | def io_write(self, str, len = -1): |
| 60 | if self.__io == None: |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 61 | return(-1) |
Daniel Veillard | c6d4a93 | 2002-09-12 15:00:57 +0000 | [diff] [blame] | 62 | if len < 0: |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 63 | return(self.__io.write(str)) |
| 64 | return(self.__io.write(str, len)) |
Daniel Veillard | c6d4a93 | 2002-09-12 15:00:57 +0000 | [diff] [blame] | 65 | |
| 66 | class ioReadWrapper(ioWrapper): |
| 67 | def __init__(self, _obj, enc = ""): |
| 68 | ioWrapper.__init__(self, _obj) |
| 69 | self._o = libxml2mod.xmlCreateInputBuffer(self, enc) |
| 70 | |
| 71 | def __del__(self): |
| 72 | print "__del__" |
| 73 | self.io_close() |
| 74 | if self._o != None: |
| 75 | libxml2mod.xmlFreeParserInputBuffer(self._o) |
| 76 | self._o = None |
| 77 | |
| 78 | def close(self): |
| 79 | self.io_close() |
| 80 | if self._o != None: |
| 81 | libxml2mod.xmlFreeParserInputBuffer(self._o) |
| 82 | self._o = None |
| 83 | |
| 84 | class ioWriteWrapper(ioWrapper): |
| 85 | def __init__(self, _obj, enc = ""): |
Daniel Veillard | 85bb5b0 | 2003-12-04 14:12:05 +0000 | [diff] [blame] | 86 | # print "ioWriteWrapper.__init__", _obj |
| 87 | if type(_obj) == type(''): |
William M. Brack | 37e6394 | 2004-07-12 16:27:37 +0000 | [diff] [blame] | 88 | print "write io from a string" |
| 89 | self.o = None |
| 90 | elif type(_obj) == types.InstanceType: |
| 91 | print "write io from instance of %s" % (_obj.__class__) |
| 92 | ioWrapper.__init__(self, _obj) |
| 93 | self._o = libxml2mod.xmlCreateOutputBuffer(self, enc) |
| 94 | else: |
| 95 | file = libxml2mod.outputBufferGetPythonFile(_obj) |
| 96 | if file != None: |
| 97 | ioWrapper.__init__(self, file) |
| 98 | else: |
| 99 | ioWrapper.__init__(self, _obj) |
| 100 | self._o = _obj |
Daniel Veillard | c6d4a93 | 2002-09-12 15:00:57 +0000 | [diff] [blame] | 101 | |
| 102 | def __del__(self): |
Daniel Veillard | 85bb5b0 | 2003-12-04 14:12:05 +0000 | [diff] [blame] | 103 | # print "__del__" |
Daniel Veillard | c6d4a93 | 2002-09-12 15:00:57 +0000 | [diff] [blame] | 104 | self.io_close() |
| 105 | if self._o != None: |
| 106 | libxml2mod.xmlOutputBufferClose(self._o) |
| 107 | self._o = None |
| 108 | |
Daniel Veillard | 85bb5b0 | 2003-12-04 14:12:05 +0000 | [diff] [blame] | 109 | def flush(self): |
| 110 | self.io_flush() |
| 111 | if self._o != None: |
| 112 | libxml2mod.xmlOutputBufferClose(self._o) |
| 113 | self._o = None |
| 114 | |
Daniel Veillard | c6d4a93 | 2002-09-12 15:00:57 +0000 | [diff] [blame] | 115 | def close(self): |
Daniel Veillard | 85bb5b0 | 2003-12-04 14:12:05 +0000 | [diff] [blame] | 116 | self.io_flush() |
Daniel Veillard | c6d4a93 | 2002-09-12 15:00:57 +0000 | [diff] [blame] | 117 | if self._o != None: |
| 118 | libxml2mod.xmlOutputBufferClose(self._o) |
| 119 | self._o = None |
| 120 | |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 121 | # |
| 122 | # Example of a class to handle SAX events |
| 123 | # |
| 124 | class SAXCallback: |
| 125 | """Base class for SAX handlers""" |
| 126 | def startDocument(self): |
| 127 | """called at the start of the document""" |
| 128 | pass |
| 129 | |
| 130 | def endDocument(self): |
| 131 | """called at the end of the document""" |
| 132 | pass |
| 133 | |
| 134 | def startElement(self, tag, attrs): |
| 135 | """called at the start of every element, tag is the name of |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 136 | the element, attrs is a dictionary of the element's attributes""" |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 137 | pass |
| 138 | |
| 139 | def endElement(self, tag): |
| 140 | """called at the start of every element, tag is the name of |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 141 | the element""" |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 142 | pass |
| 143 | |
| 144 | def characters(self, data): |
| 145 | """called when character data have been read, data is the string |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 146 | containing the data, multiple consecutive characters() callback |
| 147 | are possible.""" |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 148 | pass |
| 149 | |
| 150 | def cdataBlock(self, data): |
| 151 | """called when CDATA section have been read, data is the string |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 152 | containing the data, multiple consecutive cdataBlock() callback |
| 153 | are possible.""" |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 154 | pass |
| 155 | |
| 156 | def reference(self, name): |
| 157 | """called when an entity reference has been found""" |
| 158 | pass |
| 159 | |
| 160 | def ignorableWhitespace(self, data): |
| 161 | """called when potentially ignorable white spaces have been found""" |
| 162 | pass |
| 163 | |
| 164 | def processingInstruction(self, target, data): |
| 165 | """called when a PI has been found, target contains the PI name and |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 166 | data is the associated data in the PI""" |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 167 | pass |
| 168 | |
| 169 | def comment(self, content): |
| 170 | """called when a comment has been found, content contains the comment""" |
| 171 | pass |
| 172 | |
| 173 | def externalSubset(self, name, externalID, systemID): |
| 174 | """called when a DOCTYPE declaration has been found, name is the |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 175 | DTD name and externalID, systemID are the DTD public and system |
| 176 | identifier for that DTd if available""" |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 177 | pass |
| 178 | |
| 179 | def internalSubset(self, name, externalID, systemID): |
| 180 | """called when a DOCTYPE declaration has been found, name is the |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 181 | DTD name and externalID, systemID are the DTD public and system |
| 182 | identifier for that DTD if available""" |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 183 | pass |
| 184 | |
| 185 | def entityDecl(self, name, type, externalID, systemID, content): |
| 186 | """called when an ENTITY declaration has been found, name is the |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 187 | entity name and externalID, systemID are the entity public and |
| 188 | system identifier for that entity if available, type indicates |
| 189 | the entity type, and content reports it's string content""" |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 190 | pass |
| 191 | |
| 192 | def notationDecl(self, name, externalID, systemID): |
| 193 | """called when an NOTATION declaration has been found, name is the |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 194 | notation name and externalID, systemID are the notation public and |
| 195 | system identifier for that notation if available""" |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 196 | pass |
| 197 | |
| 198 | def attributeDecl(self, elem, name, type, defi, defaultValue, nameList): |
| 199 | """called when an ATTRIBUTE definition has been found""" |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 200 | pass |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 201 | |
| 202 | def elementDecl(self, name, type, content): |
| 203 | """called when an ELEMENT definition has been found""" |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 204 | pass |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 205 | |
| 206 | def entityDecl(self, name, publicId, systemID, notationName): |
| 207 | """called when an unparsed ENTITY declaration has been found, |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 208 | name is the entity name and publicId,, systemID are the entity |
| 209 | public and system identifier for that entity if available, |
| 210 | and notationName indicate the associated NOTATION""" |
Daniel Veillard | 8d24cc1 | 2002-03-05 15:41:29 +0000 | [diff] [blame] | 211 | pass |
| 212 | |
| 213 | def warning(self, msg): |
| 214 | print msg |
| 215 | |
| 216 | def error(self, msg): |
| 217 | raise parserError(msg) |
| 218 | |
| 219 | def fatalError(self, msg): |
| 220 | raise parserError(msg) |
| 221 | |
| 222 | # |
Daniel Veillard | 1971ee2 | 2002-01-31 20:29:19 +0000 | [diff] [blame] | 223 | # This class is the ancestor of all the Node classes. It provides |
| 224 | # the basic functionalities shared by all nodes (and handle |
| 225 | # gracefylly the exception), like name, navigation in the tree, |
Daniel Veillard | 1e77438 | 2002-03-06 17:35:40 +0000 | [diff] [blame] | 226 | # doc reference, content access and serializing to a string or URI |
Daniel Veillard | 1971ee2 | 2002-01-31 20:29:19 +0000 | [diff] [blame] | 227 | # |
Daniel Veillard | 36ed529 | 2002-01-30 23:49:06 +0000 | [diff] [blame] | 228 | class xmlCore: |
Daniel Veillard | d2897fd | 2002-01-30 16:37:32 +0000 | [diff] [blame] | 229 | def __init__(self, _obj=None): |
| 230 | if _obj != None: |
Daniel Veillard | 01a6d41 | 2002-02-11 18:42:20 +0000 | [diff] [blame] | 231 | self._o = _obj; |
| 232 | return |
| 233 | self._o = None |
Daniel Veillard | d2897fd | 2002-01-30 16:37:32 +0000 | [diff] [blame] | 234 | def get_parent(self): |
Daniel Veillard | 01a6d41 | 2002-02-11 18:42:20 +0000 | [diff] [blame] | 235 | ret = libxml2mod.parent(self._o) |
| 236 | if ret == None: |
| 237 | return None |
| 238 | return xmlNode(_obj=ret) |
Daniel Veillard | d2897fd | 2002-01-30 16:37:32 +0000 | [diff] [blame] | 239 | def get_children(self): |
Daniel Veillard | 01a6d41 | 2002-02-11 18:42:20 +0000 | [diff] [blame] | 240 | ret = libxml2mod.children(self._o) |
| 241 | if ret == None: |
| 242 | return None |
| 243 | return xmlNode(_obj=ret) |
Daniel Veillard | d2897fd | 2002-01-30 16:37:32 +0000 | [diff] [blame] | 244 | def get_last(self): |
Daniel Veillard | 01a6d41 | 2002-02-11 18:42:20 +0000 | [diff] [blame] | 245 | ret = libxml2mod.last(self._o) |
| 246 | if ret == None: |
| 247 | return None |
| 248 | return xmlNode(_obj=ret) |
Daniel Veillard | d2897fd | 2002-01-30 16:37:32 +0000 | [diff] [blame] | 249 | def get_next(self): |
Daniel Veillard | 01a6d41 | 2002-02-11 18:42:20 +0000 | [diff] [blame] | 250 | ret = libxml2mod.next(self._o) |
| 251 | if ret == None: |
| 252 | return None |
| 253 | return xmlNode(_obj=ret) |
Daniel Veillard | 1971ee2 | 2002-01-31 20:29:19 +0000 | [diff] [blame] | 254 | def get_properties(self): |
Daniel Veillard | 01a6d41 | 2002-02-11 18:42:20 +0000 | [diff] [blame] | 255 | ret = libxml2mod.properties(self._o) |
| 256 | if ret == None: |
| 257 | return None |
| 258 | return xmlAttr(_obj=ret) |
Daniel Veillard | d2897fd | 2002-01-30 16:37:32 +0000 | [diff] [blame] | 259 | def get_prev(self): |
Daniel Veillard | 01a6d41 | 2002-02-11 18:42:20 +0000 | [diff] [blame] | 260 | ret = libxml2mod.prev(self._o) |
| 261 | if ret == None: |
| 262 | return None |
| 263 | return xmlNode(_obj=ret) |
Daniel Veillard | d2897fd | 2002-01-30 16:37:32 +0000 | [diff] [blame] | 264 | def get_content(self): |
Daniel Veillard | 01a6d41 | 2002-02-11 18:42:20 +0000 | [diff] [blame] | 265 | return libxml2mod.xmlNodeGetContent(self._o) |
Daniel Veillard | 51a447a | 2003-01-04 19:42:46 +0000 | [diff] [blame] | 266 | getContent = get_content # why is this duplicate naming needed ? |
Daniel Veillard | d2897fd | 2002-01-30 16:37:32 +0000 | [diff] [blame] | 267 | def get_name(self): |
Daniel Veillard | 01a6d41 | 2002-02-11 18:42:20 +0000 | [diff] [blame] | 268 | return libxml2mod.name(self._o) |
Daniel Veillard | d2897fd | 2002-01-30 16:37:32 +0000 | [diff] [blame] | 269 | def get_type(self): |
Daniel Veillard | 01a6d41 | 2002-02-11 18:42:20 +0000 | [diff] [blame] | 270 | return libxml2mod.type(self._o) |
Daniel Veillard | 51a447a | 2003-01-04 19:42:46 +0000 | [diff] [blame] | 271 | def get_doc(self): |
| 272 | ret = libxml2mod.doc(self._o) |
| 273 | if ret == None: |
| 274 | if self.type in ["document_xml", "document_html"]: |
| 275 | return xmlDoc(_obj=self._o) |
| 276 | else: |
| 277 | return None |
| 278 | return xmlDoc(_obj=ret) |
| 279 | # |
| 280 | # Those are common attributes to nearly all type of nodes |
| 281 | # defined as python2 properties |
| 282 | # |
| 283 | import sys |
| 284 | if float(sys.version[0:3]) < 2.2: |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 285 | def __getattr__(self, attr): |
| 286 | if attr == "parent": |
| 287 | ret = libxml2mod.parent(self._o) |
| 288 | if ret == None: |
| 289 | return None |
| 290 | return xmlNode(_obj=ret) |
| 291 | elif attr == "properties": |
| 292 | ret = libxml2mod.properties(self._o) |
| 293 | if ret == None: |
| 294 | return None |
| 295 | return xmlAttr(_obj=ret) |
| 296 | elif attr == "children": |
| 297 | ret = libxml2mod.children(self._o) |
| 298 | if ret == None: |
| 299 | return None |
| 300 | return xmlNode(_obj=ret) |
| 301 | elif attr == "last": |
| 302 | ret = libxml2mod.last(self._o) |
| 303 | if ret == None: |
| 304 | return None |
| 305 | return xmlNode(_obj=ret) |
| 306 | elif attr == "next": |
| 307 | ret = libxml2mod.next(self._o) |
| 308 | if ret == None: |
| 309 | return None |
| 310 | return xmlNode(_obj=ret) |
| 311 | elif attr == "prev": |
| 312 | ret = libxml2mod.prev(self._o) |
| 313 | if ret == None: |
| 314 | return None |
| 315 | return xmlNode(_obj=ret) |
| 316 | elif attr == "content": |
| 317 | return libxml2mod.xmlNodeGetContent(self._o) |
| 318 | elif attr == "name": |
| 319 | return libxml2mod.name(self._o) |
| 320 | elif attr == "type": |
| 321 | return libxml2mod.type(self._o) |
| 322 | elif attr == "doc": |
| 323 | ret = libxml2mod.doc(self._o) |
| 324 | if ret == None: |
| 325 | if self.type == "document_xml" or self.type == "document_html": |
| 326 | return xmlDoc(_obj=self._o) |
| 327 | else: |
| 328 | return None |
| 329 | return xmlDoc(_obj=ret) |
| 330 | raise AttributeError,attr |
Daniel Veillard | 51a447a | 2003-01-04 19:42:46 +0000 | [diff] [blame] | 331 | else: |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 332 | parent = property(get_parent, None, None, "Parent node") |
| 333 | children = property(get_children, None, None, "First child node") |
| 334 | last = property(get_last, None, None, "Last sibling node") |
| 335 | next = property(get_next, None, None, "Next sibling node") |
| 336 | prev = property(get_prev, None, None, "Previous sibling node") |
| 337 | properties = property(get_properties, None, None, "List of properies") |
| 338 | content = property(get_content, None, None, "Content of this node") |
| 339 | name = property(get_name, None, None, "Node name") |
| 340 | type = property(get_type, None, None, "Node type") |
| 341 | doc = property(get_doc, None, None, "The document this node belongs to") |
Daniel Veillard | 1e77438 | 2002-03-06 17:35:40 +0000 | [diff] [blame] | 342 | |
| 343 | # |
| 344 | # Serialization routines, the optional arguments have the following |
| 345 | # meaning: |
| 346 | # encoding: string to ask saving in a specific encoding |
Daniel Veillard | 51a447a | 2003-01-04 19:42:46 +0000 | [diff] [blame] | 347 | # indent: if 1 the serializer is asked to indent the output |
Daniel Veillard | 1e77438 | 2002-03-06 17:35:40 +0000 | [diff] [blame] | 348 | # |
| 349 | def serialize(self, encoding = None, format = 0): |
| 350 | return libxml2mod.serializeNode(self._o, encoding, format) |
| 351 | def saveTo(self, file, encoding = None, format = 0): |
| 352 | return libxml2mod.saveNodeTo(self._o, file, encoding, format) |
Daniel Veillard | 01a6d41 | 2002-02-11 18:42:20 +0000 | [diff] [blame] | 353 | |
Daniel Veillard | f742d34 | 2002-03-07 00:05:35 +0000 | [diff] [blame] | 354 | # |
Daniel Veillard | d5e198a | 2004-03-09 09:03:28 +0000 | [diff] [blame] | 355 | # Canonicalization routines: |
| 356 | # |
| 357 | # nodes: the node set (tuple or list) to be included in the |
| 358 | # canonized image or None if all document nodes should be |
| 359 | # included. |
| 360 | # exclusive: the exclusive flag (0 - non-exclusive |
| 361 | # canonicalization; otherwise - exclusive canonicalization) |
| 362 | # prefixes: the list of inclusive namespace prefixes (strings), |
| 363 | # or None if there is no inclusive namespaces (only for |
| 364 | # exclusive canonicalization, ignored otherwise) |
| 365 | # with_comments: include comments in the result (!=0) or not |
| 366 | # (==0) |
| 367 | def c14nMemory(self, |
| 368 | nodes=None, |
| 369 | exclusive=0, |
| 370 | prefixes=None, |
| 371 | with_comments=0): |
| 372 | if nodes: |
| 373 | nodes = map(lambda n: n._o, nodes) |
| 374 | return libxml2mod.xmlC14NDocDumpMemory( |
| 375 | self.get_doc()._o, |
| 376 | nodes, |
| 377 | exclusive != 0, |
| 378 | prefixes, |
| 379 | with_comments != 0) |
| 380 | def c14nSaveTo(self, |
| 381 | file, |
| 382 | nodes=None, |
| 383 | exclusive=0, |
| 384 | prefixes=None, |
| 385 | with_comments=0): |
| 386 | if nodes: |
| 387 | nodes = map(lambda n: n._o, nodes) |
| 388 | return libxml2mod.xmlC14NDocSaveTo( |
| 389 | self.get_doc()._o, |
| 390 | nodes, |
| 391 | exclusive != 0, |
| 392 | prefixes, |
| 393 | with_comments != 0, |
| 394 | file) |
| 395 | |
| 396 | # |
Daniel Veillard | f742d34 | 2002-03-07 00:05:35 +0000 | [diff] [blame] | 397 | # Selecting nodes using XPath, a bit slow because the context |
| 398 | # is allocated/freed every time but convenient. |
| 399 | # |
| 400 | def xpathEval(self, expr): |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 401 | doc = self.doc |
| 402 | if doc == None: |
| 403 | return None |
| 404 | ctxt = doc.xpathNewContext() |
| 405 | ctxt.setContextNode(self) |
| 406 | res = ctxt.xpathEval(expr) |
| 407 | ctxt.xpathFreeContext() |
| 408 | return res |
Daniel Veillard | 51a447a | 2003-01-04 19:42:46 +0000 | [diff] [blame] | 409 | |
Daniel Veillard | f88d8cf | 2003-12-08 10:25:02 +0000 | [diff] [blame] | 410 | # # |
| 411 | # # Selecting nodes using XPath, faster because the context |
| 412 | # # is allocated just once per xmlDoc. |
| 413 | # # |
| 414 | # # Removed: DV memleaks c.f. #126735 |
| 415 | # # |
| 416 | # def xpathEval2(self, expr): |
| 417 | # doc = self.doc |
| 418 | # if doc == None: |
| 419 | # return None |
| 420 | # try: |
| 421 | # doc._ctxt.setContextNode(self) |
| 422 | # except: |
| 423 | # doc._ctxt = doc.xpathNewContext() |
| 424 | # doc._ctxt.setContextNode(self) |
| 425 | # res = doc._ctxt.xpathEval(expr) |
| 426 | # return res |
Daniel Veillard | 51a447a | 2003-01-04 19:42:46 +0000 | [diff] [blame] | 427 | def xpathEval2(self, expr): |
Daniel Veillard | f88d8cf | 2003-12-08 10:25:02 +0000 | [diff] [blame] | 428 | return self.xpathEval(expr) |
Daniel Veillard | 51a447a | 2003-01-04 19:42:46 +0000 | [diff] [blame] | 429 | |
| 430 | # support for python2 iterators |
| 431 | def walk_depth_first(self): |
| 432 | return xmlCoreDepthFirstItertor(self) |
| 433 | def walk_breadth_first(self): |
| 434 | return xmlCoreBreadthFirstItertor(self) |
| 435 | __iter__ = walk_depth_first |
| 436 | |
| 437 | def free(self): |
| 438 | try: |
| 439 | self.doc._ctxt.xpathFreeContext() |
| 440 | except: |
| 441 | pass |
Daniel Veillard | f88d8cf | 2003-12-08 10:25:02 +0000 | [diff] [blame] | 442 | libxml2mod.xmlFreeDoc(self._o) |
Daniel Veillard | 51a447a | 2003-01-04 19:42:46 +0000 | [diff] [blame] | 443 | |
| 444 | |
| 445 | # |
| 446 | # implements the depth-first iterator for libxml2 DOM tree |
| 447 | # |
| 448 | class xmlCoreDepthFirstItertor: |
| 449 | def __init__(self, node): |
| 450 | self.node = node |
| 451 | self.parents = [] |
| 452 | def __iter__(self): |
| 453 | return self |
| 454 | def next(self): |
| 455 | while 1: |
| 456 | if self.node: |
| 457 | ret = self.node |
| 458 | self.parents.append(self.node) |
| 459 | self.node = self.node.children |
| 460 | return ret |
| 461 | try: |
| 462 | parent = self.parents.pop() |
| 463 | except IndexError: |
| 464 | raise StopIteration |
| 465 | self.node = parent.next |
| 466 | |
| 467 | # |
| 468 | # implements the breadth-first iterator for libxml2 DOM tree |
| 469 | # |
| 470 | class xmlCoreBreadthFirstItertor: |
| 471 | def __init__(self, node): |
| 472 | self.node = node |
| 473 | self.parents = [] |
| 474 | def __iter__(self): |
| 475 | return self |
| 476 | def next(self): |
| 477 | while 1: |
| 478 | if self.node: |
| 479 | ret = self.node |
| 480 | self.parents.append(self.node) |
| 481 | self.node = self.node.next |
| 482 | return ret |
| 483 | try: |
| 484 | parent = self.parents.pop() |
| 485 | except IndexError: |
| 486 | raise StopIteration |
| 487 | self.node = parent.children |
| 488 | |
Daniel Veillard | 36ed529 | 2002-01-30 23:49:06 +0000 | [diff] [blame] | 489 | # |
Daniel Veillard | 1971ee2 | 2002-01-31 20:29:19 +0000 | [diff] [blame] | 490 | # converters to present a nicer view of the XPath returns |
| 491 | # |
| 492 | def nodeWrap(o): |
| 493 | # TODO try to cast to the most appropriate node class |
Daniel Veillard | 1f8658a | 2004-08-14 21:46:31 +0000 | [diff] [blame] | 494 | name = libxml2mod.type(o) |
Daniel Veillard | 1971ee2 | 2002-01-31 20:29:19 +0000 | [diff] [blame] | 495 | if name == "element" or name == "text": |
| 496 | return xmlNode(_obj=o) |
| 497 | if name == "attribute": |
| 498 | return xmlAttr(_obj=o) |
| 499 | if name[0:8] == "document": |
| 500 | return xmlDoc(_obj=o) |
Daniel Veillard | 1f8658a | 2004-08-14 21:46:31 +0000 | [diff] [blame] | 501 | if name == "namespace": |
Daniel Veillard | 1971ee2 | 2002-01-31 20:29:19 +0000 | [diff] [blame] | 502 | return xmlNs(_obj=o) |
| 503 | if name == "elem_decl": |
| 504 | return xmlElement(_obj=o) |
| 505 | if name == "attribute_decl": |
Daniel Veillard | 1f8658a | 2004-08-14 21:46:31 +0000 | [diff] [blame] | 506 | return xmlAttribute(_obj=o) |
Daniel Veillard | 1971ee2 | 2002-01-31 20:29:19 +0000 | [diff] [blame] | 507 | if name == "entity_decl": |
| 508 | return xmlEntity(_obj=o) |
| 509 | if name == "dtd": |
Daniel Veillard | e59494f | 2003-01-04 16:35:29 +0000 | [diff] [blame] | 510 | return xmlDtd(_obj=o) |
Daniel Veillard | 1971ee2 | 2002-01-31 20:29:19 +0000 | [diff] [blame] | 511 | return xmlNode(_obj=o) |
| 512 | |
| 513 | def xpathObjectRet(o): |
| 514 | if type(o) == type([]) or type(o) == type(()): |
| 515 | ret = map(lambda x: nodeWrap(x), o) |
Daniel Veillard | 01a6d41 | 2002-02-11 18:42:20 +0000 | [diff] [blame] | 516 | return ret |
Daniel Veillard | 1971ee2 | 2002-01-31 20:29:19 +0000 | [diff] [blame] | 517 | return o |
| 518 | |
| 519 | # |
Daniel Veillard | a7340c8 | 2002-02-01 17:56:45 +0000 | [diff] [blame] | 520 | # register an XPath function |
| 521 | # |
| 522 | def registerXPathFunction(ctxt, name, ns_uri, f): |
Daniel Veillard | 5e5c2d0 | 2002-02-09 18:03:01 +0000 | [diff] [blame] | 523 | ret = libxml2mod.xmlRegisterXPathFunction(ctxt, name, ns_uri, f) |
Daniel Veillard | a7340c8 | 2002-02-01 17:56:45 +0000 | [diff] [blame] | 524 | |
| 525 | # |
Daniel Veillard | f25b4ca | 2002-12-27 15:18:35 +0000 | [diff] [blame] | 526 | # For the xmlTextReader parser configuration |
| 527 | # |
| 528 | PARSER_LOADDTD=1 |
| 529 | PARSER_DEFAULTATTRS=2 |
| 530 | PARSER_VALIDATE=3 |
Daniel Veillard | e18fc18 | 2002-12-28 22:56:33 +0000 | [diff] [blame] | 531 | PARSER_SUBST_ENTITIES=4 |
Daniel Veillard | f25b4ca | 2002-12-27 15:18:35 +0000 | [diff] [blame] | 532 | |
| 533 | # |
Daniel Veillard | 417be3a | 2003-01-20 21:26:34 +0000 | [diff] [blame] | 534 | # For the error callback severities |
Daniel Veillard | 26f7026 | 2003-01-16 22:45:08 +0000 | [diff] [blame] | 535 | # |
Daniel Veillard | 417be3a | 2003-01-20 21:26:34 +0000 | [diff] [blame] | 536 | PARSER_SEVERITY_VALIDITY_WARNING=1 |
| 537 | PARSER_SEVERITY_VALIDITY_ERROR=2 |
| 538 | PARSER_SEVERITY_WARNING=3 |
| 539 | PARSER_SEVERITY_ERROR=4 |
Daniel Veillard | 26f7026 | 2003-01-16 22:45:08 +0000 | [diff] [blame] | 540 | |
| 541 | # |
Daniel Veillard | 3e20a29 | 2003-01-10 13:14:40 +0000 | [diff] [blame] | 542 | # register the libxml2 error handler |
Daniel Veillard | 36ed529 | 2002-01-30 23:49:06 +0000 | [diff] [blame] | 543 | # |
Daniel Veillard | 3e20a29 | 2003-01-10 13:14:40 +0000 | [diff] [blame] | 544 | def registerErrorHandler(f, ctx): |
| 545 | """Register a Python written function to for error reporting. |
| 546 | The function is called back as f(ctx, error). """ |
| 547 | import sys |
| 548 | if not sys.modules.has_key('libxslt'): |
| 549 | # normal behaviour when libxslt is not imported |
| 550 | ret = libxml2mod.xmlRegisterErrorHandler(f,ctx) |
| 551 | else: |
| 552 | # when libxslt is already imported, one must |
| 553 | # use libxst's error handler instead |
| 554 | import libxslt |
| 555 | ret = libxslt.registerErrorHandler(f,ctx) |
| 556 | return ret |
| 557 | |
Daniel Veillard | e6227e0 | 2003-01-14 11:42:39 +0000 | [diff] [blame] | 558 | class parserCtxtCore: |
| 559 | |
| 560 | def __init__(self, _obj=None): |
| 561 | if _obj != None: |
| 562 | self._o = _obj; |
| 563 | return |
| 564 | self._o = None |
| 565 | |
| 566 | def __del__(self): |
| 567 | if self._o != None: |
| 568 | libxml2mod.xmlFreeParserCtxt(self._o) |
William M. Brack | 1d75c8a | 2003-10-27 13:48:16 +0000 | [diff] [blame] | 569 | self._o = None |
Daniel Veillard | e6227e0 | 2003-01-14 11:42:39 +0000 | [diff] [blame] | 570 | |
Daniel Veillard | 417be3a | 2003-01-20 21:26:34 +0000 | [diff] [blame] | 571 | def setErrorHandler(self,f,arg): |
| 572 | """Register an error handler that will be called back as |
| 573 | f(arg,msg,severity,reserved). |
| 574 | |
| 575 | @reserved is currently always None.""" |
| 576 | libxml2mod.xmlParserCtxtSetErrorHandler(self._o,f,arg) |
Daniel Veillard | e6227e0 | 2003-01-14 11:42:39 +0000 | [diff] [blame] | 577 | |
Daniel Veillard | 417be3a | 2003-01-20 21:26:34 +0000 | [diff] [blame] | 578 | def getErrorHandler(self): |
| 579 | """Return (f,arg) as previously registered with setErrorHandler |
| 580 | or (None,None).""" |
| 581 | return libxml2mod.xmlParserCtxtGetErrorHandler(self._o) |
| 582 | |
Daniel Veillard | 5439624 | 2003-04-23 07:36:50 +0000 | [diff] [blame] | 583 | def addLocalCatalog(self, uri): |
| 584 | """Register a local catalog with the parser""" |
| 585 | return libxml2mod.addLocalCatalog(self._o, uri) |
| 586 | |
| 587 | |
Daniel Veillard | 417be3a | 2003-01-20 21:26:34 +0000 | [diff] [blame] | 588 | def _xmlTextReaderErrorFunc((f,arg),msg,severity,locator): |
| 589 | """Intermediate callback to wrap the locator""" |
| 590 | return f(arg,msg,severity,xmlTextReaderLocator(locator)) |
Daniel Veillard | e6227e0 | 2003-01-14 11:42:39 +0000 | [diff] [blame] | 591 | |
Daniel Veillard | 26f7026 | 2003-01-16 22:45:08 +0000 | [diff] [blame] | 592 | class xmlTextReaderCore: |
| 593 | |
| 594 | def __init__(self, _obj=None): |
| 595 | self.input = None |
| 596 | if _obj != None:self._o = _obj;return |
| 597 | self._o = None |
| 598 | |
| 599 | def __del__(self): |
| 600 | if self._o != None: |
| 601 | libxml2mod.xmlFreeTextReader(self._o) |
| 602 | self._o = None |
| 603 | |
Daniel Veillard | 417be3a | 2003-01-20 21:26:34 +0000 | [diff] [blame] | 604 | def SetErrorHandler(self,f,arg): |
Daniel Veillard | 26f7026 | 2003-01-16 22:45:08 +0000 | [diff] [blame] | 605 | """Register an error handler that will be called back as |
Daniel Veillard | 417be3a | 2003-01-20 21:26:34 +0000 | [diff] [blame] | 606 | f(arg,msg,severity,locator).""" |
| 607 | if f is None: |
| 608 | libxml2mod.xmlTextReaderSetErrorHandler(\ |
| 609 | self._o,None,None) |
| 610 | else: |
| 611 | libxml2mod.xmlTextReaderSetErrorHandler(\ |
| 612 | self._o,_xmlTextReaderErrorFunc,(f,arg)) |
Daniel Veillard | 26f7026 | 2003-01-16 22:45:08 +0000 | [diff] [blame] | 613 | |
Daniel Veillard | 417be3a | 2003-01-20 21:26:34 +0000 | [diff] [blame] | 614 | def GetErrorHandler(self): |
Daniel Veillard | 26f7026 | 2003-01-16 22:45:08 +0000 | [diff] [blame] | 615 | """Return (f,arg) as previously registered with setErrorHandler |
| 616 | or (None,None).""" |
Daniel Veillard | 417be3a | 2003-01-20 21:26:34 +0000 | [diff] [blame] | 617 | f,arg = libxml2mod.xmlTextReaderGetErrorHandler(self._o) |
| 618 | if f is None: |
| 619 | return None,None |
| 620 | else: |
| 621 | # assert f is _xmlTextReaderErrorFunc |
| 622 | return arg |
Daniel Veillard | 26f7026 | 2003-01-16 22:45:08 +0000 | [diff] [blame] | 623 | |
Daniel Veillard | f93a866 | 2004-07-01 12:56:30 +0000 | [diff] [blame] | 624 | # |
| 625 | # The cleanup now goes though a wrappe in libxml.c |
| 626 | # |
| 627 | def cleanupParser(): |
| 628 | libxml2mod.xmlPythonCleanupParser() |
Daniel Veillard | 87ab1c1 | 2003-12-21 13:01:56 +0000 | [diff] [blame] | 629 | |
Daniel Veillard | 3e20a29 | 2003-01-10 13:14:40 +0000 | [diff] [blame] | 630 | # WARNING WARNING WARNING WARNING WARNING WARNING WARNING WARNING WARNING |
| 631 | # |
| 632 | # Everything before this line comes from libxml.py |
| 633 | # Everything after this line is automatically generated |
| 634 | # |
| 635 | # WARNING WARNING WARNING WARNING WARNING WARNING WARNING WARNING WARNING |
Daniel Veillard | 1971ee2 | 2002-01-31 20:29:19 +0000 | [diff] [blame] | 636 | |