← 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-07 10:13:43 UTC
  • mfrom: (0.1.95)
  • Revision ID: mbp@sourcefrog.net-20050707101342-aa2ef950004cb467
todo

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/inventory.py
27
27
 
28
28
from bzrlib.osutils import uuid, quotefn, splitpath, joinpath, appendpath
29
29
from bzrlib.trace import mutter
30
 
from bzrlib.errors import NotVersionedError
31
 
        
32
30
 
33
31
class InventoryEntry(object):
34
32
    """Description of a versioned file.
55
53
    >>> i.path2id('')
56
54
    'TREE_ROOT'
57
55
    >>> i.add(InventoryEntry('123', 'src', 'directory', ROOT_ID))
58
 
    InventoryEntry('123', 'src', kind='directory', parent_id='TREE_ROOT')
59
56
    >>> i.add(InventoryEntry('2323', 'hello.c', 'file', parent_id='123'))
60
 
    InventoryEntry('2323', 'hello.c', kind='file', parent_id='123')
61
57
    >>> for j in i.iter_entries():
62
58
    ...   print j
63
59
    ... 
68
64
    ...
69
65
    BzrError: inventory already contains entry with id {2323}
70
66
    >>> i.add(InventoryEntry('2324', 'bye.c', 'file', '123'))
71
 
    InventoryEntry('2324', 'bye.c', kind='file', parent_id='123')
72
67
    >>> i.add(InventoryEntry('2325', 'wibble', 'directory', '123'))
73
 
    InventoryEntry('2325', 'wibble', kind='directory', parent_id='123')
74
68
    >>> i.path2id('src/wibble')
75
69
    '2325'
76
70
    >>> '2325' in i
77
71
    True
78
72
    >>> i.add(InventoryEntry('2326', 'wibble.c', 'file', '2325'))
79
 
    InventoryEntry('2326', 'wibble.c', kind='file', parent_id='2325')
80
73
    >>> i['2326']
81
74
    InventoryEntry('2326', 'wibble.c', kind='file', parent_id='2325')
82
75
    >>> for j in i.iter_entries():
99
92
    # TODO: split InventoryEntry into subclasses for files,
100
93
    # directories, etc etc.
101
94
 
102
 
    __slots__ = ['text_sha1', 'text_size', 'file_id', 'name', 'kind',
103
 
                 'text_id', 'parent_id', 'children', ]
104
 
 
 
95
    text_sha1 = None
 
96
    text_size = None
 
97
    
105
98
    def __init__(self, file_id, name, kind, parent_id, text_id=None):
106
99
        """Create an InventoryEntry
107
100
        
120
113
        if '/' in name or '\\' in name:
121
114
            raise BzrCheckError('InventoryEntry name %r is invalid' % name)
122
115
        
123
 
        self.text_sha1 = None
124
 
        self.text_size = None
125
 
    
126
116
        self.file_id = file_id
127
117
        self.name = name
128
118
        self.kind = kind
273
263
 
274
264
    >>> inv = Inventory()
275
265
    >>> inv.add(InventoryEntry('123-123', 'hello.c', 'file', ROOT_ID))
276
 
    InventoryEntry('123-123', 'hello.c', kind='file', parent_id='TREE_ROOT')
277
266
    >>> inv['123-123'].name
278
267
    'hello.c'
279
268
 
288
277
 
289
278
    >>> [x[0] for x in inv.iter_entries()]
290
279
    ['hello.c']
291
 
    >>> inv = Inventory('TREE_ROOT-12345678-12345678')
292
 
    >>> inv.add(InventoryEntry('123-123', 'hello.c', 'file', ROOT_ID))
293
 
    InventoryEntry('123-123', 'hello.c', kind='file', parent_id='TREE_ROOT-12345678-12345678')
294
280
    """
295
 
    def __init__(self, root_id=ROOT_ID):
 
281
    def __init__(self):
296
282
        """Create or read an inventory.
297
283
 
298
284
        If a working directory is specified, the inventory is read
302
288
        The inventory is created with a default root directory, with
303
289
        an id of None.
