← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/newinventory.py

  • Committer: Tarmac
  • Author(s): Vincent Ladeuil
  • Date: 2017-01-30 14:42:05 UTC
  • mfrom: (6620.1.1 trunk)
  • Revision ID: tarmac-20170130144205-r8fh2xpmiuxyozpv
Merge  2.7 into trunk including fix for bug #1657238 [r=vila]

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': 'bogus-root-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
 
 
63
 
def write_slacker_inventory(inv, f):
64
 
    def descend(ie):
65
 
        kind = ie.kind
66
 
        f.write('<%s name="%s" id="%s" ' % (kind, ie.name, ie.file_id))
67
 
 
68
 
        if kind == 'file':
69
 
            if ie.text_id:
70
 
                f.write('text_id="%s" ' % ie.text_id)
71
 
            if ie.text_sha1:
72
 
                f.write('text_sha1="%s" ' % ie.text_sha1)
73
 
            if ie.text_size != None:
74
 
                f.write('text_size="%d" ' % ie.text_size)
75
 
            f.write('/>\n')
76
 
        elif kind == 'directory':
77
 
            f.write('>\n')
78
 
            
79
 
            l = ie.children.items()
80
 
            l.sort()
81
 
            for child_name, child_ie in l:
82
 
                descend(child_ie)
83
 
 
84
 
            f.write('</directory>\n')
85
 
        else:
86
 
            bailout('unknown InventoryEntry kind %r' % kind)
87
 
 
88
 
    f.write('<inventory>\n')
89
 
    f.write('<root_directory id="bogus-root-id">\n')
90
 
 
91
 
    l = inv._root.children.items()
92
 
    l.sort()
93
 
    for entry_name, ie in l:
94
 
        descend(ie)
95
 
 
96
 
    f.write('</root_directory>\n')
97
 
    f.write('</inventory>\n')
98
 
    
99
 
 
100
 
 
101
 
def read_new_inventory(f):
102
 
    from inventory import Inventory, InventoryEntry
103
 
    
104
 
    def descend(parent_ie, el):
105
 
        kind = el.tag
106
 
        name = el.get('name')
107
 
        file_id = el.get('id')
108
 
        ie = InventoryEntry(file_id, name, el.tag)
109
 
        parent_ie.children[name] = ie
110
 
        inv._byid[file_id] = ie
111
 
        if kind == 'directory':
112
 
            for child_el in el:
113
 
                descend(ie, child_el)
114
 
        elif kind == 'file':
115
 
            assert len(el) == 0
116
 
            ie.text_id = el.get('text_id')
117
 
            v = el.get('text_size')
118
 
            ie.text_size = v and int(v)
119
 
            ie.text_sha1 = el.get('text_sha1')
120
 
        else:
121
 
            bailout("unknown inventory entry %r" % kind)
122
 
 
123
 
    inv_el = ElementTree().parse(f)
124
 
    assert inv_el.tag == 'inventory'
125
 
    root_el = inv_el[0]
126
 
    assert root_el.tag == 'root_directory'
127
 
 
128
 
    inv = Inventory()
129
 
    for el in root_el:
130
 
        descend(inv._root, el)