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
31
39
except ImportError:
32
40
mutter('WARNING: using slower ElementTree; consider installing cElementTree'
33
41
" and make sure it's on your PYTHONPATH")
42
# this copy is shipped with bzr
34
43
from util.elementtree.ElementTree import (ElementTree, SubElement,
35
44
Element, XMLTreeBuilder,
36
45
fromstring, tostring)
37
46
import util.elementtree as elementtree
47
from xml.parsers.expat import ExpatError as ParseError
39
49
from bzrlib import errors
42
52
class Serializer(object):
43
53
"""Abstract object serialize/deserialize"""
44
55
def write_inventory(self, inv, f):
45
56
"""Write inventory to a file"""
46
elt = self._pack_inventory(inv)
47
self._write_element(elt, f)
57
raise NotImplementedError(self.write_inventory)
49
59
def write_inventory_to_string(self, inv):
50
return tostring(self._pack_inventory(inv)) + '\n'
52
def read_inventory_from_string(self, xml_string):
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
54
return self._unpack_inventory(fromstring(xml_string))
55
except SyntaxError, e:
71
return self._unpack_inventory(fromstring(xml_string), revision_id)
56
73
raise errors.UnexpectedInventoryFormat(e)
58
def read_inventory(self, f):
75
def read_inventory(self, f, revision_id=None):
60
return self._unpack_inventory(self._read_element(f))
61
except SyntaxError, e:
77
return self._unpack_inventory(self._read_element(f),
62
80
raise errors.UnexpectedInventoryFormat(e)
64
82
def write_revision(self, rev, f):
146
164
elementtree.ElementTree._raise_serialization_error(text)
148
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')