← 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-18 14:28:13 UTC
  • Revision ID: mbp@sourcefrog.net-20050718142813-253ec02b9636bd98
- add stubbed-out test for clashing replace and delete

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
 
                 'text_version', 'entry_version', ]
105
 
 
106
 
 
 
95
    text_sha1 = None
 
96
    text_size = None
 
97
    
107
98
    def __init__(self, file_id, name, kind, parent_id, text_id=None):
108
99
        """Create an InventoryEntry
109
100
        
119
110
        Traceback (most recent call last):
120
111
        BzrCheckError: InventoryEntry name 'src/hello.c' is invalid
121
112
        """
122
 
        assert isinstance(name, basestring), name
123
113
        if '/' in name or '\\' in name:
124
114
            raise BzrCheckError('InventoryEntry name %r is invalid' % name)
125
115
        
126
 
        self.text_version = None
127
 
        self.entry_version = None
128
 
        self.text_sha1 = None
129
 
        self.text_size = None
130
116
        self.file_id = file_id
131
117
        self.name = name
132
118
        self.kind = kind
166
152
                   self.parent_id))
167
153
 
168
154
    
 
155
    def to_element(self):
 
156
        """Convert to XML element"""
 
157
        from bzrlib.xml import Element
 
158
        
 
159
        e = Element('entry')
 
160
 
 
161
        e.set('name', self.name)
 
162
        e.set('file_id', self.file_id)
 
163
        e.set('kind', self.kind)
 
164
 
 
165
        if self.text_size != None:
 
166
            e.set('text_size', '%d' % self.text_size)
 
167
            
 
168
        for f in ['text_id', 'text_sha1']:
 
169
            v = getattr(self, f)
 
170
            if v != None:
 
171
                e.set(f, v)
 
172
 
 
173
        # to be conservative, we don't externalize the root pointers
 
174
        # for now, leaving them as null in the xml form.  in a future
 
175
        # version it will be implied by nested elements.
 
176
        if self.parent_id != ROOT_ID:
 
177
            assert isinstance(self.parent_id, basestring)
 
178
            e.set('parent_id', self.parent_id)
 
179
 
 
180
        e.tail = '\n'
 
181
            
 
182
        return e
 
183
 
 
184
 
 
185
    def from_element(cls, elt):
 
186
        assert elt.tag == 'entry'
 
187
 
 
188
        ## original format inventories don't have a parent_id for
 
189
        ## nodes in the root directory, but it's cleaner to use one
 
190
        ## internally.
 
191
        parent_id = elt.get('parent_id')
 
192
        if parent_id == None:
 
193
            parent_id = ROOT_ID
 
194
 
 
195
        self = cls(elt.get('file_id'), elt.get('name'), elt.get('kind'), parent_id)
 
196
        self.text_id = elt.get('text_id')
 
197
        self.text_sha1 = elt.get('text_sha1')
 
198
        
 
199
        ## mutter("read inventoryentry: %r" % (elt.attrib))
 
200
 
 
201
        v = elt.get('text_size')
 
202
        self.text_size = v and int(v)
 
203
 
 
204
        return self
 
205
            
 
206
 
 
207
    from_element = classmethod(from_element)
 
208
 
169
209
    def __eq__(self, other):
170
210
        if not isinstance(other, InventoryEntry):
171
211
            return NotImplemented
176
216
               and (self.text_size == other.text_size) \
177
217
               and (self.text_id == other.text_id) \
178
218
               and (self.parent_id == other.parent_id) \
179
 
               and (self.kind == other.kind) \
180
 
               and (self.text_version == other.text_version) \
181
 
               and (self.entry_version == other.entry_version)
 
219
               and (self.kind == other.kind)
182
220
 
183
221
 
184
222
    def __ne__(self, other):
225
263
 
226
264
    >>> inv = Inventory()
227
265
    >>> inv.add(InventoryEntry('123-123', 'hello.c', 'file', ROOT_ID))
228
 
    InventoryEntry('123-123', 'hello.c', kind='file', parent_id='TREE_ROOT')
229
266
    >>> inv['123-123'].name
230
267
    'hello.c'
231
268
 
242
279
    ['hello.c']
243
280
    >>> inv = Inventory('TREE_ROOT-12345678-12345678')
244
281
    >>> inv.add(InventoryEntry('123-123', 'hello.c', 'file', ROOT_ID))
245
 
    InventoryEntry('123-123', 'hello.c', kind='file', parent_id='TREE_ROOT-12345678-12345678')
246
282
    """
247
283
    def __init__(self, root_id=ROOT_ID):
248
284
        """Create or read an inventory.
330
366
 
331
367
        >>> inv = Inventory()
332
368
        >>> inv.add(InventoryEntry('123', 'foo.c', 'file', ROOT_ID))
333
 
        InventoryEntry('123', 'foo.c', kind='file', parent_id='TREE_ROOT')
334
369
        >>> '123' in inv
335
370
        True
336
371
        >>> '456' in inv
344
379
 
