~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/inventory.py

  • Committer: aaron.bentley at utoronto
  • Date: 2005-09-04 03:32:17 UTC
  • mfrom: (974.1.52)
  • mto: (1185.3.4)
  • mto: This revision was merged to the branch mainline in revision 1178.
  • Revision ID: aaron.bentley@utoronto.ca-20050904033217-821a797652305c14
Disabled urlgrabber

Show diffs side-by-side

added added

removed removed

Lines of Context:
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
 
18
 
# TODO: Maybe also keep the full path of the entry, and the children?
19
 
# But those depend on its position within a particular inventory, and
20
 
# it would be nice not to need to hold the backpointer here.
21
 
 
22
 
# TODO: Perhaps split InventoryEntry into subclasses for files,
23
 
# directories, etc etc.
24
 
 
25
 
 
26
18
# This should really be an id randomly assigned when the tree is
27
19
# created, but it's not for now.
28
20
ROOT_ID = "TREE_ROOT"
44
36
    An InventoryEntry has the following fields, which are also
45
37
    present in the XML inventory-entry element:
46
38
 
47
 
    file_id
48
 
 
49
 
    name
50
 
        (within the parent directory)
51
 
 
52
 
    kind
53
 
        'directory' or 'file'
54
 
 
55
 
    parent_id
56
 
        file_id of the parent directory, or ROOT_ID
57
 
 
58
 
    entry_version
59
 
        the revision_id in which the name or parent of this file was
60
 
        last changed
61
 
 
62
 
    text_sha1
63
 
        sha-1 of the text of the file
64
 
        
65
 
    text_size
66
 
        size in bytes of the text of the file
67
 
        
68
 
    text_version
69
 
        the revision_id in which the text of this file was introduced
70
 
 
71
 
    (reading a version 4 tree created a text_id field.)
 
39
    * *file_id*
 
40
    * *name*: (only the basename within the directory, must not
 
41
      contain slashes)
 
42
    * *kind*: "directory" or "file"
 
43
    * *directory_id*: (if absent/null means the branch root directory)
 
44
    * *text_sha1*: only for files
 
45
    * *text_size*: in bytes, only for files 
 
46
    * *text_id*: identifier for the text version, only for files
 
47
 
 
48
    InventoryEntries can also exist inside a WorkingTree
 
49
    inventory, in which case they are not yet bound to a
 
50
    particular revision of the file.  In that case the text_sha1,
 
51
    text_size and text_id are absent.
 
52
 
72
53
 
73
54
    >>> i = Inventory()
74
55
    >>> i.path2id('')
109
90
    src/wibble/wibble.c
110
91
    >>> i.id2path('2326')
111
92
    'src/wibble/wibble.c'
 
93
 
 
94
    TODO: Maybe also keep the full path of the entry, and the children?
 
95
           But those depend on its position within a particular inventory, and
 
96
           it would be nice not to need to hold the backpointer here.
112
97
    """
113
 
    
 
98
 
 
99
    # TODO: split InventoryEntry into subclasses for files,
 
100
    # directories, etc etc.
 
101
 
114
102
    __slots__ = ['text_sha1', 'text_size', 'file_id', 'name', 'kind',
115
 
                 'text_id', 'parent_id', 'children',
116
 
                 'text_version', 'entry_version', ]
117
 
 
 
103
                 'text_id', 'parent_id', 'children', ]
118
104
 
119
105
    def __init__(self, file_id, name, kind, parent_id, text_id=None):
120
106
        """Create an InventoryEntry
131
117
        Traceback (most recent call last):
132
118
        BzrCheckError: InventoryEntry name 'src/hello.c' is invalid
133
119
        """
134
 
        assert isinstance(name, basestring), name
135
120
        if '/' in name or '\\' in name:
136
121
            raise BzrCheckError('InventoryEntry name %r is invalid' % name)
137
122
        
138
 
        self.text_version = None
139
 
        self.entry_version = None
140
123
        self.text_sha1 = None
141
124
        self.text_size = None
 
125
    
142
126
        self.file_id = file_id
143
127
        self.name = name
144
128
        self.kind = kind
161
145
 
162
146
    def copy(self):
163
147
        other = InventoryEntry(self.file_id, self.name, self.kind,
164
 
                               self.parent_id)
165
 
        other.text_id = self.text_id
 
148
                               self.parent_id, text_id=self.text_id)
166
149
        other.text_sha1 = self.text_sha1
167
150
        other.text_size = self.text_size
168
 
        other.text_version = self.text_version
169
 
        other.entry_version = self.entry_version
170
151
        # note that children are *not* copied; they're pulled across when
171
152
        # others are added
172
153
        return other
181
162
                   self.parent_id))
182
163
 
183
164
    
 
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
 
184
219
    def __eq__(self, other):
185
220
        if not isinstance(other, InventoryEntry):
186
221
            return NotImplemented
191
226
               and (self.text_size == other.text_size) \
192
227
               and (self.text_id == other.text_id) \
193
228
               and (self.parent_id == other.parent_id) \
194
 
               and (self.kind == other.kind) \
195
 
               and (self.text_version == other.text_version) \
196
 
               and (self.entry_version == other.entry_version)
 
229
               and (self.kind == other.kind)
197
230
 
198
231
 
199
232
    def __ne__(self, other):
277
310
        self._byid = {self.root.file_id: self.root}
278
311
 
279
312
 
280
 
    def copy(self):
281
 
        other = Inventory(self.root.file_id)
282
 
        # copy recursively so we know directories will be added before
283
 
        # their children.  There are more efficient ways than this...
284
 
        for path, entry in self.iter_entries():
285
 
            if entry == self.root:
286
 
                continue
287
 
            other.add(entry.copy())
288
 
        return other
289
 
 
290
 
 
291
313
    def __iter__(self):
292
314
        return iter(self._byid)
293
315
 
394
416
        """Add entry to inventory.
395
417
 
396
418
        To add  a file to a branch ready to be committed, use Branch.add,
397
 
        which calls this.
398
 
 
399
 
        Returns the new entry object.
400
 
        """
 
419
        which calls this."""
401
420
        if entry.file_id in self._byid:
402
421
            raise BzrError("inventory already contains entry with id {%s}" % entry.file_id)
403
422
 
421
440
    def add_path(self, relpath, kind, file_id=None):
422
441
        """Add entry from a path.
423
442
 
424
 
        The immediate parent must already be versioned.
425
 
 
426
 
        Returns the new entry object."""
 
443
        The immediate parent must already be versioned"""
427
444
        from bzrlib.branch import gen_file_id
428
445
        
429
446
        parts = bzrlib.osutils.splitpath(relpath)
470
487
        del self[ie.parent_id].children[ie.name]
471
488
 
472
489
 
 
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
 
473
528
    def __eq__(self, other):
474
529
        """Compare two sets by comparing their contents.
475
530
 
497
552
 
498
553
 
499
554
    def __ne__(self, other):
500
 
        return not self.__eq__(other)
 
555
        return not (self == other)
501
556
 
502
557
 
503
558
    def __hash__(self):
504
559
        raise ValueError('not hashable')
505
560
 
506
561
 
 
562
 
507
563
    def get_idpath(self, file_id):
508
564
        """Return a list of file_ids for the path to an entry.
509
565