← 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-09-05 08:00:35 UTC
  • Revision ID: mbp@sourcefrog.net-20050905080035-e0439293f8b6b9f9
- start splitting code for xml (de)serialization away from objects
  preparatory to supporting multiple formats by a single library

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/inventory.py
162
162
                   self.parent_id))
163
163
 
164
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
 
 
219
165
    def __eq__(self, other):
220
166
        if not isinstance(other, InventoryEntry):
221
167
            return NotImplemented
416
362
        """Add entry to inventory.
417
363
 
418
364
        To add  a file to a branch ready to be committed, use Branch.add,
419
 
        which calls this."""
 
365
        which calls this.
 
366
 
 
367
        Returns the new entry object.
 
368
        """
420
369
        if entry.file_id in self._byid:
421
370
            raise BzrError("inventory already contains entry with id {%s}" % entry.file_id)
422
371
 
440
389
    def add_path(self, relpath, kind, file_id=None):
441
390
        """Add entry from a path.
442
391
 
443
 
        The immediate parent must already be versioned"""
 
392
        The immediate parent must already be versioned.
 
393
 
 
394
        Returns the new entry object."""
444
395
        from bzrlib.branch import gen_file_id
445
396
        
446
397
        parts = bzrlib.osutils.splitpath(relpath)
487
438
        del self[ie.parent_id].children[ie.name]
488
439
 
489
440
 
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
441
    def __eq__(self, other):
529
442
        """Compare two sets by comparing their contents.
530
443
 
559
472
        raise ValueError('not hashable')
560
473
 
561
474
 
562
 
 
563
475
    def get_idpath(self, file_id):
564
476
        """Return a list of file_ids for the path to an entry.
565
477