blob: dcfe87268735ea351c6788ceeccebaa7c7a062f5 [file] [log] [blame]
Guido van Rossuma219efa1997-11-18 15:09:54 +00001# A parser for XML, using the derived class as static DTD.
Guido van Rossum5d68e8e1997-11-18 15:27:20 +00002# Author: Sjoerd Mullender.
Guido van Rossuma219efa1997-11-18 15:09:54 +00003
4import re
5import string
6
7
Guido van Rossumb35d6841999-08-26 15:52:33 +00008version = '0.3'
Guido van Rossum7e07b381998-04-03 16:02:39 +00009
Guido van Rossuma219efa1997-11-18 15:09:54 +000010# Regular expressions used for parsing
11
Guido van Rossumeeb2f321998-10-19 13:28:26 +000012_S = '[ \t\r\n]+' # white space
13_opS = '[ \t\r\n]*' # optional white space
14_Name = '[a-zA-Z_:][-a-zA-Z0-9._:]*' # valid XML name
15_QStr = "(?:'[^']*'|\"[^\"]*\")" # quoted XML string
Guido van Rossum7e07b381998-04-03 16:02:39 +000016illegal = re.compile('[^\t\r\n -\176\240-\377]') # illegal chars in content
17interesting = re.compile('[]&<]')
Guido van Rossuma219efa1997-11-18 15:09:54 +000018
Guido van Rossum7e07b381998-04-03 16:02:39 +000019amp = re.compile('&')
20ref = re.compile('&(' + _Name + '|#[0-9]+|#x[0-9a-fA-F]+)[^-a-zA-Z0-9._:]')
Guido van Rossuma219efa1997-11-18 15:09:54 +000021entityref = re.compile('&(?P<name>' + _Name + ')[^-a-zA-Z0-9._:]')
22charref = re.compile('&#(?P<char>[0-9]+[^0-9]|x[0-9a-fA-F]+[^0-9a-fA-F])')
Guido van Rossum7e07b381998-04-03 16:02:39 +000023space = re.compile(_S + '$')
Guido van Rossuma219efa1997-11-18 15:09:54 +000024newline = re.compile('\n')
25
Guido van Rossumeeb2f321998-10-19 13:28:26 +000026attrfind = re.compile(
27 _S + '(?P<name>' + _Name + ')'
28 '(' + _opS + '=' + _opS +
29 '(?P<value>'+_QStr+'|[-a-zA-Z0-9.:+*%?!()_#=~]+))?')
Guido van Rossuma219efa1997-11-18 15:09:54 +000030starttagopen = re.compile('<' + _Name)
Guido van Rossuma219efa1997-11-18 15:09:54 +000031starttagend = re.compile(_opS + '(?P<slash>/?)>')
Guido van Rossumeeb2f321998-10-19 13:28:26 +000032starttagmatch = re.compile('<(?P<tagname>'+_Name+')'
33 '(?P<attrs>(?:'+attrfind.pattern+')*)'+
34 starttagend.pattern)
35endtagopen = re.compile('</')
Guido van Rossum7e07b381998-04-03 16:02:39 +000036endbracket = re.compile(_opS + '>')
Guido van Rossumeeb2f321998-10-19 13:28:26 +000037endbracketfind = re.compile('(?:[^>\'"]|'+_QStr+')*>')
Guido van Rossuma219efa1997-11-18 15:09:54 +000038tagfind = re.compile(_Name)
Guido van Rossum02505e41998-01-29 14:55:24 +000039cdataopen = re.compile(r'<!\[CDATA\[')
40cdataclose = re.compile(r'\]\]>')
Guido van Rossum7e07b381998-04-03 16:02:39 +000041# this matches one of the following:
42# SYSTEM SystemLiteral
43# PUBLIC PubidLiteral SystemLiteral
Guido van Rossumeeb2f321998-10-19 13:28:26 +000044_SystemLiteral = '(?P<%s>'+_QStr+')'
Guido van Rossum7e07b381998-04-03 16:02:39 +000045_PublicLiteral = '(?P<%s>"[-\'()+,./:=?;!*#@$_%% \n\ra-zA-Z0-9]*"|' \
46 "'[-()+,./:=?;!*#@$_%% \n\ra-zA-Z0-9]*')"
47_ExternalId = '(?:SYSTEM|' \
48 'PUBLIC'+_S+_PublicLiteral%'pubid'+ \
49 ')'+_S+_SystemLiteral%'syslit'
50doctype = re.compile('<!DOCTYPE'+_S+'(?P<name>'+_Name+')'
51 '(?:'+_S+_ExternalId+')?'+_opS)
52xmldecl = re.compile('<\?xml'+_S+
Guido van Rossumeeb2f321998-10-19 13:28:26 +000053 'version'+_opS+'='+_opS+'(?P<version>'+_QStr+')'+
Guido van Rossum7e07b381998-04-03 16:02:39 +000054 '(?:'+_S+'encoding'+_opS+'='+_opS+
55 "(?P<encoding>'[A-Za-z][-A-Za-z0-9._]*'|"
56 '"[A-Za-z][-A-Za-z0-9._]*"))?'
57 '(?:'+_S+'standalone'+_opS+'='+_opS+
58 '(?P<standalone>\'(?:yes|no)\'|"(?:yes|no)"))?'+
59 _opS+'\?>')
60procopen = re.compile(r'<\?(?P<proc>' + _Name + ')' + _opS)
Guido van Rossum02505e41998-01-29 14:55:24 +000061procclose = re.compile(_opS + r'\?>')
Guido van Rossuma219efa1997-11-18 15:09:54 +000062commentopen = re.compile('<!--')
63commentclose = re.compile('-->')
64doubledash = re.compile('--')
Guido van Rossum7e07b381998-04-03 16:02:39 +000065attrtrans = string.maketrans(' \r\n\t', ' ')
Guido van Rossuma219efa1997-11-18 15:09:54 +000066
Guido van Rossumb083a9f1998-12-18 20:17:13 +000067# definitions for XML namespaces
68_NCName = '[a-zA-Z_][-a-zA-Z0-9._]*' # XML Name, minus the ":"
69ncname = re.compile(_NCName + '$')
70qname = re.compile('(?:(?P<prefix>' + _NCName + '):)?' # optional prefix
71 '(?P<local>' + _NCName + ')$')
72
73xmlns = re.compile('xmlns(?::(?P<ncname>'+_NCName+'))?$')
Guido van Rossuma219efa1997-11-18 15:09:54 +000074
75# XML parser base class -- find tags and call handler functions.
76# Usage: p = XMLParser(); p.feed(data); ...; p.close().
Guido van Rossum5d68e8e1997-11-18 15:27:20 +000077# The dtd is defined by deriving a class which defines methods with
78# special names to handle tags: start_foo and end_foo to handle <foo>
79# and </foo>, respectively. The data between tags is passed to the
80# parser by calling self.handle_data() with some data as argument (the
Guido van Rossumb35d6841999-08-26 15:52:33 +000081# data may be split up in arbutrary chunks).
Guido van Rossuma219efa1997-11-18 15:09:54 +000082
83class XMLParser:
Guido van Rossumb083a9f1998-12-18 20:17:13 +000084 attributes = {} # default, to be overridden
85 elements = {} # default, to be overridden
Guido van Rossuma219efa1997-11-18 15:09:54 +000086
Guido van Rossumb35d6841999-08-26 15:52:33 +000087 # parsing options, settable using keyword args in __init__
88 __accept_unquoted_attributes = 0
89 __accept_missing_endtag_name = 0
90 __map_case = 0
91 __accept_utf8 = 0
92
Guido van Rossuma219efa1997-11-18 15:09:54 +000093 # Interface -- initialize and reset this instance
Guido van Rossumb35d6841999-08-26 15:52:33 +000094 def __init__(self, **kw):
Guido van Rossumcc2c2911999-02-02 17:54:38 +000095 self.__fixed = 0
Guido van Rossumb35d6841999-08-26 15:52:33 +000096 if kw.has_key('accept_unquoted_attributes'):
97 self.__accept_unquoted_attributes = kw['accept_unquoted_attributes']
98 if kw.has_key('accept_missing_endtag_name'):
99 self.__accept_missing_endtag_name = kw['accept_missing_endtag_name']
100 if kw.has_key('map_case'):
101 self.__map_case = kw['map_case']
102 if kw.has_key('accept_utf8'):
103 self.__accept_utf8 = kw['accept_utf8']
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000104 self.reset()
Guido van Rossumfe649351999-02-01 15:35:13 +0000105
106 def __fixelements(self):
Guido van Rossumcc2c2911999-02-02 17:54:38 +0000107 self.__fixed = 1
Guido van Rossumfe649351999-02-01 15:35:13 +0000108 self.elements = {}
109 self.__fixdict(self.__dict__)
110 self.__fixclass(self.__class__)
111
112 def __fixclass(self, kl):
113 self.__fixdict(kl.__dict__)
114 for k in kl.__bases__:
115 self.__fixclass(k)
116
117 def __fixdict(self, dict):
Guido van Rossum2e3f7be1999-02-02 22:15:24 +0000118 for key in dict.keys():
Guido van Rossumfe649351999-02-01 15:35:13 +0000119 if key[:6] == 'start_':
Guido van Rossum2e3f7be1999-02-02 22:15:24 +0000120 tag = key[6:]
121 start, end = self.elements.get(tag, (None, None))
Guido van Rossumfe649351999-02-01 15:35:13 +0000122 if start is None:
Guido van Rossum2e3f7be1999-02-02 22:15:24 +0000123 self.elements[tag] = getattr(self, key), end
Guido van Rossumfe649351999-02-01 15:35:13 +0000124 elif key[:4] == 'end_':
Guido van Rossum2e3f7be1999-02-02 22:15:24 +0000125 tag = key[4:]
126 start, end = self.elements.get(tag, (None, None))
Guido van Rossumfe649351999-02-01 15:35:13 +0000127 if end is None:
Guido van Rossum2e3f7be1999-02-02 22:15:24 +0000128 self.elements[tag] = start, getattr(self, key)
Guido van Rossuma219efa1997-11-18 15:09:54 +0000129
130 # Interface -- reset this instance. Loses all unprocessed data
131 def reset(self):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000132 self.rawdata = ''
133 self.stack = []
134 self.nomoretags = 0
135 self.literal = 0
136 self.lineno = 1
137 self.__at_start = 1
138 self.__seen_doctype = None
139 self.__seen_starttag = 0
Guido van Rossum91c8f591998-12-21 18:15:28 +0000140 self.__use_namespaces = 0
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000141 self.__namespaces = {'xml':None} # xml is implicitly declared
Guido van Rossumcc2c2911999-02-02 17:54:38 +0000142 # backward compatipibility hack: if elements not overridden,
143 # fill it in ourselves
144 if self.elements is XMLParser.elements:
145 self.__fixelements()
Guido van Rossuma219efa1997-11-18 15:09:54 +0000146
147 # For derived classes only -- enter literal mode (CDATA) till EOF
148 def setnomoretags(self):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000149 self.nomoretags = self.literal = 1
Guido van Rossuma219efa1997-11-18 15:09:54 +0000150
151 # For derived classes only -- enter literal mode (CDATA)
152 def setliteral(self, *args):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000153 self.literal = 1
Guido van Rossuma219efa1997-11-18 15:09:54 +0000154
155 # Interface -- feed some data to the parser. Call this as
156 # often as you want, with as little or as much text as you
157 # want (may include '\n'). (This just saves the text, all the
158 # processing is done by goahead().)
159 def feed(self, data):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000160 self.rawdata = self.rawdata + data
161 self.goahead(0)
Guido van Rossuma219efa1997-11-18 15:09:54 +0000162
163 # Interface -- handle the remaining data
164 def close(self):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000165 self.goahead(1)
Guido van Rossumcc2c2911999-02-02 17:54:38 +0000166 if self.__fixed:
167 self.__fixed = 0
168 # remove self.elements so that we don't leak
169 del self.elements
Guido van Rossuma219efa1997-11-18 15:09:54 +0000170
171 # Interface -- translate references
Guido van Rossum7e07b381998-04-03 16:02:39 +0000172 def translate_references(self, data, all = 1):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000173 i = 0
174 while 1:
Guido van Rossum7e07b381998-04-03 16:02:39 +0000175 res = amp.search(data, i)
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000176 if res is None:
Guido van Rossum7e07b381998-04-03 16:02:39 +0000177 return data
178 res = ref.match(data, res.start(0))
179 if res is None:
180 self.syntax_error("bogus `&'")
181 i =i+1
182 continue
183 i = res.end(0)
184 if data[i - 1] != ';':
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000185 self.syntax_error("`;' missing after entity/char reference")
Guido van Rossum7e07b381998-04-03 16:02:39 +0000186 i = i-1
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000187 str = res.group(1)
Guido van Rossum7e07b381998-04-03 16:02:39 +0000188 pre = data[:res.start(0)]
189 post = data[i:]
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000190 if str[0] == '#':
191 if str[1] == 'x':
Guido van Rossum7e07b381998-04-03 16:02:39 +0000192 str = chr(string.atoi(str[2:], 16))
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000193 else:
Guido van Rossum7e07b381998-04-03 16:02:39 +0000194 str = chr(string.atoi(str[1:]))
195 data = pre + str + post
196 i = res.start(0)+len(str)
197 elif all:
198 if self.entitydefs.has_key(str):
199 data = pre + self.entitydefs[str] + post
200 i = res.start(0) # rescan substituted text
201 else:
Guido van Rossumcc2c2911999-02-02 17:54:38 +0000202 self.syntax_error("reference to unknown entity `&%s;'" % str)
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000203 # can't do it, so keep the entity ref in
Guido van Rossum7e07b381998-04-03 16:02:39 +0000204 data = pre + '&' + str + ';' + post
205 i = res.start(0) + len(str) + 2
206 else:
207 # just translating character references
208 pass # i is already postioned correctly
Guido van Rossuma219efa1997-11-18 15:09:54 +0000209
210 # Internal -- handle data as far as reasonable. May leave state
211 # and data to be processed by a subsequent call. If 'end' is
212 # true, force handling all data as if followed by EOF marker.
213 def goahead(self, end):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000214 rawdata = self.rawdata
215 i = 0
216 n = len(rawdata)
217 while i < n:
218 if i > 0:
219 self.__at_start = 0
220 if self.nomoretags:
221 data = rawdata[i:n]
222 self.handle_data(data)
223 self.lineno = self.lineno + string.count(data, '\n')
224 i = n
225 break
226 res = interesting.search(rawdata, i)
227 if res:
228 j = res.start(0)
229 else:
230 j = n
231 if i < j:
Guido van Rossum91c8f591998-12-21 18:15:28 +0000232 data = rawdata[i:j]
233 if self.__at_start and space.match(data) is None:
Guido van Rossum7e07b381998-04-03 16:02:39 +0000234 self.syntax_error('illegal data at start of file')
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000235 self.__at_start = 0
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000236 if not self.stack and space.match(data) is None:
Guido van Rossum7e07b381998-04-03 16:02:39 +0000237 self.syntax_error('data not in content')
Guido van Rossumb35d6841999-08-26 15:52:33 +0000238 if not self.__accept_utf8 and illegal.search(data):
Guido van Rossum7e07b381998-04-03 16:02:39 +0000239 self.syntax_error('illegal character in content')
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000240 self.handle_data(data)
241 self.lineno = self.lineno + string.count(data, '\n')
242 i = j
243 if i == n: break
244 if rawdata[i] == '<':
245 if starttagopen.match(rawdata, i):
246 if self.literal:
247 data = rawdata[i]
248 self.handle_data(data)
249 self.lineno = self.lineno + string.count(data, '\n')
250 i = i+1
251 continue
252 k = self.parse_starttag(i)
253 if k < 0: break
254 self.__seen_starttag = 1
255 self.lineno = self.lineno + string.count(rawdata[i:k], '\n')
256 i = k
257 continue
258 if endtagopen.match(rawdata, i):
259 k = self.parse_endtag(i)
260 if k < 0: break
261 self.lineno = self.lineno + string.count(rawdata[i:k], '\n')
262 i = k
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000263 continue
264 if commentopen.match(rawdata, i):
265 if self.literal:
266 data = rawdata[i]
267 self.handle_data(data)
268 self.lineno = self.lineno + string.count(data, '\n')
269 i = i+1
270 continue
271 k = self.parse_comment(i)
272 if k < 0: break
273 self.lineno = self.lineno + string.count(rawdata[i:k], '\n')
274 i = k
275 continue
276 if cdataopen.match(rawdata, i):
277 k = self.parse_cdata(i)
278 if k < 0: break
279 self.lineno = self.lineno + string.count(rawdata[i:i], '\n')
280 i = k
281 continue
Guido van Rossum7e07b381998-04-03 16:02:39 +0000282 res = xmldecl.match(rawdata, i)
283 if res:
284 if not self.__at_start:
285 self.syntax_error("<?xml?> declaration not at start of document")
286 version, encoding, standalone = res.group('version',
287 'encoding',
288 'standalone')
289 if version[1:-1] != '1.0':
290 raise RuntimeError, 'only XML version 1.0 supported'
291 if encoding: encoding = encoding[1:-1]
292 if standalone: standalone = standalone[1:-1]
293 self.handle_xml(encoding, standalone)
294 i = res.end(0)
295 continue
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000296 res = procopen.match(rawdata, i)
297 if res:
298 k = self.parse_proc(i)
299 if k < 0: break
300 self.lineno = self.lineno + string.count(rawdata[i:k], '\n')
301 i = k
302 continue
303 res = doctype.match(rawdata, i)
304 if res:
305 if self.literal:
306 data = rawdata[i]
307 self.handle_data(data)
308 self.lineno = self.lineno + string.count(data, '\n')
309 i = i+1
310 continue
311 if self.__seen_doctype:
312 self.syntax_error('multiple DOCTYPE elements')
313 if self.__seen_starttag:
314 self.syntax_error('DOCTYPE not at beginning of document')
315 k = self.parse_doctype(res)
316 if k < 0: break
317 self.__seen_doctype = res.group('name')
Guido van Rossumb35d6841999-08-26 15:52:33 +0000318 if self.__map_case:
319 self.__seen_doctype = string.lower(self.__seen_doctype)
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000320 self.lineno = self.lineno + string.count(rawdata[i:k], '\n')
321 i = k
322 continue
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000323 elif rawdata[i] == '&':
Guido van Rossum204b65c1998-12-08 13:25:00 +0000324 if self.literal:
325 data = rawdata[i]
326 self.handle_data(data)
327 i = i+1
328 continue
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000329 res = charref.match(rawdata, i)
330 if res is not None:
331 i = res.end(0)
332 if rawdata[i-1] != ';':
333 self.syntax_error("`;' missing in charref")
334 i = i-1
Guido van Rossum7e07b381998-04-03 16:02:39 +0000335 if not self.stack:
336 self.syntax_error('data not in content')
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000337 self.handle_charref(res.group('char')[:-1])
338 self.lineno = self.lineno + string.count(res.group(0), '\n')
339 continue
340 res = entityref.match(rawdata, i)
341 if res is not None:
342 i = res.end(0)
343 if rawdata[i-1] != ';':
344 self.syntax_error("`;' missing in entityref")
345 i = i-1
Guido van Rossum7e07b381998-04-03 16:02:39 +0000346 name = res.group('name')
Guido van Rossumb35d6841999-08-26 15:52:33 +0000347 if self.__map_case:
348 name = string.lower(name)
Guido van Rossum7e07b381998-04-03 16:02:39 +0000349 if self.entitydefs.has_key(name):
350 self.rawdata = rawdata = rawdata[:res.start(0)] + self.entitydefs[name] + rawdata[i:]
351 n = len(rawdata)
352 i = res.start(0)
353 else:
Guido van Rossum7e07b381998-04-03 16:02:39 +0000354 self.unknown_entityref(name)
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000355 self.lineno = self.lineno + string.count(res.group(0), '\n')
356 continue
Guido van Rossum7e07b381998-04-03 16:02:39 +0000357 elif rawdata[i] == ']':
Guido van Rossum204b65c1998-12-08 13:25:00 +0000358 if self.literal:
359 data = rawdata[i]
360 self.handle_data(data)
361 i = i+1
362 continue
Guido van Rossum7e07b381998-04-03 16:02:39 +0000363 if n-i < 3:
364 break
365 if cdataclose.match(rawdata, i):
366 self.syntax_error("bogus `]]>'")
367 self.handle_data(rawdata[i])
368 i = i+1
369 continue
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000370 else:
371 raise RuntimeError, 'neither < nor & ??'
372 # We get here only if incomplete matches but
373 # nothing else
Guido van Rossum7e07b381998-04-03 16:02:39 +0000374 break
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000375 # end while
Guido van Rossum7e07b381998-04-03 16:02:39 +0000376 if i > 0:
377 self.__at_start = 0
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000378 if end and i < n:
Guido van Rossum7e07b381998-04-03 16:02:39 +0000379 data = rawdata[i]
380 self.syntax_error("bogus `%s'" % data)
Guido van Rossumb35d6841999-08-26 15:52:33 +0000381 if not self.__accept_utf8 and illegal.search(data):
Guido van Rossum7e07b381998-04-03 16:02:39 +0000382 self.syntax_error('illegal character in content')
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000383 self.handle_data(data)
384 self.lineno = self.lineno + string.count(data, '\n')
Guido van Rossum7e07b381998-04-03 16:02:39 +0000385 self.rawdata = rawdata[i+1:]
386 return self.goahead(end)
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000387 self.rawdata = rawdata[i:]
388 if end:
Guido van Rossum7e07b381998-04-03 16:02:39 +0000389 if not self.__seen_starttag:
390 self.syntax_error('no elements in file')
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000391 if self.stack:
392 self.syntax_error('missing end tags')
393 while self.stack:
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000394 self.finish_endtag(self.stack[-1][0])
Guido van Rossuma219efa1997-11-18 15:09:54 +0000395
396 # Internal -- parse comment, return length or -1 if not terminated
397 def parse_comment(self, i):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000398 rawdata = self.rawdata
399 if rawdata[i:i+4] <> '<!--':
400 raise RuntimeError, 'unexpected call to handle_comment'
401 res = commentclose.search(rawdata, i+4)
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000402 if res is None:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000403 return -1
Guido van Rossum7e07b381998-04-03 16:02:39 +0000404 if doubledash.search(rawdata, i+4, res.start(0)):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000405 self.syntax_error("`--' inside comment")
Guido van Rossum7e07b381998-04-03 16:02:39 +0000406 if rawdata[res.start(0)-1] == '-':
407 self.syntax_error('comment cannot end in three dashes')
Guido van Rossumb35d6841999-08-26 15:52:33 +0000408 if not self.__accept_utf8 and \
409 illegal.search(rawdata, i+4, res.start(0)):
Guido van Rossum7e07b381998-04-03 16:02:39 +0000410 self.syntax_error('illegal character in comment')
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000411 self.handle_comment(rawdata[i+4: res.start(0)])
412 return res.end(0)
Guido van Rossuma219efa1997-11-18 15:09:54 +0000413
Guido van Rossum02505e41998-01-29 14:55:24 +0000414 # Internal -- handle DOCTYPE tag, return length or -1 if not terminated
415 def parse_doctype(self, res):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000416 rawdata = self.rawdata
417 n = len(rawdata)
418 name = res.group('name')
Guido van Rossumb35d6841999-08-26 15:52:33 +0000419 if self.__map_case:
420 name = string.lower(name)
Guido van Rossum7e07b381998-04-03 16:02:39 +0000421 pubid, syslit = res.group('pubid', 'syslit')
422 if pubid is not None:
423 pubid = pubid[1:-1] # remove quotes
424 pubid = string.join(string.split(pubid)) # normalize
425 if syslit is not None: syslit = syslit[1:-1] # remove quotes
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000426 j = k = res.end(0)
Guido van Rossum7e07b381998-04-03 16:02:39 +0000427 if k >= n:
428 return -1
429 if rawdata[k] == '[':
430 level = 0
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000431 k = k+1
Guido van Rossum7e07b381998-04-03 16:02:39 +0000432 dq = sq = 0
433 while k < n:
434 c = rawdata[k]
435 if not sq and c == '"':
436 dq = not dq
437 elif not dq and c == "'":
438 sq = not sq
439 elif sq or dq:
440 pass
441 elif level <= 0 and c == ']':
442 res = endbracket.match(rawdata, k+1)
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000443 if res is None:
Guido van Rossum7e07b381998-04-03 16:02:39 +0000444 return -1
445 self.handle_doctype(name, pubid, syslit, rawdata[j+1:k])
446 return res.end(0)
447 elif c == '<':
448 level = level + 1
449 elif c == '>':
450 level = level - 1
451 if level < 0:
452 self.syntax_error("bogus `>' in DOCTYPE")
453 k = k+1
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000454 res = endbracketfind.match(rawdata, k)
455 if res is None:
Guido van Rossum7e07b381998-04-03 16:02:39 +0000456 return -1
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000457 if endbracket.match(rawdata, k) is None:
Guido van Rossum7e07b381998-04-03 16:02:39 +0000458 self.syntax_error('garbage in DOCTYPE')
459 self.handle_doctype(name, pubid, syslit, None)
460 return res.end(0)
Guido van Rossum02505e41998-01-29 14:55:24 +0000461
462 # Internal -- handle CDATA tag, return length or -1 if not terminated
Guido van Rossuma219efa1997-11-18 15:09:54 +0000463 def parse_cdata(self, i):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000464 rawdata = self.rawdata
465 if rawdata[i:i+9] <> '<![CDATA[':
Guido van Rossum7e07b381998-04-03 16:02:39 +0000466 raise RuntimeError, 'unexpected call to parse_cdata'
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000467 res = cdataclose.search(rawdata, i+9)
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000468 if res is None:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000469 return -1
Guido van Rossumb35d6841999-08-26 15:52:33 +0000470 if not self.__accept_utf8 and \
471 illegal.search(rawdata, i+9, res.start(0)):
Guido van Rossum7e07b381998-04-03 16:02:39 +0000472 self.syntax_error('illegal character in CDATA')
473 if not self.stack:
474 self.syntax_error('CDATA not in content')
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000475 self.handle_cdata(rawdata[i+9:res.start(0)])
476 return res.end(0)
Guido van Rossuma219efa1997-11-18 15:09:54 +0000477
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000478 __xml_namespace_attributes = {'ns':None, 'src':None, 'prefix':None}
Guido van Rossum02505e41998-01-29 14:55:24 +0000479 # Internal -- handle a processing instruction tag
480 def parse_proc(self, i):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000481 rawdata = self.rawdata
482 end = procclose.search(rawdata, i)
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000483 if end is None:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000484 return -1
485 j = end.start(0)
Guido van Rossumb35d6841999-08-26 15:52:33 +0000486 if not self.__accept_utf8 and illegal.search(rawdata, i+2, j):
Guido van Rossum7e07b381998-04-03 16:02:39 +0000487 self.syntax_error('illegal character in processing instruction')
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000488 res = tagfind.match(rawdata, i+2)
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000489 if res is None:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000490 raise RuntimeError, 'unexpected call to parse_proc'
491 k = res.end(0)
492 name = res.group(0)
Guido van Rossumb35d6841999-08-26 15:52:33 +0000493 if self.__map_case:
494 name = string.lower(name)
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000495 if name == 'xml:namespace':
496 self.syntax_error('old-fashioned namespace declaration')
Guido van Rossum91c8f591998-12-21 18:15:28 +0000497 self.__use_namespaces = -1
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000498 # namespace declaration
499 # this must come after the <?xml?> declaration (if any)
500 # and before the <!DOCTYPE> (if any).
501 if self.__seen_doctype or self.__seen_starttag:
502 self.syntax_error('xml:namespace declaration too late in document')
503 attrdict, namespace, k = self.parse_attributes(name, k, j)
504 if namespace:
505 self.syntax_error('namespace declaration inside namespace declaration')
506 for attrname in attrdict.keys():
507 if not self.__xml_namespace_attributes.has_key(attrname):
508 self.syntax_error("unknown attribute `%s' in xml:namespace tag" % attrname)
509 if not attrdict.has_key('ns') or not attrdict.has_key('prefix'):
510 self.syntax_error('xml:namespace without required attributes')
511 prefix = attrdict.get('prefix')
512 if ncname.match(prefix) is None:
513 self.syntax_error('xml:namespace illegal prefix value')
514 return end.end(0)
515 if self.__namespaces.has_key(prefix):
516 self.syntax_error('xml:namespace prefix not unique')
517 self.__namespaces[prefix] = attrdict['ns']
518 else:
Guido van Rossum1f2e09b1999-06-08 21:23:26 +0000519 if string.lower(name) == 'xml':
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000520 self.syntax_error('illegal processing instruction target name')
521 self.handle_proc(name, rawdata[k:j])
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000522 return end.end(0)
Guido van Rossum02505e41998-01-29 14:55:24 +0000523
524 # Internal -- parse attributes between i and j
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000525 def parse_attributes(self, tag, i, j):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000526 rawdata = self.rawdata
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000527 attrdict = {}
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000528 namespace = {}
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000529 while i < j:
530 res = attrfind.match(rawdata, i)
531 if res is None:
532 break
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000533 attrname, attrvalue = res.group('name', 'value')
Guido van Rossumb35d6841999-08-26 15:52:33 +0000534 if self.__map_case:
535 attrname = string.lower(attrname)
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000536 i = res.end(0)
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000537 if attrvalue is None:
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000538 self.syntax_error("no value specified for attribute `%s'" % attrname)
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000539 attrvalue = attrname
540 elif attrvalue[:1] == "'" == attrvalue[-1:] or \
541 attrvalue[:1] == '"' == attrvalue[-1:]:
542 attrvalue = attrvalue[1:-1]
Guido van Rossumb35d6841999-08-26 15:52:33 +0000543 elif not self.__accept_unquoted_attributes:
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000544 self.syntax_error("attribute `%s' value not quoted" % attrname)
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000545 res = xmlns.match(attrname)
546 if res is not None:
547 # namespace declaration
548 ncname = res.group('ncname')
549 namespace[ncname or ''] = attrvalue or None
Guido van Rossum91c8f591998-12-21 18:15:28 +0000550 if not self.__use_namespaces:
551 self.__use_namespaces = len(self.stack)+1
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000552 continue
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000553 if '<' in attrvalue:
554 self.syntax_error("`<' illegal in attribute value")
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000555 if attrdict.has_key(attrname):
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000556 self.syntax_error("attribute `%s' specified twice" % attrname)
Guido van Rossum7e07b381998-04-03 16:02:39 +0000557 attrvalue = string.translate(attrvalue, attrtrans)
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000558 attrdict[attrname] = self.translate_references(attrvalue)
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000559 return attrdict, namespace, i
Guido van Rossuma219efa1997-11-18 15:09:54 +0000560
561 # Internal -- handle starttag, return length or -1 if not terminated
562 def parse_starttag(self, i):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000563 rawdata = self.rawdata
564 # i points to start of tag
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000565 end = endbracketfind.match(rawdata, i+1)
566 if end is None:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000567 return -1
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000568 tag = starttagmatch.match(rawdata, i)
569 if tag is None or tag.end(0) != end.end(0):
570 self.syntax_error('garbage in starttag')
571 return end.end(0)
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000572 nstag = tagname = tag.group('tagname')
Guido van Rossumb35d6841999-08-26 15:52:33 +0000573 if self.__map_case:
574 nstag = tagname = string.lower(nstag)
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000575 if not self.__seen_starttag and self.__seen_doctype and \
576 tagname != self.__seen_doctype:
577 self.syntax_error('starttag does not match DOCTYPE')
Guido van Rossum7e07b381998-04-03 16:02:39 +0000578 if self.__seen_starttag and not self.stack:
579 self.syntax_error('multiple elements on top level')
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000580 k, j = tag.span('attrs')
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000581 attrdict, nsdict, k = self.parse_attributes(tagname, k, j)
582 self.stack.append((tagname, nsdict, nstag))
Guido van Rossum91c8f591998-12-21 18:15:28 +0000583 if self.__use_namespaces:
584 res = qname.match(tagname)
585 else:
586 res = None
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000587 if res is not None:
588 prefix, nstag = res.group('prefix', 'local')
589 if prefix is None:
590 prefix = ''
591 ns = None
592 for t, d, nst in self.stack:
593 if d.has_key(prefix):
594 ns = d[prefix]
595 if ns is None and prefix != '':
596 ns = self.__namespaces.get(prefix)
597 if ns is not None:
598 nstag = ns + ' ' + nstag
599 elif prefix != '':
600 nstag = prefix + ':' + nstag # undo split
601 self.stack[-1] = tagname, nsdict, nstag
602 # translate namespace of attributes
Guido van Rossum91c8f591998-12-21 18:15:28 +0000603 if self.__use_namespaces:
604 nattrdict = {}
605 for key, val in attrdict.items():
606 res = qname.match(key)
607 if res is not None:
608 aprefix, key = res.group('prefix', 'local')
Guido van Rossumb35d6841999-08-26 15:52:33 +0000609 if self.__map_case:
610 key = string.lower(key)
Guido van Rossum91c8f591998-12-21 18:15:28 +0000611 if aprefix is None:
612 aprefix = ''
613 ans = None
614 for t, d, nst in self.stack:
615 if d.has_key(aprefix):
616 ans = d[aprefix]
617 if ans is None and aprefix != '':
618 ans = self.__namespaces.get(aprefix)
619 if ans is not None:
620 key = ans + ' ' + key
621 elif aprefix != '':
622 key = aprefix + ':' + key
623 elif ns is not None:
624 key = ns + ' ' + key
625 nattrdict[key] = val
626 attrdict = nattrdict
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000627 attributes = self.attributes.get(nstag)
628 if attributes is not None:
629 for key in attrdict.keys():
630 if not attributes.has_key(key):
631 self.syntax_error("unknown attribute `%s' in tag `%s'" % (key, tagname))
632 for key, val in attributes.items():
633 if val is not None and not attrdict.has_key(key):
634 attrdict[key] = val
635 method = self.elements.get(nstag, (None, None))[0]
636 self.finish_starttag(nstag, attrdict, method)
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000637 if tag.group('slash') == '/':
638 self.finish_endtag(tagname)
639 return tag.end(0)
Guido van Rossuma219efa1997-11-18 15:09:54 +0000640
641 # Internal -- parse endtag
642 def parse_endtag(self, i):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000643 rawdata = self.rawdata
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000644 end = endbracketfind.match(rawdata, i+1)
645 if end is None:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000646 return -1
647 res = tagfind.match(rawdata, i+2)
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000648 if res is None:
Guido van Rossumf484a331998-12-07 21:59:56 +0000649 if self.literal:
650 self.handle_data(rawdata[i])
651 return i+1
Guido van Rossumb35d6841999-08-26 15:52:33 +0000652 if not self.__accept_missing_endtag_name:
653 self.syntax_error('no name specified in end tag')
654 tag = self.stack[-1][0]
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000655 k = i+2
656 else:
657 tag = res.group(0)
Guido van Rossumb35d6841999-08-26 15:52:33 +0000658 if self.__map_case:
659 tag = string.lower(tag)
Guido van Rossumf484a331998-12-07 21:59:56 +0000660 if self.literal:
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000661 if not self.stack or tag != self.stack[-1][0]:
Guido van Rossumf484a331998-12-07 21:59:56 +0000662 self.handle_data(rawdata[i])
663 return i+1
664 self.literal = 0
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000665 k = res.end(0)
Guido van Rossumeeb2f321998-10-19 13:28:26 +0000666 if endbracket.match(rawdata, k) is None:
Guido van Rossum7e07b381998-04-03 16:02:39 +0000667 self.syntax_error('garbage in end tag')
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000668 self.finish_endtag(tag)
669 return end.end(0)
Guido van Rossuma219efa1997-11-18 15:09:54 +0000670
671 # Internal -- finish processing of start tag
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000672 def finish_starttag(self, tagname, attrdict, method):
673 if method is not None:
674 self.handle_starttag(tagname, method, attrdict)
Guido van Rossum7e07b381998-04-03 16:02:39 +0000675 else:
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000676 self.unknown_starttag(tagname, attrdict)
Guido van Rossuma219efa1997-11-18 15:09:54 +0000677
678 # Internal -- finish processing of end tag
679 def finish_endtag(self, tag):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000680 if not tag:
681 self.syntax_error('name-less end tag')
682 found = len(self.stack) - 1
683 if found < 0:
684 self.unknown_endtag(tag)
685 return
686 else:
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000687 found = -1
688 for i in range(len(self.stack)):
689 if tag == self.stack[i][0]:
690 found = i
691 if found == -1:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000692 self.syntax_error('unopened end tag')
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000693 method = self.elements.get(tag, (None, None))[1]
694 if method is not None:
Guido van Rossum7e07b381998-04-03 16:02:39 +0000695 self.handle_endtag(tag, method)
696 else:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000697 self.unknown_endtag(tag)
698 return
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000699 while len(self.stack) > found:
700 if found < len(self.stack) - 1:
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000701 self.syntax_error('missing close tag for %s' % self.stack[-1][2])
702 nstag = self.stack[-1][2]
703 method = self.elements.get(nstag, (None, None))[1]
704 if method is not None:
705 self.handle_endtag(nstag, method)
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000706 else:
Guido van Rossumb083a9f1998-12-18 20:17:13 +0000707 self.unknown_endtag(nstag)
Guido van Rossum91c8f591998-12-21 18:15:28 +0000708 if self.__use_namespaces == len(self.stack):
709 self.__use_namespaces = 0
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000710 del self.stack[-1]
Guido van Rossuma219efa1997-11-18 15:09:54 +0000711
Guido van Rossum02505e41998-01-29 14:55:24 +0000712 # Overridable -- handle xml processing instruction
713 def handle_xml(self, encoding, standalone):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000714 pass
Guido van Rossum02505e41998-01-29 14:55:24 +0000715
716 # Overridable -- handle DOCTYPE
Guido van Rossum7e07b381998-04-03 16:02:39 +0000717 def handle_doctype(self, tag, pubid, syslit, data):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000718 pass
Guido van Rossum02505e41998-01-29 14:55:24 +0000719
Guido van Rossuma219efa1997-11-18 15:09:54 +0000720 # Overridable -- handle start tag
721 def handle_starttag(self, tag, method, attrs):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000722 method(attrs)
Guido van Rossuma219efa1997-11-18 15:09:54 +0000723
724 # Overridable -- handle end tag
725 def handle_endtag(self, tag, method):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000726 method()
Guido van Rossuma219efa1997-11-18 15:09:54 +0000727
728 # Example -- handle character reference, no need to override
729 def handle_charref(self, name):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000730 try:
731 if name[0] == 'x':
732 n = string.atoi(name[1:], 16)
733 else:
734 n = string.atoi(name)
735 except string.atoi_error:
736 self.unknown_charref(name)
737 return
738 if not 0 <= n <= 255:
739 self.unknown_charref(name)
740 return
741 self.handle_data(chr(n))
Guido van Rossuma219efa1997-11-18 15:09:54 +0000742
743 # Definition of entities -- derived classes may override
Guido van Rossum7e07b381998-04-03 16:02:39 +0000744 entitydefs = {'lt': '&#60;', # must use charref
745 'gt': '&#62;',
746 'amp': '&#38;', # must use charref
747 'quot': '&#34;',
748 'apos': '&#39;',
749 }
Guido van Rossuma219efa1997-11-18 15:09:54 +0000750
Guido van Rossuma219efa1997-11-18 15:09:54 +0000751 # Example -- handle data, should be overridden
752 def handle_data(self, data):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000753 pass
Guido van Rossuma219efa1997-11-18 15:09:54 +0000754
755 # Example -- handle cdata, could be overridden
756 def handle_cdata(self, data):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000757 pass
Guido van Rossuma219efa1997-11-18 15:09:54 +0000758
759 # Example -- handle comment, could be overridden
760 def handle_comment(self, data):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000761 pass
Guido van Rossuma219efa1997-11-18 15:09:54 +0000762
763 # Example -- handle processing instructions, could be overridden
764 def handle_proc(self, name, data):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000765 pass
Guido van Rossuma219efa1997-11-18 15:09:54 +0000766
Guido van Rossuma219efa1997-11-18 15:09:54 +0000767 # Example -- handle relatively harmless syntax errors, could be overridden
Guido van Rossum02505e41998-01-29 14:55:24 +0000768 def syntax_error(self, message):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000769 raise RuntimeError, 'Syntax error at line %d: %s' % (self.lineno, message)
Guido van Rossuma219efa1997-11-18 15:09:54 +0000770
771 # To be overridden -- handlers for unknown objects
772 def unknown_starttag(self, tag, attrs): pass
773 def unknown_endtag(self, tag): pass
774 def unknown_charref(self, ref): pass
Guido van Rossumb35d6841999-08-26 15:52:33 +0000775 def unknown_entityref(self, name):
776 self.syntax_error("reference to unknown entity `&%s;'" % name)
Guido van Rossuma219efa1997-11-18 15:09:54 +0000777
778
779class TestXMLParser(XMLParser):
780
Guido van Rossumb35d6841999-08-26 15:52:33 +0000781 def __init__(self, **kw):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000782 self.testdata = ""
Guido van Rossumb35d6841999-08-26 15:52:33 +0000783 apply(XMLParser.__init__, (self,), kw)
Guido van Rossuma219efa1997-11-18 15:09:54 +0000784
Guido van Rossum02505e41998-01-29 14:55:24 +0000785 def handle_xml(self, encoding, standalone):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000786 self.flush()
787 print 'xml: encoding =',encoding,'standalone =',standalone
Guido van Rossum02505e41998-01-29 14:55:24 +0000788
Guido van Rossum7e07b381998-04-03 16:02:39 +0000789 def handle_doctype(self, tag, pubid, syslit, data):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000790 self.flush()
791 print 'DOCTYPE:',tag, `data`
Guido van Rossum02505e41998-01-29 14:55:24 +0000792
Guido van Rossuma219efa1997-11-18 15:09:54 +0000793 def handle_data(self, data):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000794 self.testdata = self.testdata + data
795 if len(`self.testdata`) >= 70:
796 self.flush()
Guido van Rossuma219efa1997-11-18 15:09:54 +0000797
798 def flush(self):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000799 data = self.testdata
800 if data:
801 self.testdata = ""
802 print 'data:', `data`
Guido van Rossuma219efa1997-11-18 15:09:54 +0000803
804 def handle_cdata(self, data):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000805 self.flush()
806 print 'cdata:', `data`
Guido van Rossuma219efa1997-11-18 15:09:54 +0000807
808 def handle_proc(self, name, data):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000809 self.flush()
810 print 'processing:',name,`data`
Guido van Rossuma219efa1997-11-18 15:09:54 +0000811
Guido van Rossuma219efa1997-11-18 15:09:54 +0000812 def handle_comment(self, data):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000813 self.flush()
814 r = `data`
815 if len(r) > 68:
816 r = r[:32] + '...' + r[-32:]
817 print 'comment:', r
Guido van Rossuma219efa1997-11-18 15:09:54 +0000818
Guido van Rossum02505e41998-01-29 14:55:24 +0000819 def syntax_error(self, message):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000820 print 'error at line %d:' % self.lineno, message
Guido van Rossuma219efa1997-11-18 15:09:54 +0000821
822 def unknown_starttag(self, tag, attrs):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000823 self.flush()
824 if not attrs:
825 print 'start tag: <' + tag + '>'
826 else:
827 print 'start tag: <' + tag,
828 for name, value in attrs.items():
829 print name + '=' + '"' + value + '"',
830 print '>'
Guido van Rossuma219efa1997-11-18 15:09:54 +0000831
832 def unknown_endtag(self, tag):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000833 self.flush()
834 print 'end tag: </' + tag + '>'
Guido van Rossuma219efa1997-11-18 15:09:54 +0000835
836 def unknown_entityref(self, ref):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000837 self.flush()
838 print '*** unknown entity ref: &' + ref + ';'
Guido van Rossuma219efa1997-11-18 15:09:54 +0000839
840 def unknown_charref(self, ref):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000841 self.flush()
842 print '*** unknown char ref: &#' + ref + ';'
Guido van Rossuma219efa1997-11-18 15:09:54 +0000843
844 def close(self):
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000845 XMLParser.close(self)
846 self.flush()
Guido van Rossuma219efa1997-11-18 15:09:54 +0000847
848def test(args = None):
Guido van Rossumaf5add41998-12-22 14:23:50 +0000849 import sys, getopt
850 from time import time
Guido van Rossuma219efa1997-11-18 15:09:54 +0000851
852 if not args:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000853 args = sys.argv[1:]
Guido van Rossuma219efa1997-11-18 15:09:54 +0000854
Guido van Rossumaf5add41998-12-22 14:23:50 +0000855 opts, args = getopt.getopt(args, 'st')
856 klass = TestXMLParser
857 do_time = 0
858 for o, a in opts:
859 if o == '-s':
860 klass = XMLParser
861 elif o == '-t':
862 do_time = 1
Guido van Rossuma219efa1997-11-18 15:09:54 +0000863
864 if args:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000865 file = args[0]
Guido van Rossuma219efa1997-11-18 15:09:54 +0000866 else:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000867 file = 'test.xml'
Guido van Rossuma219efa1997-11-18 15:09:54 +0000868
869 if file == '-':
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000870 f = sys.stdin
Guido van Rossuma219efa1997-11-18 15:09:54 +0000871 else:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000872 try:
873 f = open(file, 'r')
874 except IOError, msg:
875 print file, ":", msg
876 sys.exit(1)
Guido van Rossuma219efa1997-11-18 15:09:54 +0000877
878 data = f.read()
879 if f is not sys.stdin:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +0000880 f.close()
Guido van Rossuma219efa1997-11-18 15:09:54 +0000881
882 x = klass()
Guido van Rossumaf5add41998-12-22 14:23:50 +0000883 t0 = time()
Guido van Rossum7e07b381998-04-03 16:02:39 +0000884 try:
Guido van Rossumaf5add41998-12-22 14:23:50 +0000885 if do_time:
886 x.feed(data)
887 x.close()
888 else:
889 for c in data:
890 x.feed(c)
891 x.close()
Guido van Rossum7e07b381998-04-03 16:02:39 +0000892 except RuntimeError, msg:
Guido van Rossumaf5add41998-12-22 14:23:50 +0000893 t1 = time()
Guido van Rossum7e07b381998-04-03 16:02:39 +0000894 print msg
Guido van Rossumaf5add41998-12-22 14:23:50 +0000895 if do_time:
896 print 'total time: %g' % (t1-t0)
Guido van Rossum7e07b381998-04-03 16:02:39 +0000897 sys.exit(1)
Guido van Rossumaf5add41998-12-22 14:23:50 +0000898 t1 = time()
899 if do_time:
900 print 'total time: %g' % (t1-t0)
Guido van Rossuma219efa1997-11-18 15:09:54 +0000901
902
903if __name__ == '__main__':
904 test()