File Explorer

1# Copyright (c) 2011 Google Inc. All rights reserved.2# Use of this source code is governed by a BSD-style license that can be3# found in the LICENSE file.4 5"""Applies a fix to CR LF TAB handling in xml.dom.6 7Fixes this: http://code.google.com/p/chromium/issues/detail?id=762938Working around this: http://bugs.python.org/issue57529TODO(bradnelson): Consider dropping this when we drop XP support.10"""11 12import xml.dom.minidom13 14 15def _Replacement_write_data(writer, data, is_attrib=False):16    """Writes datachars to writer."""17    data = data.replace("&", "&amp;").replace("<", "&lt;")18    data = data.replace('"', "&quot;").replace(">", "&gt;")19    if is_attrib:20        data = data.replace("\r", "&#xD;").replace("\n", "&#xA;").replace("\t", "&#x9;")21    writer.write(data)22 23 24def _Replacement_writexml(self, writer, indent="", addindent="", newl=""):25    # indent = current indentation26    # addindent = indentation to add to higher levels27    # newl = newline string28    writer.write(indent + "<" + self.tagName)29 30    attrs = self._get_attributes()31    a_names = sorted(attrs.keys())32 33    for a_name in a_names:34        writer.write(' %s="' % a_name)35        _Replacement_write_data(writer, attrs[a_name].value, is_attrib=True)36        writer.write('"')37    if self.childNodes:38        writer.write(">%s" % newl)39        for node in self.childNodes:40            node.writexml(writer, indent + addindent, addindent, newl)41        writer.write(f"{indent}</{self.tagName}>{newl}")42    else:43        writer.write("/>%s" % newl)44 45 46class XmlFix:47    """Object to manage temporary patching of xml.dom.minidom."""48 49    def __init__(self):50        # Preserve current xml.dom.minidom functions.51        self.write_data = xml.dom.minidom._write_data52        self.writexml = xml.dom.minidom.Element.writexml53        # Inject replacement versions of a function and a method.54        xml.dom.minidom._write_data = _Replacement_write_data55        xml.dom.minidom.Element.writexml = _Replacement_writexml56 57    def Cleanup(self):58        if self.write_data:59            xml.dom.minidom._write_data = self.write_data60            xml.dom.minidom.Element.writexml = self.writexml61            self.write_data = None62 63    def __del__(self):64        self.Cleanup()65