20
20
# "XML is like violence: if it doesn't solve your problem, you aren't
21
21
# using enough of it." -- various
23
# importing this module is fairly slow because it has to load several
26
from bzrlib.trace import mutter, warning
24
__copyright__ = "Copyright (C) 2005 Canonical Ltd."
25
__author__ = "Martin Pool <mbp@canonical.com>"
29
from cElementTree import (ElementTree, SubElement, Element,
30
XMLTreeBuilder, fromstring, tostring)
28
from cElementTree import Element, ElementTree, SubElement
31
29
except ImportError:
32
## from warnings import warn
33
## warn('using slower ElementTree; consider installing cElementTree')
34
from util.elementtree.ElementTree import (ElementTree, SubElement,
35
Element, XMLTreeBuilder,
38
from bzrlib.errors import BzrError
41
class Serializer(object):
42
"""Abstract object serialize/deserialize"""
43
def write_inventory(self, inv, f):
44
"""Write inventory to a file"""
45
elt = self._pack_inventory(inv)
46
self._write_element(elt, f)
48
def write_inventory_to_string(self, inv):
49
return tostring(self._pack_inventory(inv)) + '\n'
51
def read_inventory_from_string(self, xml_string):
52
return self._unpack_inventory(fromstring(xml_string))
54
def read_inventory(self, f):
55
return self._unpack_inventory(self._read_element(f))
57
def write_revision(self, rev, f):
58
self._write_element(self._pack_revision(rev), f)
60
def write_revision_to_string(self, rev):
61
return tostring(self._pack_revision(rev)) + '\n'
63
def read_revision(self, f):
64
return self._unpack_revision(self._read_element(f))
66
def read_revision_from_string(self, xml_string):
67
return self._unpack_revision(fromstring(xml_string))
69
def _write_element(self, elt, f):
70
ElementTree(elt).write(f, 'utf-8')
30
from elementtree.ElementTree import Element, ElementTree, SubElement
33
from trace import mutter
37
raise Exception("XMLMixin.to_element must be overridden in concrete classes")
39
def write_xml(self, f):
40
ElementTree(self.to_element()).write(f, 'utf-8')
73
def _read_element(self, f):
74
return ElementTree().parse(f)
44
return cls.from_element(ElementTree().parse(f))
46
read_xml = classmethod(read_xml)