← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/branch.py

  • Committer: Martin Pool
  • Date: 2005-06-06 04:47:33 UTC
  • Revision ID: mbp@sourcefrog.net-20050606044733-e902b05ac1747cd2
- fix invocation of testbzr when giving explicit bzr location

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/branch.py
26
26
from inventory import InventoryEntry, Inventory
27
27
from osutils import isdir, quotefn, isfile, uuid, sha_file, username, \
28
28
     format_date, compact_date, pumpfile, user_email, rand_bytes, splitpath, \
29
 
     joinpath, sha_file, sha_string, file_kind, local_time_offset, appendpath
 
29
     joinpath, sha_string, file_kind, local_time_offset, appendpath
30
30
from store import ImmutableStore
31
31
from revision import Revision
32
32
from errors import BzrError
104
104
            raise BzrError('%r is not in a branch' % orig_f)
105
105
        f = head
106
106
    
107
 
class DivergedBranches(Exception):
108
 
    def __init__(self, branch1, branch2):
109
 
        self.branch1 = branch1
110
 
        self.branch2 = branch2
111
 
        Exception.__init__(self, "These branches have diverged.")
112
 
 
113
 
 
114
 
class NoSuchRevision(BzrError):
115
 
    def __init__(self, branch, revision):
116
 
        self.branch = branch
117
 
        self.revision = revision
118
 
        msg = "Branch %s has no revision %d" % (branch, revision)
119
 
        BzrError.__init__(self, msg)
120
107
 
121
108
 
122
109
######################################################################
285
272
        os.mkdir(self.controlfilename([]))
286
273
        self.controlfile('README', 'w').write(
287
274
            "This is a Bazaar-NG control directory.\n"
288
 
            "Do not change any files in this directory.\n")
 
275
            "Do not change any files in this directory.")
289
276
        self.controlfile('branch-format', 'w').write(BZR_BRANCH_FORMAT)
290
277
        for d in ('text-store', 'inventory-store', 'revision-store'):
291
278
            os.mkdir(self.controlfilename(d))
541
528
 
542
529
    def get_revision(self, revision_id):
543
530
        """Return the Revision object for a named revision"""
544
 
        if not revision_id or not isinstance(revision_id, basestring):
545
 
            raise ValueError('invalid revision-id: %r' % revision_id)
546
531
        r = Revision.read_xml(self.revision_store[revision_id])
547
532
        assert r.revision_id == revision_id
548
533
        return r
549
534
 
550
 
    def get_revision_sha1(self, revision_id):
551
 
        """Hash the stored value of a revision, and return it."""
552
 
        # In the future, revision entries will be signed. At that
553
 
        # point, it is probably best *not* to include the signature
554
 
        # in the revision hash. Because that lets you re-sign
555
 
        # the revision, (add signatures/remove signatures) and still
556
 
        # have all hash pointers stay consistent.
557
 
        # But for now, just hash the contents.
558
 
        return sha_file(self.revision_store[revision_id])
559
 
 
560
535
 
561
536
    def get_inventory(self, inventory_id):
562
537
        """Get Inventory object by hash.
567
542
        i = Inventory.read_xml(self.inventory_store[inventory_id])
568
543
        return i
569
544
 
570
 
    def get_inventory_sha1(self, inventory_id):
571
 
        """Return the sha1 hash of the inventory entry
572
 
        """
573
 
        return sha_file(self.inventory_store[inventory_id])
574
 
 
575
545
 
576
546
    def get_revision_inventory(self, revision_id):
577
547
        """Return inventory of a past revision."""
680
650
            return None
681
651
 
682
652
 
683
 
    def missing_revisions(self, other, stop_revision=None):
684
 
        """
685
 
        If self and other have not diverged, return a list of the revisions
686
 
        present in other, but missing from self.
687
 
 
688
 
        >>> from bzrlib.commit import commit
689
 
        >>> bzrlib.trace.silent = True
690
 
        >>> br1 = ScratchBranch()
691
 
        >>> br2 = ScratchBranch()
692
 
        >>> br1.missing_revisions(br2)
693
 
        []
694
 
        >>> commit(br2, "lala!", rev_id="REVISION-ID-1")
695
 
        >>> br1.missing_revisions(br2)
