1
# -*- coding: UTF-8 -*-
1
# Copyright (C) 2005, 2006 Canonical Ltd
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
11
# GNU General Public License for more details.
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
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22
22
# importing this module is fairly slow because it has to load several
25
from bzrlib import registry
25
26
from bzrlib.trace import mutter, warning
28
from cElementTree import (ElementTree, SubElement, Element,
29
XMLTreeBuilder, fromstring, tostring)
30
# it's in this package in python2.5
31
from xml.etree.cElementTree import (ElementTree, SubElement, Element,
32
XMLTreeBuilder, fromstring, tostring)
33
import xml.etree as elementtree
35
from cElementTree import (ElementTree, SubElement, Element,
36
XMLTreeBuilder, fromstring, tostring)
38
ParseError = SyntaxError
30
39
except ImportError:
31
40
mutter('WARNING: using slower ElementTree; consider installing cElementTree'
32
41
" and make sure it's on your PYTHONPATH")
42
# this copy is shipped with bzr
33
43
from util.elementtree.ElementTree import (ElementTree, SubElement,
34
44
Element, XMLTreeBuilder,
35
45
fromstring, tostring)
46
import util.elementtree as elementtree
47
from xml.parsers.expat import ExpatError as ParseError
37
from bzrlib.errors import BzrError
49
from bzrlib import errors
40
52
class Serializer(object):
41
53
"""Abstract object serialize/deserialize"""
42
55
def write_inventory(self, inv, f):
43
56
"""Write inventory to a file"""
44
elt = self._pack_inventory(inv)
45
self._write_element(elt, f)
57
raise NotImplementedError(self.write_inventory)
47
59
def write_inventory_to_string(self, inv):
48
return tostring(self._pack_inventory(inv)) + '\n'
50
def read_inventory_from_string(self, xml_string):
51
return self._unpack_inventory(fromstring(xml_string))
53
def read_inventory(self, f):
54
return self._unpack_inventory(self._read_element(f))
60
raise NotImplementedError(self.write_inventory_to_string)
62
def read_inventory_from_string(self, xml_string, revision_id=None):
63
"""Read xml_string into an inventory object.
65
:param xml_string: The xml to read.
66
:param revision_id: If not-None, the expected revision id of the
67
inventory. Some serialisers use this to set the results' root
71
return self._unpack_inventory(fromstring(xml_string), revision_id)
73
raise errors.UnexpectedInventoryFormat(e)
75
def read_inventory(self, f, revision_id=None):
77
return self._unpack_inventory(self._read_element(f),
80
raise errors.UnexpectedInventoryFormat(e)
56
82
def write_revision(self, rev, f):
57
83
self._write_element(self._pack_revision(rev), f)
72
98
def _read_element(self, f):
73
99
return ElementTree().parse(f)
102
# performance tuning for elementree's serialiser. This should be
103
# sent upstream - RBC 20060523.
104
# the functions here are patched into elementtree at runtime.
106
escape_re = re.compile("[&'\"<>]")
109
"'":"'", # FIXME: overkill
114
def _escape_replace(match, map=escape_map):
115
return map[match.group()]
117
def _escape_attrib(text, encoding=None, replace=None):
118
# escape attribute value
122
text = elementtree.ElementTree._encode(text, encoding)
124
return elementtree.ElementTree._encode_entity(text)
126
return escape_re.sub(_escape_replace, text)
128
text = replace(text, "&", "&")
129
text = replace(text, "'", "'") # FIXME: overkill
130
text = replace(text, "\"", """)
131
text = replace(text, "<", "<")
132
text = replace(text, ">", ">")
134
except (TypeError, AttributeError):
135
elementtree.ElementTree._raise_serialization_error(text)
137
elementtree.ElementTree._escape_attrib = _escape_attrib
139
escape_cdata_re = re.compile("[&<>]")
145
def _escape_cdata_replace(match, map=escape_cdata_map):
146
return map[match.group()]
148
def _escape_cdata(text, encoding=None, replace=None):
149
# escape character data
153
text = elementtree.ElementTree._encode(text, encoding)
155
return elementtree.ElementTree._encode_entity(text)
157
return escape_cdata_re.sub(_escape_cdata_replace, text)
159
text = replace(text, "&", "&")
160
text = replace(text, "<", "<")
161
text = replace(text, ">", ">")
163
except (TypeError, AttributeError):
164
elementtree.ElementTree._raise_serialization_error(text)
166
elementtree.ElementTree._escape_cdata = _escape_cdata
169
class SerializerRegistry(registry.Registry):
170
"""Registry for serializer objects"""
173
format_registry = SerializerRegistry()
174
format_registry.register_lazy('4', 'bzrlib.xml4', 'serializer_v4')
175
format_registry.register_lazy('5', 'bzrlib.xml5', 'serializer_v5')
176
format_registry.register_lazy('6', 'bzrlib.xml6', 'serializer_v6')
177
format_registry.register_lazy('7', 'bzrlib.xml7', 'serializer_v7')