/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
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
 
from xml import ElementTree, Element
 
17
from cElementTree import Element, ElementTree, SubElement
18
18
 
19
19
 
20
20
def write_inventory(inv, f):
21
21
    el = Element('inventory', {'version': '2'})
22
 
    el.text = '\n'
23
22
    
24
23
    root = Element('root_directory', {'id': inv.root.file_id})
25
 
    root.tail = root.text = '\n'
26
24
    el.append(root)
27
25
 
28
26
    def descend(parent_el, ie):
38
36
            if ie.text_size != None:
39
37
                el.set('text_size', ('%d' % ie.text_size))
40
38
        elif kind != 'directory':
41
 
            raise BzrError('unknown InventoryEntry kind %r' % kind)
 
39
            bailout('unknown InventoryEntry kind %r' % kind)
42
40
 
43
41
        el.tail = '\n'
44
42
        parent_el.append(el)
45
43
 
46
44
        if kind == 'directory':
47
 
            el.text = '\n' # break before having children
48
45
            l = ie.children.items()
49
46
            l.sort()
50
47
            for child_name, child_ie in l:
53
50
        
54
51
    # walk down through inventory, adding all directories
55
52
 
56
 
    l = inv.root.children.items()
 
53
    l = inv._root.children.items()
57
54
    l.sort()
58
55
    for entry_name, ie in l:
59
56
        descend(root, ie)
62
59
    f.write('\n')
63
60
 
64
61
 
65
 
 
66
 
def escape_attr(text):
67
 
    return text.replace("&", "&") \
68
 
           .replace("'", "'") \
69
 
           .replace('"', """) \
70
 
           .replace("<", "&lt;") \
71
 
           .replace(">", "&gt;")
72
 
 
73
 
 
74
62
# This writes out an inventory without building an XML tree first,
75
63
# just to see if it's faster.  Not currently used.
76
64
def write_slacker_inventory(inv, f):
77
65
    def descend(ie):
78
66
        kind = ie.kind
79
 
        f.write('<%s name="%s" id="%s" ' % (kind, escape_attr(ie.name),
80
 
                                            escape_attr(ie.file_id)))
 
67
        f.write('<%s name="%s" id="%s" ' % (kind, ie.name, ie.file_id))
81
68
 
82
69
        if kind == 'file':
83
70
            if ie.text_id:
97
84
 
98
85
            f.write('</directory>\n')
99
86
        else:
100
 
            raise BzrError('unknown InventoryEntry kind %r' % kind)
 
87
            bailout('unknown InventoryEntry kind %r' % kind)
101
88
 
102
89
    f.write('<inventory>\n')
103
 
    f.write('<root_directory id="%s">\n' % escape_attr(inv.root.file_id))
 
90
    f.write('<root_directory id="bogus-root-id">\n')
104
91
 
105
 
    l = inv.root.children.items()
 
92
    l = inv._root.children.items()
106
93
    l.sort()
107
94
    for entry_name, ie in l:
108
95
        descend(ie)
132
119
            ie.text_size = v and int(v)
133
120
            ie.text_sha1 = el.get('text_sha1')
134
121
        else:
135
 
            raise BzrError("unknown inventory entry %r" % kind)
 
122
            bailout("unknown inventory entry %r" % kind)
136
123
 
137
124
    inv_el = ElementTree().parse(f)
138
125
    assert inv_el.tag == 'inventory'
139
126
    root_el = inv_el[0]
140
127
    assert root_el.tag == 'root_directory'
141
128
 
142
 
    inv = Inventory(inv_el.get('file_id'))
 
129
    inv = Inventory()
143
130
    for el in root_el:
144
 
        descend(inv.root, el)
 
131
        descend(inv._root, el)