← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/xml4.py

  • Committer: John Arbash Meinel
  • Date: 2009-06-04 17:12:29 UTC
  • mto: This revision was merged to the branch mainline in revision 4410.
  • Revision ID: john@arbash-meinel.com-20090604171229-kbgfatt63y3u3uh1
Some small tweaks to decoding strings (avoid passing over the length 2x)

Down to 1.1s (from 1.4s) for decoding all of bzr.dev.
Also, favor decoding strings and then lists in _decode_object, since that is the
frequency we have those types inside Revisions.

Show diffs side-by-side

added added

removed removed

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