304
290
        """
305
 
        # We are letting Branch(init=True) create a unique inventory
306
 
        # root id. Rather than generating a random one here.
307
 
        #if root_id is None:
308
 
        #    root_id = bzrlib.branch.gen_file_id('TREE_ROOT')
309
 
        self.root = RootEntry(root_id)
 
291
        self.root = RootEntry(ROOT_ID)
310
292
        self._byid = {self.root.file_id: self.root}
311
293
 
312
294
 
378
360
 
379
361
        >>> inv = Inventory()
380
362
        >>> inv.add(InventoryEntry('123', 'foo.c', 'file', ROOT_ID))
381
 
        InventoryEntry('123', 'foo.c', kind='file', parent_id='TREE_ROOT')
382
363
        >>> '123' in inv
383
364
        True
384
365
        >>> '456' in inv
392
373
 
393
374
        >>> inv = Inventory()
394
375
        >>> inv.add(InventoryEntry('123123', 'hello.c', 'file', ROOT_ID))
395
 
        InventoryEntry('123123', 'hello.c', kind='file', parent_id='TREE_ROOT')
396
376
        >>> inv['123123'].name
397
377
        'hello.c'
398
378
        """
420
400
        if entry.file_id in self._byid:
421
401
            raise BzrError("inventory already contains entry with id {%s}" % entry.file_id)
422
402
 
423
 
        if entry.parent_id == ROOT_ID or entry.parent_id is None:
424
 
            entry.parent_id = self.root.file_id
425
 
 
426
403
        try:
427
404
            parent = self._byid[entry.parent_id]
428
405
        except KeyError:
434
411
 
435
412
        self._byid[entry.file_id] = entry
436
413
        parent.children[entry.name] = entry
437
 
        return entry
438
414
 
439
415
 
440
416
    def add_path(self, relpath, kind, file_id=None):
441
417
        """Add entry from a path.
442
418
 
443
419
        The immediate parent must already be versioned"""
444
 
        from bzrlib.branch import gen_file_id
 
420
        from bzrlib.errors import NotVersionedError
445
421
        
446
422
        parts = bzrlib.osutils.splitpath(relpath)
447
423
        if len(parts) == 0:
448
424
            raise BzrError("cannot re-add root of inventory")
449
425
 
450
426
        if file_id == None:
 
427
            from bzrlib.branch import gen_file_id
451
428
            file_id = gen_file_id(relpath)
452
429
 
453
430
        parent_path = parts[:-1]
465
442
 
466
443
        >>> inv = Inventory()
467
444
        >>> inv.add(InventoryEntry('123', 'foo.c', 'file', ROOT_ID))
468
 
        InventoryEntry('123', 'foo.c', kind='file', parent_id='TREE_ROOT')
469
445
        >>> '123' in inv
470
446
        True
471
447
        >>> del inv['123']
493
469
        
494
470
        e = Element('inventory')
495
471
        e.text = '\n'
496
 
        if self.root.file_id not in (None, ROOT_ID):
497
 
            e.set('file_id', self.root.file_id)
498
472
        for path, ie in self.iter_entries():
499
473
            e.append(ie.to_element())
500
474
        return e
505
479
        
506
480
        >>> inv = Inventory()
507
481
        >>> 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
482
        >>> elt = inv.to_element()
510
483
        >>> inv2 = Inventory.from_element(elt)
511
484
        >>> inv2 == inv
513
486
        """
514
487
        # XXXX: doctest doesn't run this properly under python2.3
515
488
        assert elt.tag == 'inventory'
516
 
        root_id = elt.get('file_id') or ROOT_ID
517
 
        o = cls(root_id)
 
489
        o = cls()
518
490
        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)
 
491
            o.add(InventoryEntry.from_element(e))
523
492
        return o
524
493
        
525
494
    from_element = classmethod(from_element)
533
502
        >>> i1 == i2
534
503
        True
535
504
        >>> i1.add(InventoryEntry('123', 'foo', 'file', ROOT_ID))
536
 
        InventoryEntry('123', 'foo', kind='file', parent_id='TREE_ROOT')
537
505
        >>> i1 == i2
538
506
        False
539
507
        >>> i2.add(InventoryEntry('123', 'foo', 'file', ROOT_ID))
540
 
        InventoryEntry('123', 'foo', kind='file', parent_id='TREE_ROOT')
541
508
        >>> i1 == i2
542
509
        True
543
510
        """
583
550
        """Return as a list the path to file_id."""
584
551
 
585
552
        # get all names, skipping root
586
 
        p = [self._byid[fid].name for fid in self.get_idpath(file_id)[1:]]
 
553
        p = [self[fid].name for fid in self.get_idpath(file_id)[1:]]
587
554
        return os.sep.join(p)
588
555
            
589
556
 
658
625
 
659
626
 
660
627
 
661
 
_NAME_RE = None
 
628
_NAME_RE = re.compile(r'^[^/\\]+$')
662
629
 
663
630
def is_valid_name(name):
664
 
    global _NAME_RE
665
 
    if _NAME_RE == None:
666
 
        _NAME_RE = re.compile(r'^[^/\\]+$')
667
 
        
668
631
    return bool(_NAME_RE.match(name))