/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/newinventory.py

  • Committer: mbp at sourcefrog
  • Date: 2005-04-07 05:47:05 UTC
  • Revision ID: mbp@sourcefrog.net-20050407054705-be8af6cc3e31b15583f66155
- experimental remote-branch support
- fix up newinventory code

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/newinventory.py
 
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
 
 
17
from cElementTree import Element, ElementTree, SubElement
 
18
 
 
19
 
 
20
def write_inventory(inv, f):
 
21
    el = Element('inventory', {'version': '2'})
 
22
    
 
23
    root = Element('root_directory', {'id': inv.root.file_id})
 
24
    el.append(root)
 
25
 
 
26
    def descend(parent_el, ie):
 
27
        kind = ie.kind
 
28
        el = Element(kind, {'name': ie.name,
 
29
                            'id': ie.file_id,})
 
30
        
 
31
        if kind == 'file':
 
32
            if ie.text_id:
 
33
                el.set('text_id', ie.text_id)
 
34
            if ie.text_sha1:
 
35
                el.set('text_sha1', ie.text_sha1)
 
36
            if ie.text_size != None:
 
37
                el.set('text_size', ('%d' % ie.text_size))
 
38
        elif kind != 'directory':
 
39
            bailout('unknown InventoryEntry kind %r' % kind)
 
40
 
 
41
        el.tail = '\n'
 
42
        parent_el.append(el)
 
43
 
 
44
        if kind == 'directory':
 
45
            l = ie.children.items()
 
46
            l.sort()
 
47
            for child_name, child_ie in l:
 
48
                descend(el, child_ie)
 
49
                
 
50
        
 
51
    # walk down through inventory, adding all directories
 
52
 
 
53
    l = inv._root.children.items()
 
54
    l.sort()
 
55
    for entry_name, ie in l:
 
56
        descend(root, ie)
 
57
    
 
58
    ElementTree(el).write(f, 'utf-8')
 
59
    f.write('\n')
 
60
 
 
61
 
 
62
# This writes out an inventory without building an XML tree first,
 
63
# just to see if it's faster.  Not currently used.
 
64
def write_slacker_inventory(inv, f):
 
65
    def descend(ie):
 
66
        kind = ie.kind
 
67
        f.write('<%s name="%s" id="%s" ' % (kind, ie.name, ie.file_id))
 
68
 
 
69
        if kind == 'file':
 
70
            if ie.text_id:
 
71
                f.write('text_id="%s" ' % ie.text_id)
 
72
            if ie.text_sha1:
 
73
                f.write('text_sha1="%s" ' % ie.text_sha1)
 
74
            if ie.text_size != None:
 
75
                f.write('text_size="%d" ' % ie.text_size)
 
76
            f.write('/>\n')
 
77
        elif kind == 'directory':
 
78
            f.write('>\n')
 
79
            
 
80
            l = ie.children.items()
 
81
            l.sort()
 
82
            for child_name, child_ie in l:
 
83
                descend(child_ie)
 
84
 
 
85
            f.write('</directory>\n')
 
86
        else:
 
87
            bailout('unknown InventoryEntry kind %r' % kind)
 
88
 
 
89
    f.write('<inventory>\n')
 
90
    f.write('<root_directory id="bogus-root-id">\n')
 
91
 
 
92
    l = inv._root.children.items()
 
93
    l.sort()
 
94
    for entry_name, ie in l:
 
95
        descend(ie)
 
96
 
 
97
    f.write('</root_directory>\n')
 
98
    f.write('</inventory>\n')
 
99
    
 
100
 
 
101
 
 
102
def read_new_inventory(f):
 
103
    from inventory import Inventory, InventoryEntry
 
104
    
 
105
    def descend(parent_ie, el):
 
106
        kind = el.tag
 
107
        name = el.get('name')
 
108
        file_id = el.get('id')
 
109
        ie = InventoryEntry(file_id, name, el.tag)
 
110
        parent_ie.children[name] = ie
 
111
        inv._byid[file_id] = ie
 
112
        if kind == 'directory':
 
113
            for child_el in el:
 
114
                descend(ie, child_el)
 
115
        elif kind == 'file':
 
116
            assert len(el) == 0
 
117
            ie.text_id = el.get('text_id')
 
118
            v = el.get('text_size')
 
119
            ie.text_size = v and int(v)
 
120
            ie.text_sha1 = el.get('text_sha1')
 
121
        else:
 
122
            bailout("unknown inventory entry %r" % kind)
 
123
 
 
124
    inv_el = ElementTree().parse(f)
 
125
    assert inv_el.tag == 'inventory'
 
126
    root_el = inv_el[0]
 
127
    assert root_el.tag == 'root_directory'
 
128
 
 
129
    inv = Inventory()
 
130
    for el in root_el:
 
131
        descend(inv._root, el)