~bzr-pqm/bzr/bzr.dev

149 by mbp at sourcefrog
experiment with new nested inventory file format
1
# (C) 2005 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
800 by Martin Pool
Merge John's import-speedup branch:
17
from xml import ElementTree, Element
149 by mbp at sourcefrog
experiment with new nested inventory file format
18
19
20
def write_inventory(inv, f):
21
    el = Element('inventory', {'version': '2'})
189 by mbp at sourcefrog
improved new-inventory
22
    el.text = '\n'
149 by mbp at sourcefrog
experiment with new nested inventory file format
23
    
188 by mbp at sourcefrog
- experimental remote-branch support
24
    root = Element('root_directory', {'id': inv.root.file_id})
189 by mbp at sourcefrog
improved new-inventory
25
    root.tail = root.text = '\n'
149 by mbp at sourcefrog
experiment with new nested inventory file format
26
    el.append(root)
27
28
    def descend(parent_el, ie):
29
        kind = ie.kind
30
        el = Element(kind, {'name': ie.name,
31
                            'id': ie.file_id,})
32
        
33
        if kind == 'file':
34
            if ie.text_id:
35
                el.set('text_id', ie.text_id)
36
            if ie.text_sha1:
37
                el.set('text_sha1', ie.text_sha1)
38
            if ie.text_size != None:
39
                el.set('text_size', ('%d' % ie.text_size))
40
        elif kind != 'directory':
694 by Martin Pool
- weed out all remaining calls to bailout() and remove the function
41
            raise BzrError('unknown InventoryEntry kind %r' % kind)
150 by mbp at sourcefrog
experiment in writing XML by hand, not through ElementTree
42
43
        el.tail = '\n'
149 by mbp at sourcefrog
experiment with new nested inventory file format
44
        parent_el.append(el)
45
46
        if kind == 'directory':
189 by mbp at sourcefrog
improved new-inventory
47
            el.text = '\n' # break before having children
149 by mbp at sourcefrog
experiment with new nested inventory file format
48
            l = ie.children.items()
49
            l.sort()
50
            for child_name, child_ie in l:
51
                descend(el, child_ie)
52
                
53
        
54
    # walk down through inventory, adding all directories
55
189 by mbp at sourcefrog
improved new-inventory
56
    l = inv.root.children.items()
149 by mbp at sourcefrog
experiment with new nested inventory file format
57
    l.sort()
58
    for entry_name, ie in l:
59
        descend(root, ie)
60
    
61
    ElementTree(el).write(f, 'utf-8')
62
    f.write('\n')
150 by mbp at sourcefrog
experiment in writing XML by hand, not through ElementTree
63
64
191 by mbp at sourcefrog
more XML performance tests
65
66
def escape_attr(text):
67
    return text.replace("&", "&") \
68
           .replace("'", "'") \
69
           .replace('"', """) \
70
           .replace("<", "&lt;") \
71
           .replace(">", "&gt;")
72
73
188 by mbp at sourcefrog
- experimental remote-branch support
74
# This writes out an inventory without building an XML tree first,
75
# just to see if it's faster.  Not currently used.
150 by mbp at sourcefrog
experiment in writing XML by hand, not through ElementTree
76
def write_slacker_inventory(inv, f):
77
    def descend(ie):
78
        kind = ie.kind
191 by mbp at sourcefrog
more XML performance tests
79
        f.write('<%s name="%s" id="%s" ' % (kind, escape_attr(ie.name),
80
                                            escape_attr(ie.file_id)))
150 by mbp at sourcefrog
experiment in writing XML by hand, not through ElementTree
81
82
        if kind == 'file':
83
            if ie.text_id:
84
                f.write('text_id="%s" ' % ie.text_id)
85
            if ie.text_sha1:
86
                f.write('text_sha1="%s" ' % ie.text_sha1)
87
            if ie.text_size != None:
88
                f.write('text_size="%d" ' % ie.text_size)
89
            f.write('/>\n')
90
        elif kind == 'directory':
91
            f.write('>\n')
92
            
93
            l = ie.children.items()
94
            l.sort()
95
            for child_name, child_ie in l:
96
                descend(child_ie)
97
98
            f.write('</directory>\n')
99
        else:
694 by Martin Pool
- weed out all remaining calls to bailout() and remove the function
100
            raise BzrError('unknown InventoryEntry kind %r' % kind)
150 by mbp at sourcefrog
experiment in writing XML by hand, not through ElementTree
101
102
    f.write('<inventory>\n')
191 by mbp at sourcefrog
more XML performance tests
103
    f.write('<root_directory id="%s">\n' % escape_attr(inv.root.file_id))
150 by mbp at sourcefrog
experiment in writing XML by hand, not through ElementTree
104
189 by mbp at sourcefrog
improved new-inventory
105
    l = inv.root.children.items()
150 by mbp at sourcefrog
experiment in writing XML by hand, not through ElementTree
106
    l.sort()
107
    for entry_name, ie in l:
108
        descend(ie)
109
151 by mbp at sourcefrog
experimental nested-inventory load support
110
    f.write('</root_directory>\n')
150 by mbp at sourcefrog
experiment in writing XML by hand, not through ElementTree
111
    f.write('</inventory>\n')
112
    
113
114
151 by mbp at sourcefrog
experimental nested-inventory load support
115
def read_new_inventory(f):
116
    from inventory import Inventory, InventoryEntry
117
    
118
    def descend(parent_ie, el):
119
        kind = el.tag
120
        name = el.get('name')
121
        file_id = el.get('id')
122
        ie = InventoryEntry(file_id, name, el.tag)
123
        parent_ie.children[name] = ie
124
        inv._byid[file_id] = ie
125
        if kind == 'directory':
126
            for child_el in el:
127
                descend(ie, child_el)
128
        elif kind == 'file':
129
            assert len(el) == 0
130
            ie.text_id = el.get('text_id')
131
            v = el.get('text_size')
132
            ie.text_size = v and int(v)
133
            ie.text_sha1 = el.get('text_sha1')
134
        else:
694 by Martin Pool
- weed out all remaining calls to bailout() and remove the function
135
            raise BzrError("unknown inventory entry %r" % kind)
151 by mbp at sourcefrog
experimental nested-inventory load support
136
137
    inv_el = ElementTree().parse(f)
138
    assert inv_el.tag == 'inventory'
139
    root_el = inv_el[0]
140
    assert root_el.tag == 'root_directory'
141
142
    inv = Inventory()
143
    for el in root_el:
189 by mbp at sourcefrog
improved new-inventory
144
        descend(inv.root, el)