← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/inventory.py

  • Committer: Martin Pool
  • Date: 2005-07-11 07:33:45 UTC
  • Revision ID: mbp@sourcefrog.net-20050711073345-177b10e674961bda
- merge John's code to give the tree root an explicit file id

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/inventory.py
277
277
 
278
278
    >>> [x[0] for x in inv.iter_entries()]
279
279
    ['hello.c']
 
280
    >>> inv = Inventory('TREE_ROOT-12345678-12345678')
 
281
    >>> inv.add(InventoryEntry('123-123', 'hello.c', 'file', ROOT_ID))
280
282
    """
281
 
    def __init__(self):
 
283
    def __init__(self, root_id=ROOT_ID):
282
284
        """Create or read an inventory.
283
285
 
284
286
        If a working directory is specified, the inventory is read
288
290
        The inventory is created with a default root directory, with
289
291
        an id of None.
290
292
        """
291
 
        self.root = RootEntry(ROOT_ID)
 
293
        # We are letting Branch(init=True) create a unique inventory
 
294
        # root id. Rather than generating a random one here.
 
295
        #if root_id is None:
 
296
        #    root_id = bzrlib.branch.gen_file_id('TREE_ROOT')
 
297
        self.root = RootEntry(root_id)
292
298
        self._byid = {self.root.file_id: self.root}
293
299
 
294
300
 
400
406
        if entry.file_id in self._byid:
401
407
            raise BzrError("inventory already contains entry with id {%s}" % entry.file_id)
402
408
 
 
409
        if entry.parent_id == ROOT_ID or entry.parent_id is None:
 
410
            entry.parent_id = self.root.file_id
 
411
 
403
412
        try:
404
413
            parent = self._byid[entry.parent_id]
405
414
        except KeyError:
469
478
        
470
479
        e = Element('inventory')
471
480
        e.text = '\n'
 
481
        if self.root.file_id not in (None, ROOT_ID):
 
482
            e.set('file_id', self.root.file_id)
472
483
        for path, ie in self.iter_entries():
473
484
            e.append(ie.to_element())
474
485
        return e
486
497
        """
487
498
        # XXXX: doctest doesn't run this properly under python2.3
488
499
        assert elt.tag == 'inventory'
489
 
        o = cls()
 
500
        root_id = elt.get('file_id') or ROOT_ID
 
501
        o = cls(root_id)
490
502
        for e in elt:
491
 
            o.add(InventoryEntry.from_element(e))
 
503
            ie = InventoryEntry.from_element(e)
 
504
            if ie.parent_id == ROOT_ID:
 
505
                ie.parent_id = root_id
 
506
            o.add(ie)
492
507
        return o
493
508
        
494
509
    from_element = classmethod(from_element)