blob: bc35e0609d80c23b26009d89d4409694490ee60f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
"""
Utilities for XML generation/parsing.
"""
from xml.sax.saxutils import XMLGenerator
from django.utils.regex_helper import _lazy_re_compile
_xml_control_chars_re = _lazy_re_compile(r"[\x00-\x08\x0B-\x0C\x0E-\x1F]")
class UnserializableContentError(ValueError):
def __init__(self, msg="Control characters are not supported in XML 1.0"):
super().__init__(msg)
class SimplerXMLGenerator(XMLGenerator):
def addQuickElement(self, name, contents=None, attrs=None):
"Convenience method for adding an element with no children"
if attrs is None:
attrs = {}
self.startElement(name, attrs)
if contents is not None:
self.characters(contents)
self.endElement(name)
def characters(self, content):
if content and _xml_control_chars_re.search(content):
# Fail loudly when content has control chars (unsupported in XML
# 1.0) See https://www.w3.org/International/questions/qa-controls
raise UnserializableContentError
XMLGenerator.characters(self, content)
def startElement(self, name, attrs):
for value in attrs.values():
if isinstance(value, str) and _xml_control_chars_re.search(value):
raise UnserializableContentError
# Sort attrs for a deterministic output.
sorted_attrs = dict(sorted(attrs.items())) if attrs else attrs
super().startElement(name, sorted_attrs)
|
