← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/inventory.py

- fix up imports in remotebranch

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/inventory.py
100
100
    # directories, etc etc.
101
101
 
102
102
    __slots__ = ['text_sha1', 'text_size', 'file_id', 'name', 'kind',
103
 
                 'text_id', 'parent_id', 'children', ]
 
103
                 'text_id', 'parent_id', 'children',
 
104
                 'text_version', 'entry_version', ]
 
105
 
104
106
 
105
107
    def __init__(self, file_id, name, kind, parent_id, text_id=None):
106
108
        """Create an InventoryEntry
117
119
        Traceback (most recent call last):
118
120
        BzrCheckError: InventoryEntry name 'src/hello.c' is invalid
119
121
        """
 
122
        assert isinstance(name, basestring), name
120
123
        if '/' in name or '\\' in name:
121
124
            raise BzrCheckError('InventoryEntry name %r is invalid' % name)
122
125
        
 
126
        self.text_version = None
 
127
        self.entry_version = None
123
128
        self.text_sha1 = None
124
129
        self.text_size = None
125
 
    
126
130
        self.file_id = file_id
127
131
        self.name = name
128
132
        self.kind = kind
162
166
                   self.parent_id))
163
167
 
164
168
    
165
 
    def to_element(self):
166
 
        """Convert to XML element"""
167
 
        from bzrlib.xml import Element
168
 
        
169
 
        e = Element('entry')
170
 
 
171
 
        e.set('name', self.name)
172
 
        e.set('file_id', self.file_id)
173
 
        e.set('kind', self.kind)
174
 
 
175
 
        if self.text_size != None:
176
 
            e.set('text_size', '%d' % self.text_size)
177
 
            
178
 
        for f in ['text_id', 'text_sha1']:
179
 
            v = getattr(self, f)
180
 
            if v != None:
181
 
                e.set(f, v)
182
 
 
183
 
        # to be conservative, we don't externalize the root pointers
184
 
        # for now, leaving them as null in the xml form.  in a future
185
 
        # version it will be implied by nested elements.
186
 
        if self.parent_id != ROOT_ID:
187
 
            assert isinstance(self.parent_id, basestring)
188
 
            e.set('parent_id', self.parent_id)
189
 
 
190
 
        e.tail = '\n'
191
 
            
192
 
        return e
193
 
 
194
 
 
195
 
    def from_element(cls, elt):
196
 
        assert elt.tag == 'entry'
197
 
 
198
 
        ## original format inventories don't have a parent_id for
199
 
        ## nodes in the root directory, but it's cleaner to use one
200
 
        ## internally.
201
 
        parent_id = elt.get('parent_id')
202
 
        if parent_id == None:
203
 
            parent_id = ROOT_ID
204
 
 
205
 
        self = cls(elt.get('file_id'), elt.get('name'), elt.get('kind'), parent_id)
206
 
        self.text_id = elt.get('text_id')
207
 
        self.text_sha1 = elt.get('text_sha1')
208
 
        
209
 
        ## mutter("read inventoryentry: %r" % (elt.attrib))
210
 
 
211
 
        v = elt.get('text_size')
212
 
        self.text_size = v and int(v)
213
 
 
214
 
        return self
215
 
            
216
 
 
217
 
    from_element = classmethod(from_element)
218
 
 
219
169
    def __eq__(self, other):
220
170
        if not isinstance(other, InventoryEntry):
221
171
            return NotImplemented
226
176
               and (self.text_size == other.text_size) \
227
177
               and (self.text_id == other.text_id) \
228
178
               and (self.parent_id == other.parent_id) \
229
 
               and (self.kind == other.kind)
 
179
               and (self.kind == other.kind) \
 
180
               and (self.text_version == other.text_version) \
 
181
               and (self.entry_version == other.entry_version)
230
182
 
231
183
 
232
184
    def __ne__(self, other):
416
368
        """Add entry to inventory.
417
369
 
418
370
        To add  a file to a branch ready to be committed, use Branch.add,
419
 
        which calls this."""
 
371
        which calls this.
 
372
 
 
373
        Returns the new entry object.
 
374
        """
420
375
        if entry.file_id in self._byid:
421
376
            raise BzrError("inventory already contains entry with id {%s}" % entry.file_id)
422
377
 
440
395
    def add_path(self, relpath, kind, file_id=None):
441
396
        """Add entry from a path.
442
397
 
443
 
        The immediate parent must already be versioned"""
 
398
        The immediate parent must already be versioned.
 
399
 
 
400
        Returns the new entry object."""
444
401
        from bzrlib.branch import gen_file_id
445
402
        
446
403
        parts = bzrlib.osutils.splitpath(relpath)
487
444
        del self[ie.parent_id].children[ie.name]
488
445
 
489
446
 
490
 
    def to_element(self):
491
 
        """Convert to XML Element"""
492
 
        from bzrlib.xml import Element
493
 
        
494
 
        e = Element('inventory')
495
 
        e.text = '\n'
496
 
        if self.root.file_id not in (None, ROOT_ID):
497
 
            e.set('file_id', self.root.file_id)
498
 
        for path, ie in self.iter_entries():
499
 
            e.append(ie.to_element())
500
 
        return e
501
 
    
502
 
 
503
 
    def from_element(cls, elt):
504
 
        """Construct from XML Element
505
 
        
506
 
        >>> inv = Inventory()
507
 
        >>> inv.add(InventoryEntry('foo.c-123981239', 'foo.c', 'file', ROOT_ID))
508
 
        InventoryEntry('foo.c-123981239', 'foo.c', kind='file', parent_id='TREE_ROOT')
509
 
        >>> elt = inv.to_element()
510
 
        >>> inv2 = Inventory.from_element(elt)
511
 
        >>> inv2 == inv
512
 
        True
513
 
        """
514
 
        # XXXX: doctest doesn't run this properly under python2.3
515
 
        assert elt.tag == 'inventory'
516
 
        root_id = elt.get('file_id') or ROOT_ID
517
 
        o = cls(root_id)
518
 
        for e in elt:
519
 
            ie = InventoryEntry.from_element(e)
520
 
            if ie.parent_id == ROOT_ID:
521
 
                ie.parent_id = root_id
522
 
            o.add(ie)
523
 
        return o
524
 
        
525
 
    from_element = classmethod(from_element)
526
 
 
527
 
 
528
447
    def __eq__(self, other):
529
448
        """Compare two sets by comparing their contents.
530
449
 
559
478
        raise ValueError('not hashable')
560
479
 
561
480
 
562
 
 
563
481
    def get_idpath(self, file_id):
564
482
        """Return a list of file_ids for the path to an entry.
565
483