← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/inventory.py

  • Committer: Robert Collins
  • Date: 2005-09-13 09:39:26 UTC
  • mto: This revision was merged to the branch mainline in revision 1397.
  • Revision ID: robertc@robertcollins.net-20050913093926-7edba69aff28352d
bugfix symlink support - read the link from the abspath not relative path

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/inventory.py
100
100
    # directories, etc etc.
101
101
 
102
102
    __slots__ = ['text_sha1', 'text_size', 'file_id', 'name', 'kind',
103
 
                 'text_id', 'parent_id', 'children', ]
 
103
                 'text_id', 'parent_id', 'children',
 
104
                 'text_version', 'entry_version', 'symlink_target']
 
105
 
104
106
 
105
107
    def __init__(self, file_id, name, kind, parent_id, text_id=None):
106
108
        """Create an InventoryEntry
117
119
        Traceback (most recent call last):
118
120
        BzrCheckError: InventoryEntry name 'src/hello.c' is invalid
119
121
        """
 
122
        assert isinstance(name, basestring), name
120
123
        if '/' in name or '\\' in name:
121
124
            raise BzrCheckError('InventoryEntry name %r is invalid' % name)
122
125
        
 
126
        self.text_version = None
 
127
        self.entry_version = None
123
128
        self.text_sha1 = None
124
129
        self.text_size = None
125
 
    
126
130
        self.file_id = file_id
127
131
        self.name = name
128
132
        self.kind = kind
129
133
        self.text_id = text_id
130
134
        self.parent_id = parent_id
 
135
        self.symlink_target = None
131
136
        if kind == 'directory':
132
137
            self.children = {}
133
138
        elif kind == 'file':
134
139
            pass
 
140
        elif kind == 'symlink':
 
141
            pass
135
142
        else:
136
143
            raise BzrError("unhandled entry kind %r" % kind)
137
144
 
138
 
 
 
145
    def read_symlink_target(self, path):
 
146
        if self.kind == 'symlink':
 
147
            try:
 
148
                self.symlink_target = os.readlink(path)
 
149
            except OSError,e:
 
150
                raise BzrError("os.readlink error, %s" % e)
139
151
 
140
152
    def sorted_children(self):
141
153
        l = self.children.items()
148
160
                               self.parent_id, text_id=self.text_id)
149
161
        other.text_sha1 = self.text_sha1
150
162
        other.text_size = self.text_size
 
163
        other.symlink_target = self.symlink_target
151
164
        # note that children are *not* copied; they're pulled across when
152
165
        # others are added
153
166
        return other
162
175
                   self.parent_id))
163
176
 
164
177
    
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
 
 
219
178
    def __eq__(self, other):
220
179
        if not isinstance(other, InventoryEntry):
221
180
            return NotImplemented
222
181
 
223
182
        return (self.file_id == other.file_id) \
224
183
               and (self.name == other.name) \
 
184
               and (other.symlink_target == self.symlink_target) \
225
185
               and (self.text_sha1 == other.text_sha1) \
226
186
               and (self.text_size == other.text_size) \
227
187
               and (self.text_id == other.text_id) \
228
188
               and (self.parent_id == other.parent_id) \
229
 
               and (self.kind == other.kind)
230
 
 
 
189
               and (self.kind == other.kind) \
 
190
               and (self.text_version == other.text_version) \
 
191
               and (self.entry_version == other.entry_version)
231
192
 
232
193
    def __ne__(self, other):
233
194
        return not (self == other)
416
377
        """Add entry to inventory.
417
378
 
418
379
        To add  a file to a branch ready to be committed, use Branch.add,
419
 
        which calls this."""
 
380
        which calls this.
 
381
 
 
382
        Returns the new entry object.
 
383
        """
420
384
        if entry.file_id in self._byid:
421
385
            raise BzrError("inventory already contains entry with id {%s}" % entry.file_id)
422
386
 
440
404
    def add_path(self, relpath, kind, file_id=None):
441
405
        """Add entry from a path.
442
406
 
443
 
        The immediate parent must already be versioned"""
 
407
        The immediate parent must already be versioned.
 
408
 
 
409
        Returns the new entry object."""
444
410
        from bzrlib.branch import gen_file_id
445
411
        
446
412
        parts = bzrlib.osutils.splitpath(relpath)
487
453
        del self[ie.parent_id].children[ie.name]
488
454
 
489
455
 
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
 
 
528
456
    def __eq__(self, other):
529
457
        """Compare two sets by comparing their contents.
530
458
 
559
487
        raise ValueError('not hashable')
560
488
 
561
489
 
562
 
 
563
490
    def get_idpath(self, file_id):
564
491
        """Return a list of file_ids for the path to an entry.
565
492