← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/plugins/weave_fmt/xml4.py

Major code cleanup.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/plugins/weave_fmt/xml4.py
 
1
# Copyright (C) 2005-2010 Canonical Ltd
 
2
#
1
3
# This program is free software; you can redistribute it and/or modify
2
4
# it under the terms of the GNU General Public License as published by
3
5
# the Free Software Foundation; either version 2 of the License, or
4
6
# (at your option) any later version.
5
 
 
 
7
#
6
8
# This program is distributed in the hope that it will be useful,
7
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
8
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
9
11
# GNU General Public License for more details.
10
 
 
 
12
#
11
13
# You should have received a copy of the GNU General Public License
12
14
# along with this program; if not, write to the Free Software
13
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
14
 
 
15
 
 
16
 
from bzrlib.xml_serializer import ElementTree, SubElement, Element, Serializer
17
 
from bzrlib.inventory import ROOT_ID, Inventory, InventoryEntry
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
from bzrlib.xml_serializer import (
 
18
    Element,
 
19
    SubElement,
 
20
    XMLSerializer,
 
21
    escape_invalid_chars,
 
22
    )
 
23
from bzrlib.inventory import ROOT_ID, Inventory
18
24
import bzrlib.inventory as inventory
19
 
from bzrlib.revision import Revision        
 
25
from bzrlib.revision import Revision
20
26
from bzrlib.errors import BzrError
21
27
 
22
28
 
23
 
 
24
 
 
25
 
 
26
 
 
27
 
class _Serializer_v4(Serializer):
 
29
class _Serializer_v4(XMLSerializer):
28
30
    """Version 0.0.4 serializer
29
31
 
30
 
    You should use the serialzer_v4 singleton."""
31
 
    
 
32
    You should use the serializer_v4 singleton.
 
33
 
 
34
    v4 serialisation is no longer supported, only deserialisation.
 
35
    """
 
36
 
32
37
    __slots__ = []
33
 
    
34
 
    def _pack_inventory(self, inv):
35
 
        """Convert to XML Element"""
36
 
        e = Element('inventory')
37
 
        e.text = '\n'
38
 
        if inv.root.file_id not in (None, ROOT_ID):
39
 
            e.set('file_id', inv.root.file_id)
40
 
        for path, ie in inv.iter_entries():
41
 
            e.append(self._pack_entry(ie))
42
 
        return e
43
 
 
44
38
 
45
39
    def _pack_entry(self, ie):
46
40
        """Convert InventoryEntry to XML element"""
49
43
        e.set('file_id', ie.file_id)
50
44
        e.set('kind', ie.kind)
51
45
 
52
 
        if ie.text_size != None:
 
46
        if ie.text_size is not None:
53
47
            e.set('text_size', '%d' % ie.text_size)
54
48
 
55
49
        for f in ['text_id', 'text_sha1', 'symlink_target']:
56
50
            v = getattr(ie, f)
57
 
            if v != None:
 
51
            if v is not None:
58
52
                e.set(f, v)
59
53
 
60
54
        # to be conservative, we don't externalize the root pointers
61
55
        # for now, leaving them as null in the xml form.  in a future
62
56
        # version it will be implied by nested elements.
63
57
        if ie.parent_id != ROOT_ID:
64
 
            assert isinstance(ie.parent_id, basestring)
65
58
            e.set('parent_id', ie.parent_id)
66
59
 
67
60
        e.tail = '\n'
69
62
        return e
70
63
 
71
64
 
72
 
    def _unpack_inventory(self, elt):
 
65
    def _unpack_inventory(self, elt, revision_id=None, entry_cache=None,
 
66
                          return_from_cache=False):
73
67
        """Construct from XML Element
 
68
 
 
69
        :param revision_id: Ignored parameter used by xml5.
74
70
        """
75
 
        assert elt.tag == 'inventory'
76
71
        root_id = elt.get('file_id') or ROOT_ID
77
72
        inv = Inventory(root_id)
78
73
        for e in elt:
79
 
            ie = self._unpack_entry(e)
 
74
            ie = self._unpack_entry(e, entry_cache=entry_cache,
 
75
                                    return_from_cache=return_from_cache)
80
76
            if ie.parent_id == ROOT_ID:
81
77
                ie.parent_id = root_id
82
78
            inv.add(ie)
83
79
        return inv
84
80
 
85
81
 
86
 
    def _unpack_entry(self, elt):
87
 
        assert elt.tag == 'entry'
88
 
 
 
82
    def _unpack_entry(self, elt, entry_cache=None, return_from_cache=False):
89
83
        ## original format inventories don't have a parent_id for
90
84
        ## nodes in the root directory, but it's cleaner to use one
91
85
        ## internally.
92
86
        parent_id = elt.get('parent_id')
93
 
        if parent_id == None:
 
87
        if parent_id is None:
94
88
            parent_id = ROOT_ID
95
89
 
96
90
        kind = elt.get('kind')
133
127
        root.text = '\n'
134
128
 
135
129
        msg = SubElement(root, 'message')
136
 
        msg.text = rev.message
 
130
        msg.text = escape_invalid_chars(rev.message)[0]
137
131
        msg.tail = '\n'
138
132
 
139
133
        if rev.parents:
142
136
            for i, parent_id in enumerate(rev.parents):
143
137
                p = SubElement(pelts, 'revision_ref')
144
138
                p.tail = '\n'
145
 
                assert parent_id
146
139
                p.set('revision_id', parent_id)
147
140
                if i < len(rev.parent_sha1s):
148
141
                    p.set('revision_sha1', rev.parent_sha1s[i])
149
142
        return root
150
143
 
151
 
    
 
144
 
152
145
    def _unpack_revision(self, elt):
153
146
        """XML Element -> Revision object"""
154
 
        
 
147
 
155
148
        # <changeset> is deprecated...
156
149
        if elt.tag not in ('revision', 'changeset'):
157
150
            raise BzrError("unexpected tag in revision file: %r" % elt)
170
163
 
171
164
        if pelts:
172
165
            for p in pelts:
173
 
                assert p.tag == 'revision_ref', \
174
 
                       "bad parent node tag %r" % p.tag
175
166
                rev.parent_ids.append(p.get('revision_id'))
176
167
                rev.parent_sha1s.append(p.get('revision_sha1'))
177
168
            if precursor:
178
169
                # must be consistent
179
170
                prec_parent = rev.parent_ids[0]
180
 
                assert prec_parent == precursor
181
171
        elif precursor:
182
172
            # revisions written prior to 0.0.5 have a single precursor
183
173
            # give as an attribute