696
 
        [u'REVISION-ID-1']
697
 
        >>> br2.missing_revisions(br1)
698
 
        []
699
 
        >>> commit(br1, "lala!", rev_id="REVISION-ID-1")
700
 
        >>> br1.missing_revisions(br2)
701
 
        []
702
 
        >>> commit(br2, "lala!", rev_id="REVISION-ID-2A")
703
 
        >>> br1.missing_revisions(br2)
704
 
        [u'REVISION-ID-2A']
705
 
        >>> commit(br1, "lala!", rev_id="REVISION-ID-2B")
706
 
        >>> br1.missing_revisions(br2)
707
 
        Traceback (most recent call last):
708
 
        DivergedBranches: These branches have diverged.
709
 
        """
710
 
        self_history = self.revision_history()
711
 
        self_len = len(self_history)
712
 
        other_history = other.revision_history()
713
 
        other_len = len(other_history)
714
 
        common_index = min(self_len, other_len) -1
715
 
        if common_index >= 0 and \
716
 
            self_history[common_index] != other_history[common_index]:
717
 
            raise DivergedBranches(self, other)
718
 
 
719
 
        if stop_revision is None:
720
 
            stop_revision = other_len
721
 
        elif stop_revision > other_len:
722
 
            raise NoSuchRevision(self, stop_revision)
723
 
        
724
 
        return other_history[self_len:stop_revision]
725
 
 
726
 
 
727
 
    def update_revisions(self, other, stop_revision=None):
728
 
        """Pull in all new revisions from other branch.
729
 
        
730
 
        >>> from bzrlib.commit import commit
731
 
        >>> bzrlib.trace.silent = True
732
 
        >>> br1 = ScratchBranch(files=['foo', 'bar'])
733
 
        >>> br1.add('foo')
734
 
        >>> br1.add('bar')
735
 
        >>> commit(br1, "lala!", rev_id="REVISION-ID-1", verbose=False)
736
 
        >>> br2 = ScratchBranch()
737
 
        >>> br2.update_revisions(br1)
738
 
        Added 2 texts.
739
 
        Added 1 inventories.
740
 
        Added 1 revisions.
741
 
        >>> br2.revision_history()
742
 
        [u'REVISION-ID-1']
743
 
        >>> br2.update_revisions(br1)
744
 
        Added 0 texts.
745
 
        Added 0 inventories.
746
 
        Added 0 revisions.
747
 
        >>> br1.text_store.total_size() == br2.text_store.total_size()
748
 
        True
749
 
        """
750
 
        from bzrlib.progress import ProgressBar
751
 
 
752
 
        pb = ProgressBar()
753
 
 
754
 
        pb.update('comparing histories')
755
 
        revision_ids = self.missing_revisions(other, stop_revision)
756
 
        revisions = []
757
 
        needed_texts = sets.Set()
758
 
        i = 0
759
 
        for rev_id in revision_ids:
760
 
            i += 1
761
 
            pb.update('fetching revision', i, len(revision_ids))
762
 
            rev = other.get_revision(rev_id)
763
 
            revisions.append(rev)
764
 
            inv = other.get_inventory(str(rev.inventory_id))
765
 
            for key, entry in inv.iter_entries():
766
 
                if entry.text_id is None:
767
 
                    continue
768
 
                if entry.text_id not in self.text_store:
769
 
                    needed_texts.add(entry.text_id)
770
 
 
771
 
        pb.clear()
772
 
                    
773
 
        count = self.text_store.copy_multi(other.text_store, needed_texts)
774
 
        print "Added %d texts." % count 
775
 
        inventory_ids = [ f.inventory_id for f in revisions ]
776
 
        count = self.inventory_store.copy_multi(other.inventory_store, 
777
 
                                                inventory_ids)
778
 
        print "Added %d inventories." % count 
779
 
        revision_ids = [ f.revision_id for f in revisions]
780
 
        count = self.revision_store.copy_multi(other.revision_store, 
781
 
                                               revision_ids)
782
 
        for revision_id in revision_ids:
783
 
            self.append_revision(revision_id)
784
 
        print "Added %d revisions." % count
785
 
                    
786
 
        
787
653
    def commit(self, *args, **kw):
788
654
        """Deprecated"""
789
655
        from bzrlib.commit import commit