345
380
        >>> inv = Inventory()
346
381
        >>> inv.add(InventoryEntry('123123', 'hello.c', 'file', ROOT_ID))
347
 
        InventoryEntry('123123', 'hello.c', kind='file', parent_id='TREE_ROOT')
348
382
        >>> inv['123123'].name
349
383
        'hello.c'
350
384
        """
368
402
        """Add entry to inventory.
369
403
 
370
404
        To add  a file to a branch ready to be committed, use Branch.add,
371
 
        which calls this.
372
 
 
373
 
        Returns the new entry object.
374
 
        """
 
405
        which calls this."""
375
406
        if entry.file_id in self._byid:
376
407
            raise BzrError("inventory already contains entry with id {%s}" % entry.file_id)
377
408
 
389
420
 
390
421
        self._byid[entry.file_id] = entry
391
422
        parent.children[entry.name] = entry
392
 
        return entry
393
423
 
394
424
 
395
425
    def add_path(self, relpath, kind, file_id=None):
396
426
        """Add entry from a path.
397
427
 
398
 
        The immediate parent must already be versioned.
399
 
 
400
 
        Returns the new entry object."""
401
 
        from bzrlib.branch import gen_file_id
 
428
        The immediate parent must already be versioned"""
 
429
        from bzrlib.errors import NotVersionedError
402
430
        
403
431
        parts = bzrlib.osutils.splitpath(relpath)
404
432
        if len(parts) == 0:
405
433
            raise BzrError("cannot re-add root of inventory")
406
434
 
407
435
        if file_id == None:
 
436
            from bzrlib.branch import gen_file_id
408
437
            file_id = gen_file_id(relpath)
409
438
 
410
439
        parent_path = parts[:-1]
422
451
 
423
452
        >>> inv = Inventory()
424
453
        >>> inv.add(InventoryEntry('123', 'foo.c', 'file', ROOT_ID))
425
 
        InventoryEntry('123', 'foo.c', kind='file', parent_id='TREE_ROOT')
426
454
        >>> '123' in inv
427
455
        True
428
456
        >>> del inv['123']
444
472
        del self[ie.parent_id].children[ie.name]
445
473
 
446
474
 
 
475
    def to_element(self):
 
476
        """Convert to XML Element"""
 
477
        from bzrlib.xml import Element
 
478
        
 
479
        e = Element('inventory')
 
480
        e.text = '\n'
 
481
        if self.root.file_id not in (None, ROOT_ID):
 
482
            e.set('file_id', self.root.file_id)
 
483
        for path, ie in self.iter_entries():
 
484
            e.append(ie.to_element())
 
485
        return e
 
486
    
 
487
 
 
488
    def from_element(cls, elt):
 
489
        """Construct from XML Element
 
490
        
 
491
        >>> inv = Inventory()
 
492
        >>> inv.add(InventoryEntry('foo.c-123981239', 'foo.c', 'file', ROOT_ID))
 
493
        >>> elt = inv.to_element()
 
494
        >>> inv2 = Inventory.from_element(elt)
 
495
        >>> inv2 == inv
 
496
        True
 
497
        """
 
498
        # XXXX: doctest doesn't run this properly under python2.3
 
499
        assert elt.tag == 'inventory'
 
500
        root_id = elt.get('file_id') or ROOT_ID
 
501
        o = cls(root_id)
 
502
        for e in elt:
 
503
            ie = InventoryEntry.from_element(e)
 
504
            if ie.parent_id == ROOT_ID:
 
505
                ie.parent_id = root_id
 
506
            o.add(ie)
 
507
        return o
 
508
        
 
509
    from_element = classmethod(from_element)
 
510
 
 
511
 
447
512
    def __eq__(self, other):
448
513
        """Compare two sets by comparing their contents.
449
514
 
452
517
        >>> i1 == i2
453
518
        True
454
519
        >>> i1.add(InventoryEntry('123', 'foo', 'file', ROOT_ID))
455
 
        InventoryEntry('123', 'foo', kind='file', parent_id='TREE_ROOT')
456
520
        >>> i1 == i2
457
521
        False
458
522
        >>> i2.add(InventoryEntry('123', 'foo', 'file', ROOT_ID))
459
 
        InventoryEntry('123', 'foo', kind='file', parent_id='TREE_ROOT')
460
523
        >>> i1 == i2
461
524
        True
462
525
        """
478
541
        raise ValueError('not hashable')
479
542
 
480
543
 
 
544
 
481
545
    def get_idpath(self, file_id):
482
546
        """Return a list of file_ids for the path to an entry.
483
547
 
576
640
 
577
641
 
578
642
 
579
 
_NAME_RE = None
 
643
_NAME_RE = re.compile(r'^[^/\\]+$')
580
644
 
581
645
def is_valid_name(name):
582
 
    global _NAME_RE
583
 
    if _NAME_RE == None:
584
 
        _NAME_RE = re.compile(r'^[^/\\]+$')
585
 
        
586
646
    return bool(_NAME_RE.match(name))