You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
8 months ago
|
# Copyright (c) 2011 Google Inc. All rights reserved.
|
||
|
|
# Use of this source code is governed by a BSD-style license that can be
|
||
|
|
# found in the LICENSE file.
|
||
|
|
|
||
|
|
"""Applies a fix to CR LF TAB handling in xml.dom.
|
||
|
|
|
||
|
|
Fixes this: http://code.google.com/p/chromium/issues/detail?id=76293
|
||
|
|
Working around this: http://bugs.python.org/issue5752
|
||
|
|
TODO(bradnelson): Consider dropping this when we drop XP support.
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
import xml.dom.minidom
|
||
|
|
|
||
|
|
|
||
|
|
def _Replacement_write_data(writer, data, is_attrib=False):
|
||
|
|
"""Writes datachars to writer."""
|
||
|
|
data = data.replace("&", "&").replace("<", "<")
|
||
|
|
data = data.replace('"', """).replace(">", ">")
|
||
|
|
if is_attrib:
|
||
|
|
data = data.replace("\r", "
").replace("\n", "
").replace("\t", "	")
|
||
|
|
writer.write(data)
|
||
|
|
|
||
|
|
|
||
|
|
def _Replacement_writexml(self, writer, indent="", addindent="", newl=""):
|
||
|
|
# indent = current indentation
|
||
|
|
# addindent = indentation to add to higher levels
|
||
|
|
# newl = newline string
|
||
|
|
writer.write(indent + "<" + self.tagName)
|
||
|
|
|
||
|
|
attrs = self._get_attributes()
|
||
|
|
a_names = sorted(attrs.keys())
|
||
|
|
|
||
|
|
for a_name in a_names:
|
||
|
|
writer.write(' %s="' % a_name)
|
||
|
|
_Replacement_write_data(writer, attrs[a_name].value, is_attrib=True)
|
||
|
|
writer.write('"')
|
||
|
|
if self.childNodes:
|
||
|
|
writer.write(">%s" % newl)
|
||
|
|
for node in self.childNodes:
|
||
|
|
node.writexml(writer, indent + addindent, addindent, newl)
|
||
|
|
writer.write(f"{indent}</{self.tagName}>{newl}")
|
||
|
|
else:
|
||
|
|
writer.write("/>%s" % newl)
|
||
|
|
|
||
|
|
|
||
|
|
class XmlFix:
|
||
|
|
"""Object to manage temporary patching of xml.dom.minidom."""
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
# Preserve current xml.dom.minidom functions.
|
||
|
|
self.write_data = xml.dom.minidom._write_data
|
||
|
|
self.writexml = xml.dom.minidom.Element.writexml
|
||
|
|
# Inject replacement versions of a function and a method.
|
||
|
|
xml.dom.minidom._write_data = _Replacement_write_data
|
||
|
|
xml.dom.minidom.Element.writexml = _Replacement_writexml
|
||
|
|
|
||
|
|
def Cleanup(self):
|
||
|
|
if self.write_data:
|
||
|
|
xml.dom.minidom._write_data = self.write_data
|
||
|
|
xml.dom.minidom.Element.writexml = self.writexml
|
||
|
|
self.write_data = None
|
||
|
|
|
||
|
|
def __del__(self):
|
||
|
|
self.Cleanup()
|