/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/inventory.py

  • Committer: Martin Pool
  • Date: 2005-06-22 08:12:31 UTC
  • Revision ID: mbp@sourcefrog.net-20050622081231-630e514240ccb87a
- new exception NotVersionedError
- raise this from Inventory.add_path if parent isnt versioned

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/inventory.py
22
22
 
23
23
import sys, os.path, types, re
24
24
 
 
25
try:
 
26
    from cElementTree import Element, ElementTree, SubElement
 
27
except ImportError:
 
28
    from elementtree.ElementTree import Element, ElementTree, SubElement
 
29
 
 
30
from bzrlib.xml import XMLMixin
 
31
from bzrlib.errors import BzrError, BzrCheckError
 
32
 
25
33
import bzrlib
26
 
from bzrlib.errors import BzrError, BzrCheckError
27
 
 
28
34
from bzrlib.osutils import uuid, quotefn, splitpath, joinpath, appendpath
29
35
from bzrlib.trace import mutter
30
 
from bzrlib.errors import NotVersionedError
31
 
        
32
36
 
33
 
class InventoryEntry(object):
 
37
class InventoryEntry(XMLMixin):
34
38
    """Description of a versioned file.
35
39
 
36
40
    An InventoryEntry has the following fields, which are also
94
98
    # TODO: split InventoryEntry into subclasses for files,
95
99
    # directories, etc etc.
96
100
 
97
 
    __slots__ = ['text_sha1', 'text_size', 'file_id', 'name', 'kind',
98
 
                 'text_id', 'parent_id', 'children', ]
99
 
 
 
101
    text_sha1 = None
 
102
    text_size = None
 
103
    
100
104
    def __init__(self, file_id, name, kind, parent_id, text_id=None):
101
105
        """Create an InventoryEntry
102
106
        
115
119
        if '/' in name or '\\' in name:
116
120
            raise BzrCheckError('InventoryEntry name %r is invalid' % name)
117
121
        
118
 
        self.text_sha1 = None
119
 
        self.text_size = None
120
 
    
121
122
        self.file_id = file_id
122
123
        self.name = name
123
124
        self.kind = kind
159
160
    
160
161
    def to_element(self):
161
162
        """Convert to XML element"""
162
 
        from bzrlib.xml import Element
163
 
        
164
163
        e = Element('entry')
165
164
 
166
165
        e.set('name', self.name)
249
248
 
250
249
 
251
250
 
252
 
class Inventory(object):
 
251
class Inventory(XMLMixin):
253
252
    """Inventory of versioned files in a tree.
254
253
 
255
254
    This describes which file_id is present at each point in the tree,
267
266
    inserted, other than through the Inventory API.
268
267
 
269
268
    >>> inv = Inventory()
 
269
    >>> inv.write_xml(sys.stdout)
 
270
    <inventory>
 
271
    </inventory>
270
272
    >>> inv.add(InventoryEntry('123-123', 'hello.c', 'file', ROOT_ID))
271
273
    >>> inv['123-123'].name
272
274
    'hello.c'
282
284
 
283
285
    >>> [x[0] for x in inv.iter_entries()]
284
286
    ['hello.c']
285
 
    >>> inv = Inventory('TREE_ROOT-12345678-12345678')
286
 
    >>> inv.add(InventoryEntry('123-123', 'hello.c', 'file', ROOT_ID))
 
287
    
 
288
    >>> inv.write_xml(sys.stdout)
 
289
    <inventory>
 
290
    <entry file_id="123-123" kind="file" name="hello.c" />
 
291
    </inventory>
 
292
 
287
293
    """
288
 
    def __init__(self, root_id=ROOT_ID):
 
294
    def __init__(self):
289
295
        """Create or read an inventory.
290
296
 
291
297
        If a working directory is specified, the inventory is read
295
301
        The inventory is created with a default root directory, with
296
302
        an id of None.
297
303
        """
298
 
        # We are letting Branch(init=True) create a unique inventory
299
 
        # root id. Rather than generating a random one here.
300
 
        #if root_id is None:
301
 
        #    root_id = bzrlib.branch.gen_file_id('TREE_ROOT')
302
 
        self.root = RootEntry(root_id)
 
304
        self.root = RootEntry(ROOT_ID)
303
305
        self._byid = {self.root.file_id: self.root}
304
306
 
305
307
 
411
413
        if entry.file_id in self._byid:
412
414
            raise BzrError("inventory already contains entry with id {%s}" % entry.file_id)
413
415
 
414
 
        if entry.parent_id == ROOT_ID or entry.parent_id is None:
415
 
            entry.parent_id = self.root.file_id
416
 
 
417
416
        try:
418
417
            parent = self._byid[entry.parent_id]
419
418
        except KeyError:
431
430
        """Add entry from a path.
432
431
 
433
432
        The immediate parent must already be versioned"""
434
 
        from bzrlib.branch import gen_file_id
 
433
        from bzrlib.errors import NotVersionedError
435
434
        
436
435
        parts = bzrlib.osutils.splitpath(relpath)
437
436
        if len(parts) == 0:
438
437
            raise BzrError("cannot re-add root of inventory")
439
438
 
440
439
        if file_id == None:
441
 
            file_id = gen_file_id(relpath)
 
440
            file_id = bzrlib.branch.gen_file_id(relpath)
442
441
 
443
442
        parent_path = parts[:-1]
444
443
        parent_id = self.path2id(parent_path)
478
477
 
479
478
    def to_element(self):
480
479
        """Convert to XML Element"""
481
 
        from bzrlib.xml import Element
482
 
        
483
480
        e = Element('inventory')
484
481
        e.text = '\n'
485
 
        if self.root.file_id not in (None, ROOT_ID):
486
 
            e.set('file_id', self.root.file_id)
487
482
        for path, ie in self.iter_entries():
488
483
            e.append(ie.to_element())
489
484
        return e
491
486
 
492
487
    def from_element(cls, elt):
493
488
        """Construct from XML Element
494
 
        
 
489
 
495
490
        >>> inv = Inventory()
496
491
        >>> inv.add(InventoryEntry('foo.c-123981239', 'foo.c', 'file', ROOT_ID))
497
492
        >>> elt = inv.to_element()
499
494
        >>> inv2 == inv
500
495
        True
501
496
        """
502
 
        # XXXX: doctest doesn't run this properly under python2.3
503
497
        assert elt.tag == 'inventory'
504
 
        root_id = elt.get('file_id') or ROOT_ID
505
 
        o = cls(root_id)
 
498
        o = cls()
506
499
        for e in elt:
507
 
            ie = InventoryEntry.from_element(e)
508
 
            if ie.parent_id == ROOT_ID:
509
 
                ie.parent_id = root_id
510
 
            o.add(ie)
 
500
            o.add(InventoryEntry.from_element(e))
511
501
        return o
512
502
        
513
503
    from_element = classmethod(from_element)
569
559
        """Return as a list the path to file_id."""
570
560
 
571
561
        # get all names, skipping root
572
 
        p = [self._byid[fid].name for fid in self.get_idpath(file_id)[1:]]
 
562
        p = [self[fid].name for fid in self.get_idpath(file_id)[1:]]
573
563
        return os.sep.join(p)
574
564
            
575
565