13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
17
"""XML externalization support."""
22
22
# importing this module is fairly slow because it has to load several
25
from bzrlib import registry
25
from bzrlib.serializer import Serializer
26
26
from bzrlib.trace import mutter, warning
49
49
from bzrlib import errors
52
class Serializer(object):
53
"""Abstract object serialize/deserialize"""
55
def write_inventory(self, inv, f):
56
"""Write inventory to a file"""
57
raise NotImplementedError(self.write_inventory)
59
def write_inventory_to_string(self, inv):
60
raise NotImplementedError(self.write_inventory_to_string)
62
def read_inventory_from_string(self, xml_string, revision_id=None):
52
class XMLSerializer(Serializer):
53
"""Abstract XML object serialize/deserialize"""
55
squashes_xml_invalid_characters = True
57
def read_inventory_from_string(self, xml_string, revision_id=None,
63
59
"""Read xml_string into an inventory object.
65
61
:param xml_string: The xml to read.
70
66
serialised without a revision identifier can be given the right
71
67
revision id (but not for working tree inventories where users can
72
68
edit the data without triggering checksum errors or anything).
69
:param entry_cache: An optional cache of InventoryEntry objects. If
70
supplied we will look up entries via (file_id, revision_id) which
71
should map to a valid InventoryEntry (File/Directory/etc) object.
75
return self._unpack_inventory(fromstring(xml_string), revision_id)
74
return self._unpack_inventory(fromstring(xml_string), revision_id,
75
entry_cache=entry_cache)
76
76
except ParseError, e:
77
77
raise errors.UnexpectedInventoryFormat(e)
118
118
def _escape_replace(match, map=escape_map):
119
119
return map[match.group()]
121
121
def _escape_attrib(text, encoding=None, replace=None):
122
122
# escape attribute value
149
149
def _escape_cdata_replace(match, map=escape_cdata_map):
150
150
return map[match.group()]
152
152
def _escape_cdata(text, encoding=None, replace=None):
153
153
# escape character data
170
170
elementtree.ElementTree._escape_cdata = _escape_cdata
173
class SerializerRegistry(registry.Registry):
174
"""Registry for serializer objects"""
177
format_registry = SerializerRegistry()
178
format_registry.register_lazy('4', 'bzrlib.xml4', 'serializer_v4')
179
format_registry.register_lazy('5', 'bzrlib.xml5', 'serializer_v5')
180
format_registry.register_lazy('6', 'bzrlib.xml6', 'serializer_v6')
181
format_registry.register_lazy('7', 'bzrlib.xml7', 'serializer_v7')
182
format_registry.register_lazy('8', 'bzrlib.xml8', 'serializer_v8')
173
def escape_invalid_chars(message):
174
"""Escape the XML-invalid characters in a commit message.
176
:param message: Commit message to escape
177
:return: tuple with escaped message and number of characters escaped
181
# Python strings can include characters that can't be
182
# represented in well-formed XML; escape characters that
183
# aren't listed in the XML specification
184
# (http://www.w3.org/TR/REC-xml/#NT-Char).
185
return re.subn(u'[^\x09\x0A\x0D\u0020-\uD7FF\uE000-\uFFFD]+',
186
lambda match: match.group(0).encode('unicode_escape'),