← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/bundle/bundle_data.py

  • Committer: Vincent Ladeuil
  • Date: 2011-12-16 16:38:33 UTC
  • mto: This revision was merged to the branch mainline in revision 6387.
  • Revision ID: v.ladeuil+lp@free.fr-20111216163833-4igwmwi1dmxbbebw
Migrate add.maximum_file_size to the new config scheme

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/bundle/bundle_data.py
16
16
 
17
17
"""Read in a bundle stream, and process it into a BundleReader object."""
18
18
 
19
 
from __future__ import absolute_import
20
 
 
21
19
import base64
22
20
from cStringIO import StringIO
23
21
import os
576
574
            return None
577
575
        return new_path
578
576
 
579
 
    def get_root_id(self):
580
 
        return self.path2id('')
581
 
 
582
577
    def path2id(self, path):
583
578
        """Return the id of the file present at path in the target tree."""
584
579
        file_id = self._new_id.get(path)
589
584
            return None
590
585
        if old_path in self.deleted:
591
586
            return None
592
 
        return self.base_tree.path2id(old_path)
 
587
        if getattr(self.base_tree, 'path2id', None) is not None:
 
588
            return self.base_tree.path2id(old_path)
 
589
        else:
 
590
            return self.base_tree.inventory.path2id(old_path)
593
591
 
594
592
    def id2path(self, file_id):
595
593
        """Return the new path in the target tree of the file with id file_id"""
625
623
        """
626
624
        base_id = self.old_contents_id(file_id)
627
625
        if (base_id is not None and
628
 
            base_id != self.base_tree.get_root_id()):
 
626
            base_id != self.base_tree.inventory.root.file_id):
629
627
            patch_original = self.base_tree.get_file(base_id)
630
628
        else:
631
629
            patch_original = None
632
630
        file_patch = self.patches.get(self.id2path(file_id))
633
631
        if file_patch is None:
634
632
            if (patch_original is None and
635
 
                self.kind(file_id) == 'directory'):
 
633
                self.get_kind(file_id) == 'directory'):
636
634
                return StringIO()
637
635
            if patch_original is None:
638
636
                raise AssertionError("None: %s" % file_id)
651
649
        except KeyError:
652
650
            return self.base_tree.get_symlink_target(file_id)
653
651
 
654
 
    def kind(self, file_id):
 
652
    def get_kind(self, file_id):
655
653
        if file_id in self._kinds:
656
654
            return self._kinds[file_id]
657
 
        return self.base_tree.kind(file_id)
658
 
 
659
 
    def get_file_revision(self, file_id):
660
 
        path = self.id2path(file_id)
661
 
        if path in self._last_changed:
662
 
            return self._last_changed[path]
663
 
        else:
664
 
            return self.base_tree.get_file_revision(file_id)
 
655
        return self.base_tree.inventory[file_id].kind
665
656
 
666
657
    def is_executable(self, file_id):
667
658
        path = self.id2path(file_id)
668
659
        if path in self._executable:
669
660
            return self._executable[path]
670
661
        else:
671
 
            return self.base_tree.is_executable(file_id)
 
662
            return self.base_tree.inventory[file_id].executable
672
663
 
673
664
    def get_last_changed(self, file_id):
674
665
        path = self.id2path(file_id)
687
678
        if new_path not in self.patches:
688
679
            # If the entry does not have a patch, then the
689
680
            # contents must be the same as in the base_tree
690
 
            text_size = self.base_tree.get_file_size(file_id)
691
 
            text_sha1 = self.base_tree.get_file_sha1(file_id)
692
 
            return text_size, text_sha1
 
681
            ie = self.base_tree.inventory[file_id]
 
682
            if ie.text_size is None:
 
683
                return ie.text_size, ie.text_sha1
 
684
            return int(ie.text_size), ie.text_sha1
693
685
        fileobj = self.get_file(file_id)
694
686
        content = fileobj.read()
695
687
        return len(content), sha_string(content)
700
692
        This need to be called before ever accessing self.inventory
701
693
        """
702
694
        from os.path import dirname, basename
 
695
        base_inv = self.base_tree.inventory
703
696
        inv = Inventory(None, self.revision_id)
704
697
 
705
698
        def add_entry(file_id):
712
705
                parent_path = dirname(path)
713
706
                parent_id = self.path2id(parent_path)
714
707
 
715
 
            kind = self.kind(file_id)
 
708
            kind = self.get_kind(file_id)
716
709
            revision_id = self.get_last_changed(file_id)
717
710
 
718
711
            name = basename(path)
746
739
    # at that instant
747
740
    inventory = property(_get_inventory)
748
741
 
749
 
    root_inventory = property(_get_inventory)
750
 
 
751
 
    def all_file_ids(self):
752
 
        return set(
753
 
            [entry.file_id for path, entry in self.inventory.iter_entries()])
 
742
    def __iter__(self):
 
743
        for path, entry in self.inventory.iter_entries():
 
744
            yield entry.file_id
754
745
 
755
746
    def list_files(self, include_root=False, from_dir=None, recursive=True):
756
747
        # The only files returned by this are those from the version