~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/branch.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2011-06-30 18:28:17 UTC
  • mfrom: (5967.10.2 test-cat)
  • Revision ID: pqm@pqm.ubuntu.com-20110630182817-83a5q9r9rxfkdn8r
(mbp) don't use subprocesses for testing cat (Martin Pool)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2012 Canonical Ltd
 
1
# Copyright (C) 2005-2011 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
 
from __future__ import absolute_import
18
 
 
19
 
import bzrlib.bzrdir
20
17
 
21
18
from cStringIO import StringIO
 
19
import sys
22
20
 
23
21
from bzrlib.lazy_import import lazy_import
24
22
lazy_import(globals(), """
25
 
import itertools
 
23
from itertools import chain
26
24
from bzrlib import (
27
 
    bzrdir,
28
 
    controldir,
29
 
    cache_utf8,
30
 
    cleanup,
31
 
    config as _mod_config,
32
 
    debug,
33
 
    errors,
34
 
    fetch,
35
 
    graph as _mod_graph,
36
 
    lockdir,
37
 
    lockable_files,
38
 
    remote,
39
 
    repository,
40
 
    revision as _mod_revision,
41
 
    rio,
42
 
    shelf,
43
 
    tag as _mod_tag,
44
 
    transport,
45
 
    ui,
46
 
    urlutils,
47
 
    vf_search,
 
25
        bzrdir,
 
26
        cache_utf8,
 
27
        cleanup,
 
28
        config as _mod_config,
 
29
        debug,
 
30
        errors,
 
31
        fetch,
 
32
        graph as _mod_graph,
 
33
        lockdir,
 
34
        lockable_files,
 
35
        remote,
 
36
        repository,
 
37
        revision as _mod_revision,
 
38
        rio,
 
39
        transport,
 
40
        ui,
 
41
        urlutils,
 
42
        )
 
43
from bzrlib.config import BranchConfig, TransportConfig
 
44
from bzrlib.tag import (
 
45
    BasicTags,
 
46
    DisabledTags,
48
47
    )
49
 
from bzrlib.i18n import gettext, ngettext
50
48
""")
51
49
 
52
 
# Explicitly import bzrlib.bzrdir so that the BzrProber
53
 
# is guaranteed to be registered.
54
 
import bzrlib.bzrdir
55
 
 
56
50
from bzrlib import (
57
 
    bzrdir,
58
51
    controldir,
59
52
    )
60
53
from bzrlib.decorators import (
73
66
from bzrlib.trace import mutter, mutter_callsite, note, is_quiet
74
67
 
75
68
 
 
69
BZR_BRANCH_FORMAT_4 = "Bazaar-NG branch, format 0.0.4\n"
 
70
BZR_BRANCH_FORMAT_5 = "Bazaar-NG branch, format 5\n"
 
71
BZR_BRANCH_FORMAT_6 = "Bazaar Branch Format 6 (bzr 0.15)\n"
 
72
 
 
73
 
76
74
class Branch(controldir.ControlComponent):
77
75
    """Branch holding a history of revisions.
78
76
 
95
93
    def user_transport(self):
96
94
        return self.bzrdir.user_transport
97
95
 
98
 
    def __init__(self, possible_transports=None):
 
96
    def __init__(self, *ignored, **ignored_too):
99
97
        self.tags = self._format.make_tags(self)
100
98
        self._revision_history_cache = None
101
99
        self._revision_id_to_revno_cache = None
105
103
        self._last_revision_info_cache = None
106
104
        self._master_branch_cache = None
107
105
        self._merge_sorted_revisions_cache = None
108
 
        self._open_hook(possible_transports)
 
106
        self._open_hook()
109
107
        hooks = Branch.hooks['open']
110
108
        for hook in hooks:
111
109
            hook(self)
112
110
 
113
 
    def _open_hook(self, possible_transports):
 
111
    def _open_hook(self):
114
112
        """Called by init to allow simpler extension of the base class."""
115
113
 
116
 
    def _activate_fallback_location(self, url, possible_transports):
 
114
    def _activate_fallback_location(self, url):
117
115
        """Activate the branch/repository from url as a fallback repository."""
118
116
        for existing_fallback_repo in self.repository._fallback_repositories:
119
117
            if existing_fallback_repo.user_url == url:
120
118
                # This fallback is already configured.  This probably only
121
 
                # happens because ControlDir.sprout is a horrible mess.  To avoid
 
119
                # happens because BzrDir.sprout is a horrible mess.  To avoid
122
120
                # confusing _unstack we don't add this a second time.
123
121
                mutter('duplicate activation of fallback %r on %r', url, self)
124
122
                return
125
 
        repo = self._get_fallback_repository(url, possible_transports)
 
123
        repo = self._get_fallback_repository(url)
126
124
        if repo.has_same_location(self.repository):
127
125
            raise errors.UnstackableLocationError(self.user_url, url)
128
126
        self.repository.add_fallback_repository(repo)
182
180
        For instance, if the branch is at URL/.bzr/branch,
183
181
        Branch.open(URL) -> a Branch instance.
184
182
        """
185
 
        control = controldir.ControlDir.open(base,
186
 
            possible_transports=possible_transports, _unsupported=_unsupported)
187
 
        return control.open_branch(unsupported=_unsupported,
188
 
            possible_transports=possible_transports)
 
183
        control = bzrdir.BzrDir.open(base, _unsupported,
 
184
                                     possible_transports=possible_transports)
 
185
        return control.open_branch(unsupported=_unsupported)
189
186
 
190
187
    @staticmethod
191
 
    def open_from_transport(transport, name=None, _unsupported=False,
192
 
            possible_transports=None):
 
188
    def open_from_transport(transport, name=None, _unsupported=False):
193
189
        """Open the branch rooted at transport"""
194
 
        control = controldir.ControlDir.open_from_transport(transport, _unsupported)
195
 
        return control.open_branch(name=name, unsupported=_unsupported,
196
 
            possible_transports=possible_transports)
 
190
        control = bzrdir.BzrDir.open_from_transport(transport, _unsupported)
 
191
        return control.open_branch(name=name, unsupported=_unsupported)
197
192
 
198
193
    @staticmethod
199
194
    def open_containing(url, possible_transports=None):
207
202
        format, UnknownFormatError or UnsupportedFormatError are raised.
208
203
        If there is one, it is returned, along with the unused portion of url.
209
204
        """
210
 
        control, relpath = controldir.ControlDir.open_containing(url,
 
205
        control, relpath = bzrdir.BzrDir.open_containing(url,
211
206
                                                         possible_transports)
212
 
        branch = control.open_branch(possible_transports=possible_transports)
213
 
        return (branch, relpath)
 
207
        return control.open_branch(), relpath
214
208
 
215
209
    def _push_should_merge_tags(self):
216
210
        """Should _basic_push merge this branch's tags into the target?
228
222
 
229
223
        :return: A bzrlib.config.BranchConfig.
230
224
        """
231
 
        return _mod_config.BranchConfig(self)
232
 
 
233
 
    def get_config_stack(self):
234
 
        """Get a bzrlib.config.BranchStack for this Branch.
235
 
 
236
 
        This can then be used to get and set configuration options for the
237
 
        branch.
238
 
 
239
 
        :return: A bzrlib.config.BranchStack.
240
 
        """
241
 
        return _mod_config.BranchStack(self)
 
225
        return BranchConfig(self)
242
226
 
243
227
    def _get_config(self):
244
228
        """Get the concrete config for just the config in this branch.
252
236
        """
253
237
        raise NotImplementedError(self._get_config)
254
238
 
255
 
    def store_uncommitted(self, creator):
256
 
        """Store uncommitted changes from a ShelfCreator.
257
 
 
258
 
        :param creator: The ShelfCreator containing uncommitted changes, or
259
 
            None to delete any stored changes.
260
 
        :raises: ChangesAlreadyStored if the branch already has changes.
261
 
        """
262
 
        raise NotImplementedError(self.store_uncommitted)
263
 
 
264
 
    def get_unshelver(self, tree):
265
 
        """Return a shelf.Unshelver for this branch and tree.
266
 
 
267
 
        :param tree: The tree to use to construct the Unshelver.
268
 
        :return: an Unshelver or None if no changes are stored.
269
 
        """
270
 
        raise NotImplementedError(self.get_unshelver)
271
 
 
272
 
    def _get_fallback_repository(self, url, possible_transports):
 
239
    def _get_fallback_repository(self, url):
273
240
        """Get the repository we fallback to at url."""
274
241
        url = urlutils.join(self.base, url)
275
 
        a_branch = Branch.open(url, possible_transports=possible_transports)
 
242
        a_branch = Branch.open(url,
 
243
            possible_transports=[self.bzrdir.root_transport])
276
244
        return a_branch.repository
277
245
 
278
246
    @needs_read_lock
552
520
                    # The decision to include the start or not
553
521
                    # depends on the stop_rule if a stop is provided
554
522
                    # so pop this node back into the iterator
555
 
                    rev_iter = itertools.chain(iter([node]), rev_iter)
 
523
                    rev_iter = chain(iter([node]), rev_iter)
556
524
                    break
557
525
        if stop_revision_id is None:
558
526
            # Yield everything
683
651
        """
684
652
        raise errors.UpgradeRequired(self.user_url)
685
653
 
686
 
    def get_append_revisions_only(self):
687
 
        """Whether it is only possible to append revisions to the history.
688
 
        """
689
 
        if not self._format.supports_set_append_revisions_only():
690
 
            return False
691
 
        return self.get_config_stack().get('append_revisions_only')
692
 
 
693
654
    def set_append_revisions_only(self, enabled):
694
655
        if not self._format.supports_set_append_revisions_only():
695
656
            raise errors.UpgradeRequired(self.user_url)
696
 
        self.get_config_stack().set('append_revisions_only', enabled)
 
657
        if enabled:
 
658
            value = 'True'
 
659
        else:
 
660
            value = 'False'
 
661
        self.get_config().set_user_option('append_revisions_only', value,
 
662
            warn_masked=True)
697
663
 
698
664
    def set_reference_info(self, file_id, tree_path, branch_location):
699
665
        """Set the branch location to use for a tree reference."""
728
694
        """
729
695
        raise errors.UpgradeRequired(self.user_url)
730
696
 
731
 
    def get_commit_builder(self, parents, config_stack=None, timestamp=None,
 
697
    def get_commit_builder(self, parents, config=None, timestamp=None,
732
698
                           timezone=None, committer=None, revprops=None,
733
699
                           revision_id=None, lossy=False):
734
700
        """Obtain a CommitBuilder for this branch.
744
710
            represented, when pushing to a foreign VCS 
745
711
        """
746
712
 
747
 
        if config_stack is None:
748
 
            config_stack = self.get_config_stack()
 
713
        if config is None:
 
714
            config = self.get_config()
749
715
 
750
 
        return self.repository.get_commit_builder(self, parents, config_stack,
 
716
        return self.repository.get_commit_builder(self, parents, config,
751
717
            timestamp, timezone, committer, revprops, revision_id,
752
718
            lossy)
753
719
 
758
724
        """
759
725
        return None
760
726
 
761
 
    @deprecated_method(deprecated_in((2, 5, 0)))
762
727
    def get_revision_delta(self, revno):
763
728
        """Return the delta for one revision.
764
729
 
765
730
        The delta is relative to its mainline predecessor, or the
766
731
        empty tree for revision 1.
767
732
        """
768
 
        try:
769
 
            revid = self.get_rev_id(revno)
770
 
        except errors.NoSuchRevision:
 
733
        rh = self.revision_history()
 
734
        if not (1 <= revno <= len(rh)):
771
735
            raise errors.InvalidRevisionNumber(revno)
772
 
        return self.repository.get_revision_delta(revid)
 
736
        return self.repository.get_revision_delta(rh[revno-1])
773
737
 
774
738
    def get_stacked_on_url(self):
775
739
        """Get the URL this branch is stacked against.
784
748
        """Print `file` to stdout."""
785
749
        raise NotImplementedError(self.print_file)
786
750
 
 
751
    @deprecated_method(deprecated_in((2, 4, 0)))
 
752
    def set_revision_history(self, rev_history):
 
753
        """See Branch.set_revision_history."""
 
754
        self._set_revision_history(rev_history)
 
755
 
 
756
    @needs_write_lock
 
757
    def _set_revision_history(self, rev_history):
 
758
        if len(rev_history) == 0:
 
759
            revid = _mod_revision.NULL_REVISION
 
760
        else:
 
761
            revid = rev_history[-1]
 
762
        if rev_history != self._lefthand_history(revid):
 
763
            raise errors.NotLefthandHistory(rev_history)
 
764
        self.set_last_revision_info(len(rev_history), revid)
 
765
        self._cache_revision_history(rev_history)
 
766
        for hook in Branch.hooks['set_rh']:
 
767
            hook(self, rev_history)
 
768
 
787
769
    @needs_write_lock
788
770
    def set_last_revision_info(self, revno, revision_id):
789
771
        """Set the last revision of this branch.
803
785
                                  other_branch=None):
804
786
        """See Branch.generate_revision_history"""
805
787
        graph = self.repository.get_graph()
806
 
        (last_revno, last_revid) = self.last_revision_info()
807
788
        known_revision_ids = [
808
 
            (last_revid, last_revno),
 
789
            self.last_revision_info(),
809
790
            (_mod_revision.NULL_REVISION, 0),
810
791
            ]
811
792
        if last_rev is not None:
856
837
                return
857
838
            self._unstack()
858
839
        else:
859
 
            self._activate_fallback_location(url,
860
 
                possible_transports=[self.bzrdir.root_transport])
 
840
            self._activate_fallback_location(url)
861
841
        # write this out after the repository is stacked to avoid setting a
862
842
        # stacked config that doesn't work.
863
843
        self._set_config_location('stacked_on_location', url)
869
849
        """
870
850
        pb = ui.ui_factory.nested_progress_bar()
871
851
        try:
872
 
            pb.update(gettext("Unstacking"))
 
852
            pb.update("Unstacking")
873
853
            # The basic approach here is to fetch the tip of the branch,
874
854
            # including all available ghosts, from the existing stacked
875
855
            # repository into a new repository object without the fallbacks. 
889
869
            # stream from one of them to the other.  This does mean doing
890
870
            # separate SSH connection setup, but unstacking is not a
891
871
            # common operation so it's tolerable.
892
 
            new_bzrdir = controldir.ControlDir.open(
893
 
                self.bzrdir.root_transport.base)
 
872
            new_bzrdir = bzrdir.BzrDir.open(self.bzrdir.root_transport.base)
894
873
            new_repository = new_bzrdir.find_repository()
895
874
            if new_repository._fallback_repositories:
896
875
                raise AssertionError("didn't expect %r to have "
939
918
                    tags_to_fetch = set(self.tags.get_reverse_tag_dict())
940
919
                except errors.TagsNotSupported:
941
920
                    tags_to_fetch = set()
942
 
                fetch_spec = vf_search.NotInOtherForRevs(self.repository,
 
921
                fetch_spec = _mod_graph.NotInOtherForRevs(self.repository,
943
922
                    old_repository, required_ids=[self.last_revision()],
944
923
                    if_present_ids=tags_to_fetch, find_ghosts=True).execute()
945
924
                self.repository.fetch(old_repository, fetch_spec=fetch_spec)
986
965
        This means the next call to revision_history will need to call
987
966
        _gen_revision_history.
988
967
 
989
 
        This API is semi-public; it is only for use by subclasses, all other
990
 
        code should consider it to be private.
 
968
        This API is semi-public; it only for use by subclasses, all other code
 
969
        should consider it to be private.
991
970
        """
992
971
        self._revision_history_cache = None
993
972
        self._revision_id_to_revno_cache = None
1013
992
        """
1014
993
        raise NotImplementedError(self._gen_revision_history)
1015
994
 
1016
 
    def _revision_history(self):
 
995
    @needs_read_lock
 
996
    def revision_history(self):
 
997
        """Return sequence of revision ids on this branch.
 
998
 
 
999
        This method will cache the revision history for as long as it is safe to
 
1000
        do so.
 
1001
        """
1017
1002
        if 'evil' in debug.debug_flags:
1018
1003
            mutter_callsite(3, "revision_history scales with history.")
1019
1004
        if self._revision_history_cache is not None:
1052
1037
    def _read_last_revision_info(self):
1053
1038
        raise NotImplementedError(self._read_last_revision_info)
1054
1039
 
 
1040
    @deprecated_method(deprecated_in((2, 4, 0)))
 
1041
    def import_last_revision_info(self, source_repo, revno, revid):
 
1042
        """Set the last revision info, importing from another repo if necessary.
 
1043
 
 
1044
        :param source_repo: Source repository to optionally fetch from
 
1045
        :param revno: Revision number of the new tip
 
1046
        :param revid: Revision id of the new tip
 
1047
        """
 
1048
        if not self.repository.has_same_location(source_repo):
 
1049
            self.repository.fetch(source_repo, revision_id=revid)
 
1050
        self.set_last_revision_info(revno, revid)
 
1051
 
1055
1052
    def import_last_revision_info_and_tags(self, source, revno, revid,
1056
1053
                                           lossy=False):
1057
1054
        """Set the last revision info, importing from another repo if necessary.
1077
1074
        """Given a revision id, return its revno"""
1078
1075
        if _mod_revision.is_null(revision_id):
1079
1076
            return 0
1080
 
        history = self._revision_history()
 
1077
        history = self.revision_history()
1081
1078
        try:
1082
1079
            return history.index(revision_id) + 1
1083
1080
        except ValueError:
1148
1145
    def _set_config_location(self, name, url, config=None,
1149
1146
                             make_relative=False):
1150
1147
        if config is None:
1151
 
            config = self.get_config_stack()
 
1148
            config = self.get_config()
1152
1149
        if url is None:
1153
1150
            url = ''
1154
1151
        elif make_relative:
1155
1152
            url = urlutils.relative_url(self.base, url)
1156
 
        config.set(name, url)
 
1153
        config.set_user_option(name, url, warn_masked=True)
1157
1154
 
1158
1155
    def _get_config_location(self, name, config=None):
1159
1156
        if config is None:
1160
 
            config = self.get_config_stack()
1161
 
        location = config.get(name)
 
1157
            config = self.get_config()
 
1158
        location = config.get_user_option(name)
1162
1159
        if location == '':
1163
1160
            location = None
1164
1161
        return location
1165
1162
 
1166
1163
    def get_child_submit_format(self):
1167
1164
        """Return the preferred format of submissions to this branch."""
1168
 
        return self.get_config_stack().get('child_submit_format')
 
1165
        return self.get_config().get_user_option("child_submit_format")
1169
1166
 
1170
1167
    def get_submit_branch(self):
1171
1168
        """Return the submit location of the branch.
1174
1171
        pattern is that the user can override it by specifying a
1175
1172
        location.
1176
1173
        """
1177
 
        return self.get_config_stack().get('submit_branch')
 
1174
        return self.get_config().get_user_option('submit_branch')
1178
1175
 
1179
1176
    def set_submit_branch(self, location):
1180
1177
        """Return the submit location of the branch.
1183
1180
        pattern is that the user can override it by specifying a
1184
1181
        location.
1185
1182
        """
1186
 
        self.get_config_stack().set('submit_branch', location)
 
1183
        self.get_config().set_user_option('submit_branch', location,
 
1184
            warn_masked=True)
1187
1185
 
1188
1186
    def get_public_branch(self):
1189
1187
        """Return the public location of the branch.
1202
1200
        self._set_config_location('public_branch', location)
1203
1201
 
1204
1202
    def get_push_location(self):
1205
 
        """Return None or the location to push this branch to."""
1206
 
        return self.get_config_stack().get('push_location')
 
1203
        """Return the None or the location to push this branch to."""
 
1204
        push_loc = self.get_config().get_user_option('push_location')
 
1205
        return push_loc
1207
1206
 
1208
1207
    def set_push_location(self, location):
1209
1208
        """Set a new push location for this branch."""
1294
1293
            if repository_policy is not None:
1295
1294
                repository_policy.configure_branch(result)
1296
1295
            self.copy_content_into(result, revision_id=revision_id)
1297
 
            master_url = self.get_bound_location()
1298
 
            if master_url is None:
 
1296
            master_branch = self.get_master_branch()
 
1297
            if master_branch is None:
1299
1298
                result.set_parent(self.bzrdir.root_transport.base)
1300
1299
            else:
1301
 
                result.set_parent(master_url)
 
1300
                result.set_parent(master_branch.bzrdir.root_transport.base)
1302
1301
        finally:
1303
1302
            result.unlock()
1304
1303
        return result
1378
1377
        # TODO: We should probably also check that self.revision_history
1379
1378
        # matches the repository for older branch formats.
1380
1379
        # If looking for the code that cross-checks repository parents against
1381
 
        # the Graph.iter_lefthand_ancestry output, that is now a repository
 
1380
        # the iter_reverse_revision_history output, that is now a repository
1382
1381
        # specific check.
1383
1382
        return result
1384
1383
 
1385
 
    def _get_checkout_format(self, lightweight=False):
 
1384
    def _get_checkout_format(self):
1386
1385
        """Return the most suitable metadir for a checkout of this branch.
1387
1386
        Weaves are used if this branch's repository uses weaves.
1388
1387
        """
1434
1433
        """
1435
1434
        t = transport.get_transport(to_location)
1436
1435
        t.ensure_base()
1437
 
        format = self._get_checkout_format(lightweight=lightweight)
1438
 
        try:
 
1436
        if lightweight:
 
1437
            format = self._get_checkout_format()
1439
1438
            checkout = format.initialize_on_transport(t)
1440
 
        except errors.AlreadyControlDirError:
1441
 
            # It's fine if the control directory already exists,
1442
 
            # as long as there is no existing branch and working tree.
1443
 
            checkout = controldir.ControlDir.open_from_transport(t)
1444
 
            try:
1445
 
                checkout.open_branch()
1446
 
            except errors.NotBranchError:
1447
 
                pass
1448
 
            else:
1449
 
                raise errors.AlreadyControlDirError(t.base)
1450
 
            if checkout.control_transport.base == self.bzrdir.control_transport.base:
1451
 
                # When checking out to the same control directory,
1452
 
                # always create a lightweight checkout
1453
 
                lightweight = True
1454
 
 
1455
 
        if lightweight:
1456
 
            from_branch = checkout.set_branch_reference(target_branch=self)
 
1439
            from_branch = BranchReferenceFormat().initialize(checkout, 
 
1440
                target_branch=self)
1457
1441
        else:
1458
 
            policy = checkout.determine_repository_policy()
1459
 
            repo = policy.acquire_repository()[0]
1460
 
            checkout_branch = checkout.create_branch()
 
1442
            format = self._get_checkout_format()
 
1443
            checkout_branch = bzrdir.BzrDir.create_branch_convenience(
 
1444
                to_location, force_new_tree=False, format=format)
 
1445
            checkout = checkout_branch.bzrdir
1461
1446
            checkout_branch.bind(self)
1462
1447
            # pull up to the specified revision_id to set the initial
1463
1448
            # branch tip correctly, and seed it with history.
1464
1449
            checkout_branch.pull(self, stop_revision=revision_id)
1465
 
            from_branch = None
 
1450
            from_branch=None
1466
1451
        tree = checkout.create_workingtree(revision_id,
1467
1452
                                           from_branch=from_branch,
1468
1453
                                           accelerator_tree=accelerator_tree,
1557
1542
            heads that must be fetched if present, but no error is necessary if
1558
1543
            they are not present.
1559
1544
        """
1560
 
        # For bzr native formats must_fetch is just the tip, and
1561
 
        # if_present_fetch are the tags.
 
1545
        # For bzr native formats must_fetch is just the tip, and if_present_fetch
 
1546
        # are the tags.
1562
1547
        must_fetch = set([self.last_revision()])
1563
 
        if_present_fetch = set()
1564
 
        if self.get_config_stack().get('branch.fetch_tags'):
1565
 
            try:
1566
 
                if_present_fetch = set(self.tags.get_reverse_tag_dict())
1567
 
            except errors.TagsNotSupported:
1568
 
                pass
 
1548
        try:
 
1549
            if_present_fetch = set(self.tags.get_reverse_tag_dict())
 
1550
        except errors.TagsNotSupported:
 
1551
            if_present_fetch = set()
1569
1552
        must_fetch.discard(_mod_revision.NULL_REVISION)
1570
1553
        if_present_fetch.discard(_mod_revision.NULL_REVISION)
1571
1554
        return must_fetch, if_present_fetch
1576
1559
 
1577
1560
    Formats provide three things:
1578
1561
     * An initialization routine,
1579
 
     * a format description
 
1562
     * a format string,
1580
1563
     * an open routine.
1581
1564
 
1582
1565
    Formats are placed in an dict by their format string for reference
1589
1572
    object will be created every time regardless.
1590
1573
    """
1591
1574
 
 
1575
    can_set_append_revisions_only = True
 
1576
 
1592
1577
    def __eq__(self, other):
1593
1578
        return self.__class__ is other.__class__
1594
1579
 
1595
1580
    def __ne__(self, other):
1596
1581
        return not (self == other)
1597
1582
 
1598
 
    def get_reference(self, controldir, name=None):
1599
 
        """Get the target reference of the branch in controldir.
 
1583
    @classmethod
 
1584
    def find_format(klass, a_bzrdir, name=None):
 
1585
        """Return the format for the branch object in a_bzrdir."""
 
1586
        try:
 
1587
            transport = a_bzrdir.get_branch_transport(None, name=name)
 
1588
            format_string = transport.get_bytes("format")
 
1589
            return format_registry.get(format_string)
 
1590
        except errors.NoSuchFile:
 
1591
            raise errors.NotBranchError(path=transport.base, bzrdir=a_bzrdir)
 
1592
        except KeyError:
 
1593
            raise errors.UnknownFormatError(format=format_string, kind='branch')
 
1594
 
 
1595
    @classmethod
 
1596
    @deprecated_method(deprecated_in((2, 4, 0)))
 
1597
    def get_default_format(klass):
 
1598
        """Return the current default format."""
 
1599
        return format_registry.get_default()
 
1600
 
 
1601
    @classmethod
 
1602
    @deprecated_method(deprecated_in((2, 4, 0)))
 
1603
    def get_formats(klass):
 
1604
        """Get all the known formats.
 
1605
 
 
1606
        Warning: This triggers a load of all lazy registered formats: do not
 
1607
        use except when that is desireed.
 
1608
        """
 
1609
        return format_registry._get_all()
 
1610
 
 
1611
    def get_reference(self, a_bzrdir, name=None):
 
1612
        """Get the target reference of the branch in a_bzrdir.
1600
1613
 
1601
1614
        format probing must have been completed before calling
1602
1615
        this method - it is assumed that the format of the branch
1603
 
        in controldir is correct.
 
1616
        in a_bzrdir is correct.
1604
1617
 
1605
 
        :param controldir: The controldir to get the branch data from.
 
1618
        :param a_bzrdir: The bzrdir to get the branch data from.
1606
1619
        :param name: Name of the colocated branch to fetch
1607
1620
        :return: None if the branch is not a reference branch.
1608
1621
        """
1609
1622
        return None
1610
1623
 
1611
1624
    @classmethod
1612
 
    def set_reference(self, controldir, name, to_branch):
1613
 
        """Set the target reference of the branch in controldir.
 
1625
    def set_reference(self, a_bzrdir, name, to_branch):
 
1626
        """Set the target reference of the branch in a_bzrdir.
1614
1627
 
1615
1628
        format probing must have been completed before calling
1616
1629
        this method - it is assumed that the format of the branch
1617
 
        in controldir is correct.
 
1630
        in a_bzrdir is correct.
1618
1631
 
1619
 
        :param controldir: The controldir to set the branch reference for.
 
1632
        :param a_bzrdir: The bzrdir to set the branch reference for.
1620
1633
        :param name: Name of colocated branch to set, None for default
1621
1634
        :param to_branch: branch that the checkout is to reference
1622
1635
        """
1623
1636
        raise NotImplementedError(self.set_reference)
1624
1637
 
 
1638
    def get_format_string(self):
 
1639
        """Return the ASCII format string that identifies this format."""
 
1640
        raise NotImplementedError(self.get_format_string)
 
1641
 
1625
1642
    def get_format_description(self):
1626
1643
        """Return the short format description for this format."""
1627
1644
        raise NotImplementedError(self.get_format_description)
1628
1645
 
1629
 
    def _run_post_branch_init_hooks(self, controldir, name, branch):
 
1646
    def _run_post_branch_init_hooks(self, a_bzrdir, name, branch):
1630
1647
        hooks = Branch.hooks['post_branch_init']
1631
1648
        if not hooks:
1632
1649
            return
1633
 
        params = BranchInitHookParams(self, controldir, name, branch)
 
1650
        params = BranchInitHookParams(self, a_bzrdir, name, branch)
1634
1651
        for hook in hooks:
1635
1652
            hook(params)
1636
1653
 
1637
 
    def initialize(self, controldir, name=None, repository=None,
1638
 
                   append_revisions_only=None):
1639
 
        """Create a branch of this format in controldir.
1640
 
 
 
1654
    def initialize(self, a_bzrdir, name=None, repository=None):
 
1655
        """Create a branch of this format in a_bzrdir.
 
1656
        
1641
1657
        :param name: Name of the colocated branch to create.
1642
1658
        """
1643
1659
        raise NotImplementedError(self.initialize)
1663
1679
        Note that it is normal for branch to be a RemoteBranch when using tags
1664
1680
        on a RemoteBranch.
1665
1681
        """
1666
 
        return _mod_tag.DisabledTags(branch)
 
1682
        return DisabledTags(branch)
1667
1683
 
1668
1684
    def network_name(self):
1669
1685
        """A simple byte string uniquely identifying this format for RPC calls.
1675
1691
        """
1676
1692
        raise NotImplementedError(self.network_name)
1677
1693
 
1678
 
    def open(self, controldir, name=None, _found=False, ignore_fallbacks=False,
1679
 
            found_repository=None, possible_transports=None):
1680
 
        """Return the branch object for controldir.
 
1694
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False,
 
1695
            found_repository=None):
 
1696
        """Return the branch object for a_bzrdir
1681
1697
 
1682
 
        :param controldir: A ControlDir that contains a branch.
 
1698
        :param a_bzrdir: A BzrDir that contains a branch.
1683
1699
        :param name: Name of colocated branch to open
1684
1700
        :param _found: a private parameter, do not use it. It is used to
1685
1701
            indicate if format probing has already be done.
1688
1704
        """
1689
1705
        raise NotImplementedError(self.open)
1690
1706
 
 
1707
    @classmethod
 
1708
    @deprecated_method(deprecated_in((2, 4, 0)))
 
1709
    def register_format(klass, format):
 
1710
        """Register a metadir format.
 
1711
 
 
1712
        See MetaDirBranchFormatFactory for the ability to register a format
 
1713
        without loading the code the format needs until it is actually used.
 
1714
        """
 
1715
        format_registry.register(format)
 
1716
 
 
1717
    @classmethod
 
1718
    @deprecated_method(deprecated_in((2, 4, 0)))
 
1719
    def set_default_format(klass, format):
 
1720
        format_registry.set_default(format)
 
1721
 
1691
1722
    def supports_set_append_revisions_only(self):
1692
1723
        """True if this format supports set_append_revisions_only."""
1693
1724
        return False
1700
1731
        """True if this format supports leaving locks in place."""
1701
1732
        return False # by default
1702
1733
 
 
1734
    @classmethod
 
1735
    @deprecated_method(deprecated_in((2, 4, 0)))
 
1736
    def unregister_format(klass, format):
 
1737
        format_registry.remove(format)
 
1738
 
1703
1739
    def __str__(self):
1704
1740
        return self.get_format_description().rstrip()
1705
1741
 
1707
1743
        """True if this format supports tags stored in the branch"""
1708
1744
        return False  # by default
1709
1745
 
1710
 
    def tags_are_versioned(self):
1711
 
        """Whether the tag container for this branch versions tags."""
1712
 
        return False
1713
 
 
1714
 
    def supports_tags_referencing_ghosts(self):
1715
 
        """True if tags can reference ghost revisions."""
1716
 
        return True
1717
 
 
1718
1746
 
1719
1747
class MetaDirBranchFormatFactory(registry._LazyObjectGetter):
1720
1748
    """A factory for a BranchFormat object, permitting simple lazy registration.
1734
1762
        """
1735
1763
        registry._LazyObjectGetter.__init__(self, module_name, member_name)
1736
1764
        self._format_string = format_string
1737
 
 
 
1765
        
1738
1766
    def get_format_string(self):
1739
1767
        """See BranchFormat.get_format_string."""
1740
1768
        return self._format_string
1747
1775
class BranchHooks(Hooks):
1748
1776
    """A dictionary mapping hook name to a list of callables for branch hooks.
1749
1777
 
1750
 
    e.g. ['post_push'] Is the list of items to be called when the
1751
 
    push function is invoked.
 
1778
    e.g. ['set_rh'] Is the list of items to be called when the
 
1779
    set_revision_history function is invoked.
1752
1780
    """
1753
1781
 
1754
1782
    def __init__(self):
1758
1786
        notified.
1759
1787
        """
1760
1788
        Hooks.__init__(self, "bzrlib.branch", "Branch.hooks")
 
1789
        self.add_hook('set_rh',
 
1790
            "Invoked whenever the revision history has been set via "
 
1791
            "set_revision_history. The api signature is (branch, "
 
1792
            "revision_history), and the branch will be write-locked. "
 
1793
            "The set_rh hook can be expensive for bzr to trigger, a better "
 
1794
            "hook to use is Branch.post_change_branch_tip.", (0, 15))
1761
1795
        self.add_hook('open',
1762
1796
            "Called with the Branch object that has been opened after a "
1763
1797
            "branch is opened.", (1, 8))
1879
1913
    There are 4 fields that hooks may wish to access:
1880
1914
 
1881
1915
    :ivar format: the branch format
1882
 
    :ivar bzrdir: the ControlDir where the branch will be/has been initialized
 
1916
    :ivar bzrdir: the BzrDir where the branch will be/has been initialized
1883
1917
    :ivar name: name of colocated branch, if any (or None)
1884
1918
    :ivar branch: the branch created
1885
1919
 
1888
1922
    branch, which refer to the original branch.
1889
1923
    """
1890
1924
 
1891
 
    def __init__(self, format, controldir, name, branch):
 
1925
    def __init__(self, format, a_bzrdir, name, branch):
1892
1926
        """Create a group of BranchInitHook parameters.
1893
1927
 
1894
1928
        :param format: the branch format
1895
 
        :param controldir: the ControlDir where the branch will be/has been
 
1929
        :param a_bzrdir: the BzrDir where the branch will be/has been
1896
1930
            initialized
1897
1931
        :param name: name of colocated branch, if any (or None)
1898
1932
        :param branch: the branch created
1902
1936
        in branch, which refer to the original branch.
1903
1937
        """
1904
1938
        self.format = format
1905
 
        self.bzrdir = controldir
 
1939
        self.bzrdir = a_bzrdir
1906
1940
        self.name = name
1907
1941
        self.branch = branch
1908
1942
 
1918
1952
 
1919
1953
    There are 4 fields that hooks may wish to access:
1920
1954
 
1921
 
    :ivar control_dir: ControlDir of the checkout to change
 
1955
    :ivar control_dir: BzrDir of the checkout to change
1922
1956
    :ivar to_branch: branch that the checkout is to reference
1923
1957
    :ivar force: skip the check for local commits in a heavy checkout
1924
1958
    :ivar revision_id: revision ID to switch to (or None)
1927
1961
    def __init__(self, control_dir, to_branch, force, revision_id):
1928
1962
        """Create a group of SwitchHook parameters.
1929
1963
 
1930
 
        :param control_dir: ControlDir of the checkout to change
 
1964
        :param control_dir: BzrDir of the checkout to change
1931
1965
        :param to_branch: branch that the checkout is to reference
1932
1966
        :param force: skip the check for local commits in a heavy checkout
1933
1967
        :param revision_id: revision ID to switch to (or None)
1946
1980
            self.revision_id)
1947
1981
 
1948
1982
 
1949
 
class BranchFormatMetadir(bzrdir.BzrFormat, BranchFormat):
1950
 
    """Base class for branch formats that live in meta directories.
1951
 
    """
1952
 
 
1953
 
    def __init__(self):
1954
 
        BranchFormat.__init__(self)
1955
 
        bzrdir.BzrFormat.__init__(self)
1956
 
 
1957
 
    @classmethod
1958
 
    def find_format(klass, controldir, name=None):
1959
 
        """Return the format for the branch object in controldir."""
1960
 
        try:
1961
 
            transport = controldir.get_branch_transport(None, name=name)
1962
 
        except errors.NoSuchFile:
1963
 
            raise errors.NotBranchError(path=name, bzrdir=controldir)
1964
 
        try:
1965
 
            format_string = transport.get_bytes("format")
1966
 
        except errors.NoSuchFile:
1967
 
            raise errors.NotBranchError(path=transport.base, bzrdir=controldir)
1968
 
        return klass._find_format(format_registry, 'branch', format_string)
 
1983
class BranchFormatMetadir(BranchFormat):
 
1984
    """Common logic for meta-dir based branch formats."""
1969
1985
 
1970
1986
    def _branch_class(self):
1971
1987
        """What class to instantiate on open calls."""
1972
1988
        raise NotImplementedError(self._branch_class)
1973
1989
 
1974
 
    def _get_initial_config(self, append_revisions_only=None):
1975
 
        if append_revisions_only:
1976
 
            return "append_revisions_only = True\n"
1977
 
        else:
1978
 
            # Avoid writing anything if append_revisions_only is disabled,
1979
 
            # as that is the default.
1980
 
            return ""
1981
 
 
1982
1990
    def _initialize_helper(self, a_bzrdir, utf8_files, name=None,
1983
1991
                           repository=None):
1984
 
        """Initialize a branch in a control dir, with specified files
 
1992
        """Initialize a branch in a bzrdir, with specified files
1985
1993
 
1986
1994
        :param a_bzrdir: The bzrdir to initialize the branch in
1987
1995
        :param utf8_files: The files to create as a list of
1989
1997
        :param name: Name of colocated branch to create, if any
1990
1998
        :return: a branch in this format
1991
1999
        """
1992
 
        if name is None:
1993
 
            name = a_bzrdir._get_selected_branch()
1994
2000
        mutter('creating branch %r in %s', self, a_bzrdir.user_url)
1995
2001
        branch_transport = a_bzrdir.get_branch_transport(self, name=name)
1996
2002
        control_files = lockable_files.LockableFiles(branch_transport,
1998
2004
        control_files.create_lock()
1999
2005
        control_files.lock_write()
2000
2006
        try:
2001
 
            utf8_files += [('format', self.as_string())]
 
2007
            utf8_files += [('format', self.get_format_string())]
2002
2008
            for (filename, content) in utf8_files:
2003
2009
                branch_transport.put_bytes(
2004
2010
                    filename, content,
2010
2016
        self._run_post_branch_init_hooks(a_bzrdir, name, branch)
2011
2017
        return branch
2012
2018
 
 
2019
    def network_name(self):
 
2020
        """A simple byte string uniquely identifying this format for RPC calls.
 
2021
 
 
2022
        Metadir branch formats use their format string.
 
2023
        """
 
2024
        return self.get_format_string()
 
2025
 
2013
2026
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False,
2014
 
            found_repository=None, possible_transports=None):
 
2027
            found_repository=None):
2015
2028
        """See BranchFormat.open()."""
2016
 
        if name is None:
2017
 
            name = a_bzrdir._get_selected_branch()
2018
2029
        if not _found:
2019
 
            format = BranchFormatMetadir.find_format(a_bzrdir, name=name)
 
2030
            format = BranchFormat.find_format(a_bzrdir, name=name)
2020
2031
            if format.__class__ != self.__class__:
2021
2032
                raise AssertionError("wrong format %r found for %r" %
2022
2033
                    (format, self))
2031
2042
                              name=name,
2032
2043
                              a_bzrdir=a_bzrdir,
2033
2044
                              _repository=found_repository,
2034
 
                              ignore_fallbacks=ignore_fallbacks,
2035
 
                              possible_transports=possible_transports)
 
2045
                              ignore_fallbacks=ignore_fallbacks)
2036
2046
        except errors.NoSuchFile:
2037
2047
            raise errors.NotBranchError(path=transport.base, bzrdir=a_bzrdir)
2038
2048
 
2039
 
    @property
2040
 
    def _matchingbzrdir(self):
2041
 
        ret = bzrdir.BzrDirMetaFormat1()
2042
 
        ret.set_branch_format(self)
2043
 
        return ret
 
2049
    def __init__(self):
 
2050
        super(BranchFormatMetadir, self).__init__()
 
2051
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
 
2052
        self._matchingbzrdir.set_branch_format(self)
2044
2053
 
2045
2054
    def supports_tags(self):
2046
2055
        return True
2048
2057
    def supports_leaving_lock(self):
2049
2058
        return True
2050
2059
 
2051
 
    def check_support_status(self, allow_unsupported, recommend_upgrade=True,
2052
 
            basedir=None):
2053
 
        BranchFormat.check_support_status(self,
2054
 
            allow_unsupported=allow_unsupported, recommend_upgrade=recommend_upgrade,
2055
 
            basedir=basedir)
2056
 
        bzrdir.BzrFormat.check_support_status(self, allow_unsupported=allow_unsupported,
2057
 
            recommend_upgrade=recommend_upgrade, basedir=basedir)
 
2060
 
 
2061
class BzrBranchFormat5(BranchFormatMetadir):
 
2062
    """Bzr branch format 5.
 
2063
 
 
2064
    This format has:
 
2065
     - a revision-history file.
 
2066
     - a format string
 
2067
     - a lock dir guarding the branch itself
 
2068
     - all of this stored in a branch/ subdirectory
 
2069
     - works with shared repositories.
 
2070
 
 
2071
    This format is new in bzr 0.8.
 
2072
    """
 
2073
 
 
2074
    def _branch_class(self):
 
2075
        return BzrBranch5
 
2076
 
 
2077
    def get_format_string(self):
 
2078
        """See BranchFormat.get_format_string()."""
 
2079
        return "Bazaar-NG branch format 5\n"
 
2080
 
 
2081
    def get_format_description(self):
 
2082
        """See BranchFormat.get_format_description()."""
 
2083
        return "Branch format 5"
 
2084
 
 
2085
    def initialize(self, a_bzrdir, name=None, repository=None):
 
2086
        """Create a branch of this format in a_bzrdir."""
 
2087
        utf8_files = [('revision-history', ''),
 
2088
                      ('branch-name', ''),
 
2089
                      ]
 
2090
        return self._initialize_helper(a_bzrdir, utf8_files, name, repository)
 
2091
 
 
2092
    def supports_tags(self):
 
2093
        return False
2058
2094
 
2059
2095
 
2060
2096
class BzrBranchFormat6(BranchFormatMetadir):
2071
2107
    def _branch_class(self):
2072
2108
        return BzrBranch6
2073
2109
 
2074
 
    @classmethod
2075
 
    def get_format_string(cls):
 
2110
    def get_format_string(self):
2076
2111
        """See BranchFormat.get_format_string()."""
2077
2112
        return "Bazaar Branch Format 6 (bzr 0.15)\n"
2078
2113
 
2080
2115
        """See BranchFormat.get_format_description()."""
2081
2116
        return "Branch format 6"
2082
2117
 
2083
 
    def initialize(self, a_bzrdir, name=None, repository=None,
2084
 
                   append_revisions_only=None):
 
2118
    def initialize(self, a_bzrdir, name=None, repository=None):
2085
2119
        """Create a branch of this format in a_bzrdir."""
2086
2120
        utf8_files = [('last-revision', '0 null:\n'),
2087
 
                      ('branch.conf',
2088
 
                          self._get_initial_config(append_revisions_only)),
 
2121
                      ('branch.conf', ''),
2089
2122
                      ('tags', ''),
2090
2123
                      ]
2091
2124
        return self._initialize_helper(a_bzrdir, utf8_files, name, repository)
2092
2125
 
2093
2126
    def make_tags(self, branch):
2094
2127
        """See bzrlib.branch.BranchFormat.make_tags()."""
2095
 
        return _mod_tag.BasicTags(branch)
 
2128
        return BasicTags(branch)
2096
2129
 
2097
2130
    def supports_set_append_revisions_only(self):
2098
2131
        return True
2104
2137
    def _branch_class(self):
2105
2138
        return BzrBranch8
2106
2139
 
2107
 
    @classmethod
2108
 
    def get_format_string(cls):
 
2140
    def get_format_string(self):
2109
2141
        """See BranchFormat.get_format_string()."""
2110
2142
        return "Bazaar Branch Format 8 (needs bzr 1.15)\n"
2111
2143
 
2113
2145
        """See BranchFormat.get_format_description()."""
2114
2146
        return "Branch format 8"
2115
2147
 
2116
 
    def initialize(self, a_bzrdir, name=None, repository=None,
2117
 
                   append_revisions_only=None):
 
2148
    def initialize(self, a_bzrdir, name=None, repository=None):
2118
2149
        """Create a branch of this format in a_bzrdir."""
2119
2150
        utf8_files = [('last-revision', '0 null:\n'),
2120
 
                      ('branch.conf',
2121
 
                          self._get_initial_config(append_revisions_only)),
 
2151
                      ('branch.conf', ''),
2122
2152
                      ('tags', ''),
2123
2153
                      ('references', '')
2124
2154
                      ]
2126
2156
 
2127
2157
    def make_tags(self, branch):
2128
2158
        """See bzrlib.branch.BranchFormat.make_tags()."""
2129
 
        return _mod_tag.BasicTags(branch)
 
2159
        return BasicTags(branch)
2130
2160
 
2131
2161
    def supports_set_append_revisions_only(self):
2132
2162
        return True
2146
2176
    This format was introduced in bzr 1.6.
2147
2177
    """
2148
2178
 
2149
 
    def initialize(self, a_bzrdir, name=None, repository=None,
2150
 
                   append_revisions_only=None):
 
2179
    def initialize(self, a_bzrdir, name=None, repository=None):
2151
2180
        """Create a branch of this format in a_bzrdir."""
2152
2181
        utf8_files = [('last-revision', '0 null:\n'),
2153
 
                      ('branch.conf',
2154
 
                          self._get_initial_config(append_revisions_only)),
 
2182
                      ('branch.conf', ''),
2155
2183
                      ('tags', ''),
2156
2184
                      ]
2157
2185
        return self._initialize_helper(a_bzrdir, utf8_files, name, repository)
2159
2187
    def _branch_class(self):
2160
2188
        return BzrBranch7
2161
2189
 
2162
 
    @classmethod
2163
 
    def get_format_string(cls):
 
2190
    def get_format_string(self):
2164
2191
        """See BranchFormat.get_format_string()."""
2165
2192
        return "Bazaar Branch Format 7 (needs bzr 1.6)\n"
2166
2193
 
2176
2203
 
2177
2204
    def make_tags(self, branch):
2178
2205
        """See bzrlib.branch.BranchFormat.make_tags()."""
2179
 
        return _mod_tag.BasicTags(branch)
 
2206
        return BasicTags(branch)
2180
2207
 
2181
2208
    supports_reference_locations = False
2182
2209
 
2183
2210
 
2184
 
class BranchReferenceFormat(BranchFormatMetadir):
 
2211
class BranchReferenceFormat(BranchFormat):
2185
2212
    """Bzr branch reference format.
2186
2213
 
2187
2214
    Branch references are used in implementing checkouts, they
2192
2219
     - a format string
2193
2220
    """
2194
2221
 
2195
 
    @classmethod
2196
 
    def get_format_string(cls):
 
2222
    def get_format_string(self):
2197
2223
        """See BranchFormat.get_format_string()."""
2198
2224
        return "Bazaar-NG Branch Reference Format 1\n"
2199
2225
 
2212
2238
        location = transport.put_bytes('location', to_branch.base)
2213
2239
 
2214
2240
    def initialize(self, a_bzrdir, name=None, target_branch=None,
2215
 
            repository=None, append_revisions_only=None):
 
2241
            repository=None):
2216
2242
        """Create a branch of this format in a_bzrdir."""
2217
2243
        if target_branch is None:
2218
2244
            # this format does not implement branch itself, thus the implicit
2219
2245
            # creation contract must see it as uninitializable
2220
2246
            raise errors.UninitializableFormat(self)
2221
2247
        mutter('creating branch reference in %s', a_bzrdir.user_url)
2222
 
        if a_bzrdir._format.fixed_components:
2223
 
            raise errors.IncompatibleFormat(self, a_bzrdir._format)
2224
 
        if name is None:
2225
 
            name = a_bzrdir._get_selected_branch()
2226
2248
        branch_transport = a_bzrdir.get_branch_transport(self, name=name)
2227
2249
        branch_transport.put_bytes('location',
2228
 
            target_branch.user_url)
2229
 
        branch_transport.put_bytes('format', self.as_string())
2230
 
        branch = self.open(a_bzrdir, name, _found=True,
 
2250
            target_branch.bzrdir.user_url)
 
2251
        branch_transport.put_bytes('format', self.get_format_string())
 
2252
        branch = self.open(
 
2253
            a_bzrdir, name, _found=True,
2231
2254
            possible_transports=[target_branch.bzrdir.root_transport])
2232
2255
        self._run_post_branch_init_hooks(a_bzrdir, name, branch)
2233
2256
        return branch
2234
2257
 
 
2258
    def __init__(self):
 
2259
        super(BranchReferenceFormat, self).__init__()
 
2260
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
 
2261
        self._matchingbzrdir.set_branch_format(self)
 
2262
 
2235
2263
    def _make_reference_clone_function(format, a_branch):
2236
2264
        """Create a clone() routine for a branch dynamically."""
2237
2265
        def clone(to_bzrdir, revision_id=None,
2259
2287
            a_bzrdir.
2260
2288
        :param possible_transports: An optional reusable transports list.
2261
2289
        """
2262
 
        if name is None:
2263
 
            name = a_bzrdir._get_selected_branch()
2264
2290
        if not _found:
2265
 
            format = BranchFormatMetadir.find_format(a_bzrdir, name=name)
 
2291
            format = BranchFormat.find_format(a_bzrdir, name=name)
2266
2292
            if format.__class__ != self.__class__:
2267
2293
                raise AssertionError("wrong format %r found for %r" %
2268
2294
                    (format, self))
2269
2295
        if location is None:
2270
2296
            location = self.get_reference(a_bzrdir, name)
2271
 
        real_bzrdir = controldir.ControlDir.open(
 
2297
        real_bzrdir = bzrdir.BzrDir.open(
2272
2298
            location, possible_transports=possible_transports)
2273
 
        result = real_bzrdir.open_branch(ignore_fallbacks=ignore_fallbacks,
2274
 
            possible_transports=possible_transports)
 
2299
        result = real_bzrdir.open_branch(name=name, 
 
2300
            ignore_fallbacks=ignore_fallbacks)
2275
2301
        # this changes the behaviour of result.clone to create a new reference
2276
2302
        # rather than a copy of the content of the branch.
2277
2303
        # I did not use a proxy object because that needs much more extensive
2311
2337
 
2312
2338
# formats which have no format string are not discoverable
2313
2339
# and not independently creatable, so are not registered.
 
2340
__format5 = BzrBranchFormat5()
2314
2341
__format6 = BzrBranchFormat6()
2315
2342
__format7 = BzrBranchFormat7()
2316
2343
__format8 = BzrBranchFormat8()
2317
 
format_registry.register_lazy(
2318
 
    "Bazaar-NG branch format 5\n", "bzrlib.branchfmt.fullhistory", "BzrBranchFormat5")
 
2344
format_registry.register(__format5)
2319
2345
format_registry.register(BranchReferenceFormat())
2320
2346
format_registry.register(__format6)
2321
2347
format_registry.register(__format7)
2358
2384
 
2359
2385
    def __init__(self, _format=None,
2360
2386
                 _control_files=None, a_bzrdir=None, name=None,
2361
 
                 _repository=None, ignore_fallbacks=False,
2362
 
                 possible_transports=None):
 
2387
                 _repository=None, ignore_fallbacks=False):
2363
2388
        """Create new branch object at a particular location."""
2364
2389
        if a_bzrdir is None:
2365
2390
            raise ValueError('a_bzrdir must be supplied')
2366
 
        if name is None:
2367
 
            raise ValueError('name must be supplied')
2368
 
        self.bzrdir = a_bzrdir
2369
 
        self._user_transport = self.bzrdir.transport.clone('..')
2370
 
        if name != "":
2371
 
            self._user_transport.set_segment_parameter(
2372
 
                "branch", urlutils.escape(name))
2373
 
        self._base = self._user_transport.base
 
2391
        else:
 
2392
            self.bzrdir = a_bzrdir
 
2393
        self._base = self.bzrdir.transport.clone('..').base
2374
2394
        self.name = name
 
2395
        # XXX: We should be able to just do
 
2396
        #   self.base = self.bzrdir.root_transport.base
 
2397
        # but this does not quite work yet -- mbp 20080522
2375
2398
        self._format = _format
2376
2399
        if _control_files is None:
2377
2400
            raise ValueError('BzrBranch _control_files is None')
2378
2401
        self.control_files = _control_files
2379
2402
        self._transport = _control_files._transport
2380
2403
        self.repository = _repository
2381
 
        self.conf_store = None
2382
 
        Branch.__init__(self, possible_transports)
 
2404
        Branch.__init__(self)
2383
2405
 
2384
2406
    def __str__(self):
2385
 
        return '%s(%s)' % (self.__class__.__name__, self.user_url)
 
2407
        if self.name is None:
 
2408
            return '%s(%s)' % (self.__class__.__name__, self.user_url)
 
2409
        else:
 
2410
            return '%s(%s,%s)' % (self.__class__.__name__, self.user_url,
 
2411
                self.name)
2386
2412
 
2387
2413
    __repr__ = __str__
2388
2414
 
2392
2418
 
2393
2419
    base = property(_get_base, doc="The URL for the root of this branch.")
2394
2420
 
2395
 
    @property
2396
 
    def user_transport(self):
2397
 
        return self._user_transport
2398
 
 
2399
2421
    def _get_config(self):
2400
 
        return _mod_config.TransportConfig(self._transport, 'branch.conf')
2401
 
 
2402
 
    def _get_config_store(self):
2403
 
        if self.conf_store is None:
2404
 
            self.conf_store =  _mod_config.BranchStore(self)
2405
 
        return self.conf_store
2406
 
 
2407
 
    def _uncommitted_branch(self):
2408
 
        """Return the branch that may contain uncommitted changes."""
2409
 
        master = self.get_master_branch()
2410
 
        if master is not None:
2411
 
            return master
2412
 
        else:
2413
 
            return self
2414
 
 
2415
 
    def store_uncommitted(self, creator):
2416
 
        """Store uncommitted changes from a ShelfCreator.
2417
 
 
2418
 
        :param creator: The ShelfCreator containing uncommitted changes, or
2419
 
            None to delete any stored changes.
2420
 
        :raises: ChangesAlreadyStored if the branch already has changes.
2421
 
        """
2422
 
        branch = self._uncommitted_branch()
2423
 
        if creator is None:
2424
 
            branch._transport.delete('stored-transform')
2425
 
            return
2426
 
        if branch._transport.has('stored-transform'):
2427
 
            raise errors.ChangesAlreadyStored
2428
 
        transform = StringIO()
2429
 
        creator.write_shelf(transform)
2430
 
        transform.seek(0)
2431
 
        branch._transport.put_file('stored-transform', transform)
2432
 
 
2433
 
    def get_unshelver(self, tree):
2434
 
        """Return a shelf.Unshelver for this branch and tree.
2435
 
 
2436
 
        :param tree: The tree to use to construct the Unshelver.
2437
 
        :return: an Unshelver or None if no changes are stored.
2438
 
        """
2439
 
        branch = self._uncommitted_branch()
2440
 
        try:
2441
 
            transform = branch._transport.get('stored-transform')
2442
 
        except errors.NoSuchFile:
2443
 
            return None
2444
 
        return shelf.Unshelver.from_tree_and_shelf(tree, transform)
 
2422
        return TransportConfig(self._transport, 'branch.conf')
2445
2423
 
2446
2424
    def is_locked(self):
2447
2425
        return self.control_files.is_locked()
2455
2433
        """
2456
2434
        if not self.is_locked():
2457
2435
            self._note_lock('w')
 
2436
        # All-in-one needs to always unlock/lock.
 
2437
        repo_control = getattr(self.repository, 'control_files', None)
 
2438
        if self.control_files == repo_control or not self.is_locked():
2458
2439
            self.repository._warn_if_deprecated(self)
2459
2440
            self.repository.lock_write()
2460
2441
            took_lock = True
2475
2456
        """
2476
2457
        if not self.is_locked():
2477
2458
            self._note_lock('r')
 
2459
        # All-in-one needs to always unlock/lock.
 
2460
        repo_control = getattr(self.repository, 'control_files', None)
 
2461
        if self.control_files == repo_control or not self.is_locked():
2478
2462
            self.repository._warn_if_deprecated(self)
2479
2463
            self.repository.lock_read()
2480
2464
            took_lock = True
2490
2474
 
2491
2475
    @only_raises(errors.LockNotHeld, errors.LockBroken)
2492
2476
    def unlock(self):
2493
 
        if self.control_files._lock_count == 1 and self.conf_store is not None:
2494
 
            self.conf_store.save_changes()
2495
2477
        try:
2496
2478
            self.control_files.unlock()
2497
2479
        finally:
 
2480
            # All-in-one needs to always unlock/lock.
 
2481
            repo_control = getattr(self.repository, 'control_files', None)
 
2482
            if (self.control_files == repo_control or
 
2483
                not self.control_files.is_locked()):
 
2484
                self.repository.unlock()
2498
2485
            if not self.control_files.is_locked():
2499
 
                self.repository.unlock()
2500
2486
                # we just released the lock
2501
2487
                self._clear_cached_state()
2502
2488
 
2520
2506
            raise errors.InvalidRevisionId(revision_id=revision_id, branch=self)
2521
2507
        revision_id = _mod_revision.ensure_null(revision_id)
2522
2508
        old_revno, old_revid = self.last_revision_info()
2523
 
        if self.get_append_revisions_only():
 
2509
        if self._get_append_revisions_only():
2524
2510
            self._check_history_violation(revision_id)
2525
2511
        self._run_pre_change_branch_tip_hooks(revno, revision_id)
2526
2512
        self._write_last_revision_info(revno, revision_id)
2669
2655
        self._transport.put_bytes('last-revision', out_string,
2670
2656
            mode=self.bzrdir._get_file_mode())
2671
2657
 
2672
 
    @needs_write_lock
2673
 
    def update_feature_flags(self, updated_flags):
2674
 
        """Update the feature flags for this branch.
2675
 
 
2676
 
        :param updated_flags: Dictionary mapping feature names to necessities
2677
 
            A necessity can be None to indicate the feature should be removed
 
2658
 
 
2659
class FullHistoryBzrBranch(BzrBranch):
 
2660
    """Bzr branch which contains the full revision history."""
 
2661
 
 
2662
    @needs_write_lock
 
2663
    def set_last_revision_info(self, revno, revision_id):
 
2664
        if not revision_id or not isinstance(revision_id, basestring):
 
2665
            raise errors.InvalidRevisionId(revision_id=revision_id, branch=self)
 
2666
        revision_id = _mod_revision.ensure_null(revision_id)
 
2667
        # this old format stores the full history, but this api doesn't
 
2668
        # provide it, so we must generate, and might as well check it's
 
2669
        # correct
 
2670
        history = self._lefthand_history(revision_id)
 
2671
        if len(history) != revno:
 
2672
            raise AssertionError('%d != %d' % (len(history), revno))
 
2673
        self._set_revision_history(history)
 
2674
 
 
2675
    def _read_last_revision_info(self):
 
2676
        rh = self.revision_history()
 
2677
        revno = len(rh)
 
2678
        if revno:
 
2679
            return (revno, rh[-1])
 
2680
        else:
 
2681
            return (0, _mod_revision.NULL_REVISION)
 
2682
 
 
2683
    @deprecated_method(deprecated_in((2, 4, 0)))
 
2684
    @needs_write_lock
 
2685
    def set_revision_history(self, rev_history):
 
2686
        """See Branch.set_revision_history."""
 
2687
        self._set_revision_history(rev_history)
 
2688
 
 
2689
    def _set_revision_history(self, rev_history):
 
2690
        if 'evil' in debug.debug_flags:
 
2691
            mutter_callsite(3, "set_revision_history scales with history.")
 
2692
        check_not_reserved_id = _mod_revision.check_not_reserved_id
 
2693
        for rev_id in rev_history:
 
2694
            check_not_reserved_id(rev_id)
 
2695
        if Branch.hooks['post_change_branch_tip']:
 
2696
            # Don't calculate the last_revision_info() if there are no hooks
 
2697
            # that will use it.
 
2698
            old_revno, old_revid = self.last_revision_info()
 
2699
        if len(rev_history) == 0:
 
2700
            revid = _mod_revision.NULL_REVISION
 
2701
        else:
 
2702
            revid = rev_history[-1]
 
2703
        self._run_pre_change_branch_tip_hooks(len(rev_history), revid)
 
2704
        self._write_revision_history(rev_history)
 
2705
        self._clear_cached_state()
 
2706
        self._cache_revision_history(rev_history)
 
2707
        for hook in Branch.hooks['set_rh']:
 
2708
            hook(self, rev_history)
 
2709
        if Branch.hooks['post_change_branch_tip']:
 
2710
            self._run_post_change_branch_tip_hooks(old_revno, old_revid)
 
2711
 
 
2712
    def _write_revision_history(self, history):
 
2713
        """Factored out of set_revision_history.
 
2714
 
 
2715
        This performs the actual writing to disk.
 
2716
        It is intended to be called by set_revision_history."""
 
2717
        self._transport.put_bytes(
 
2718
            'revision-history', '\n'.join(history),
 
2719
            mode=self.bzrdir._get_file_mode())
 
2720
 
 
2721
    def _gen_revision_history(self):
 
2722
        history = self._transport.get_bytes('revision-history').split('\n')
 
2723
        if history[-1:] == ['']:
 
2724
            # There shouldn't be a trailing newline, but just in case.
 
2725
            history.pop()
 
2726
        return history
 
2727
 
 
2728
    def _synchronize_history(self, destination, revision_id):
 
2729
        if not isinstance(destination, FullHistoryBzrBranch):
 
2730
            super(BzrBranch, self)._synchronize_history(
 
2731
                destination, revision_id)
 
2732
            return
 
2733
        if revision_id == _mod_revision.NULL_REVISION:
 
2734
            new_history = []
 
2735
        else:
 
2736
            new_history = self.revision_history()
 
2737
        if revision_id is not None and new_history != []:
 
2738
            try:
 
2739
                new_history = new_history[:new_history.index(revision_id) + 1]
 
2740
            except ValueError:
 
2741
                rev = self.repository.get_revision(revision_id)
 
2742
                new_history = rev.get_history(self.repository)[1:]
 
2743
        destination._set_revision_history(new_history)
 
2744
 
 
2745
    @needs_write_lock
 
2746
    def generate_revision_history(self, revision_id, last_rev=None,
 
2747
        other_branch=None):
 
2748
        """Create a new revision history that will finish with revision_id.
 
2749
 
 
2750
        :param revision_id: the new tip to use.
 
2751
        :param last_rev: The previous last_revision. If not None, then this
 
2752
            must be a ancestory of revision_id, or DivergedBranches is raised.
 
2753
        :param other_branch: The other branch that DivergedBranches should
 
2754
            raise with respect to.
2678
2755
        """
2679
 
        self._format._update_feature_flags(updated_flags)
2680
 
        self.control_transport.put_bytes('format', self._format.as_string())
 
2756
        self._set_revision_history(self._lefthand_history(revision_id,
 
2757
            last_rev, other_branch))
 
2758
 
 
2759
 
 
2760
class BzrBranch5(FullHistoryBzrBranch):
 
2761
    """A format 5 branch. This supports new features over plain branches.
 
2762
 
 
2763
    It has support for a master_branch which is the data for bound branches.
 
2764
    """
2681
2765
 
2682
2766
 
2683
2767
class BzrBranch8(BzrBranch):
2684
2768
    """A branch that stores tree-reference locations."""
2685
2769
 
2686
 
    def _open_hook(self, possible_transports=None):
 
2770
    def _open_hook(self):
2687
2771
        if self._ignore_fallbacks:
2688
2772
            return
2689
 
        if possible_transports is None:
2690
 
            possible_transports = [self.bzrdir.root_transport]
2691
2773
        try:
2692
2774
            url = self.get_stacked_on_url()
2693
2775
        except (errors.UnstackableRepositoryFormat, errors.NotStacked,
2701
2783
                    raise AssertionError(
2702
2784
                        "'transform_fallback_location' hook %s returned "
2703
2785
                        "None, not a URL." % hook_name)
2704
 
            self._activate_fallback_location(url,
2705
 
                possible_transports=possible_transports)
 
2786
            self._activate_fallback_location(url)
2706
2787
 
2707
2788
    def __init__(self, *args, **kwargs):
2708
2789
        self._ignore_fallbacks = kwargs.get('ignore_fallbacks', False)
2826
2907
        """See Branch.set_push_location."""
2827
2908
        self._master_branch_cache = None
2828
2909
        result = None
2829
 
        conf = self.get_config_stack()
 
2910
        config = self.get_config()
2830
2911
        if location is None:
2831
 
            if not conf.get('bound'):
 
2912
            if config.get_user_option('bound') != 'True':
2832
2913
                return False
2833
2914
            else:
2834
 
                conf.set('bound', 'False')
 
2915
                config.set_user_option('bound', 'False', warn_masked=True)
2835
2916
                return True
2836
2917
        else:
2837
2918
            self._set_config_location('bound_location', location,
2838
 
                                      config=conf)
2839
 
            conf.set('bound', 'True')
 
2919
                                      config=config)
 
2920
            config.set_user_option('bound', 'True', warn_masked=True)
2840
2921
        return True
2841
2922
 
2842
2923
    def _get_bound_location(self, bound):
2843
2924
        """Return the bound location in the config file.
2844
2925
 
2845
2926
        Return None if the bound parameter does not match"""
2846
 
        conf = self.get_config_stack()
2847
 
        if conf.get('bound') != bound:
 
2927
        config = self.get_config()
 
2928
        config_bound = (config.get_user_option('bound') == 'True')
 
2929
        if config_bound != bound:
2848
2930
            return None
2849
 
        return self._get_config_location('bound_location', config=conf)
 
2931
        return self._get_config_location('bound_location', config=config)
2850
2932
 
2851
2933
    def get_bound_location(self):
2852
 
        """See Branch.get_bound_location."""
 
2934
        """See Branch.set_push_location."""
2853
2935
        return self._get_bound_location(True)
2854
2936
 
2855
2937
    def get_old_bound_location(self):
2862
2944
        ## self._check_stackable_repo()
2863
2945
        # stacked_on_location is only ever defined in branch.conf, so don't
2864
2946
        # waste effort reading the whole stack of config files.
2865
 
        conf = _mod_config.BranchOnlyStack(self)
 
2947
        config = self.get_config()._get_branch_data_config()
2866
2948
        stacked_url = self._get_config_location('stacked_on_location',
2867
 
                                                config=conf)
 
2949
            config=config)
2868
2950
        if stacked_url is None:
2869
2951
            raise errors.NotStacked(self)
2870
 
        return stacked_url.encode('utf-8')
 
2952
        return stacked_url
 
2953
 
 
2954
    def _get_append_revisions_only(self):
 
2955
        return self.get_config(
 
2956
            ).get_user_option_as_bool('append_revisions_only')
2871
2957
 
2872
2958
    @needs_read_lock
2873
2959
    def get_rev_id(self, revno, history=None):
2903
2989
            except errors.RevisionNotPresent, e:
2904
2990
                raise errors.GhostRevisionsHaveNoRevno(revision_id, e.revision_id)
2905
2991
            index = len(self._partial_revision_history_cache) - 1
2906
 
            if index < 0:
2907
 
                raise errors.NoSuchRevision(self, revision_id)
2908
2992
            if self._partial_revision_history_cache[index] != revision_id:
2909
2993
                raise errors.NoSuchRevision(self, revision_id)
2910
2994
        return self.revno() - index
2962
3046
    :ivar local_branch: target branch if there is a Master, else None
2963
3047
    :ivar target_branch: Target/destination branch object. (write locked)
2964
3048
    :ivar tag_conflicts: A list of tag conflicts, see BasicTags.merge_to
2965
 
    :ivar tag_updates: A dict with new tags, see BasicTags.merge_to
2966
3049
    """
2967
3050
 
 
3051
    @deprecated_method(deprecated_in((2, 3, 0)))
 
3052
    def __int__(self):
 
3053
        """Return the relative change in revno.
 
3054
 
 
3055
        :deprecated: Use `new_revno` and `old_revno` instead.
 
3056
        """
 
3057
        return self.new_revno - self.old_revno
 
3058
 
2968
3059
    def report(self, to_file):
2969
 
        tag_conflicts = getattr(self, "tag_conflicts", None)
2970
 
        tag_updates = getattr(self, "tag_updates", None)
2971
3060
        if not is_quiet():
2972
 
            if self.old_revid != self.new_revid:
 
3061
            if self.old_revid == self.new_revid:
 
3062
                to_file.write('No revisions to pull.\n')
 
3063
            else:
2973
3064
                to_file.write('Now on revision %d.\n' % self.new_revno)
2974
 
            if tag_updates:
2975
 
                to_file.write('%d tag(s) updated.\n' % len(tag_updates))
2976
 
            if self.old_revid == self.new_revid and not tag_updates:
2977
 
                if not tag_conflicts:
2978
 
                    to_file.write('No revisions or tags to pull.\n')
2979
 
                else:
2980
 
                    to_file.write('No revisions to pull.\n')
2981
3065
        self._show_tag_conficts(to_file)
2982
3066
 
2983
3067
 
3000
3084
        target, otherwise it will be None.
3001
3085
    """
3002
3086
 
 
3087
    @deprecated_method(deprecated_in((2, 3, 0)))
 
3088
    def __int__(self):
 
3089
        """Return the relative change in revno.
 
3090
 
 
3091
        :deprecated: Use `new_revno` and `old_revno` instead.
 
3092
        """
 
3093
        return self.new_revno - self.old_revno
 
3094
 
3003
3095
    def report(self, to_file):
3004
 
        # TODO: This function gets passed a to_file, but then
3005
 
        # ignores it and calls note() instead. This is also
3006
 
        # inconsistent with PullResult(), which writes to stdout.
3007
 
        # -- JRV20110901, bug #838853
3008
 
        tag_conflicts = getattr(self, "tag_conflicts", None)
3009
 
        tag_updates = getattr(self, "tag_updates", None)
3010
 
        if not is_quiet():
3011
 
            if self.old_revid != self.new_revid:
3012
 
                note(gettext('Pushed up to revision %d.') % self.new_revno)
3013
 
            if tag_updates:
3014
 
                note(ngettext('%d tag updated.', '%d tags updated.', len(tag_updates)) % len(tag_updates))
3015
 
            if self.old_revid == self.new_revid and not tag_updates:
3016
 
                if not tag_conflicts:
3017
 
                    note(gettext('No new revisions or tags to push.'))
3018
 
                else:
3019
 
                    note(gettext('No new revisions to push.'))
 
3096
        """Write a human-readable description of the result."""
 
3097
        if self.old_revid == self.new_revid:
 
3098
            note('No new revisions to push.')
 
3099
        else:
 
3100
            note('Pushed up to revision %d.' % self.new_revno)
3020
3101
        self._show_tag_conficts(to_file)
3021
3102
 
3022
3103
 
3036
3117
        :param verbose: Requests more detailed display of what was checked,
3037
3118
            if any.
3038
3119
        """
3039
 
        note(gettext('checked branch {0} format {1}').format(
3040
 
                                self.branch.user_url, self.branch._format))
 
3120
        note('checked branch %s format %s', self.branch.user_url,
 
3121
            self.branch._format)
3041
3122
        for error in self.errors:
3042
 
            note(gettext('found error:%s'), error)
 
3123
            note('found error:%s', error)
3043
3124
 
3044
3125
 
3045
3126
class Converter5to6(object):
3052
3133
 
3053
3134
        # Copy source data into target
3054
3135
        new_branch._write_last_revision_info(*branch.last_revision_info())
3055
 
        new_branch.lock_write()
3056
 
        try:
3057
 
            new_branch.set_parent(branch.get_parent())
3058
 
            new_branch.set_bound_location(branch.get_bound_location())
3059
 
            new_branch.set_push_location(branch.get_push_location())
3060
 
        finally:
3061
 
            new_branch.unlock()
 
3136
        new_branch.set_parent(branch.get_parent())
 
3137
        new_branch.set_bound_location(branch.get_bound_location())
 
3138
        new_branch.set_push_location(branch.get_push_location())
3062
3139
 
3063
3140
        # New branch has no tags by default
3064
3141
        new_branch.tags._set_tag_dict({})
3065
3142
 
3066
3143
        # Copying done; now update target format
3067
3144
        new_branch._transport.put_bytes('format',
3068
 
            format.as_string(),
 
3145
            format.get_format_string(),
3069
3146
            mode=new_branch.bzrdir._get_file_mode())
3070
3147
 
3071
3148
        # Clean up old files
3072
3149
        new_branch._transport.delete('revision-history')
3073
 
        branch.lock_write()
3074
3150
        try:
3075
 
            try:
3076
 
                branch.set_parent(None)
3077
 
            except errors.NoSuchFile:
3078
 
                pass
3079
 
            branch.set_bound_location(None)
3080
 
        finally:
3081
 
            branch.unlock()
 
3151
            branch.set_parent(None)
 
3152
        except errors.NoSuchFile:
 
3153
            pass
 
3154
        branch.set_bound_location(None)
3082
3155
 
3083
3156
 
3084
3157
class Converter6to7(object):
3088
3161
        format = BzrBranchFormat7()
3089
3162
        branch._set_config_location('stacked_on_location', '')
3090
3163
        # update target format
3091
 
        branch._transport.put_bytes('format', format.as_string())
 
3164
        branch._transport.put_bytes('format', format.get_format_string())
3092
3165
 
3093
3166
 
3094
3167
class Converter7to8(object):
3095
 
    """Perform an in-place upgrade of format 7 to format 8"""
 
3168
    """Perform an in-place upgrade of format 6 to format 7"""
3096
3169
 
3097
3170
    def convert(self, branch):
3098
3171
        format = BzrBranchFormat8()
3099
3172
        branch._transport.put_bytes('references', '')
3100
3173
        # update target format
3101
 
        branch._transport.put_bytes('format', format.as_string())
 
3174
        branch._transport.put_bytes('format', format.get_format_string())
3102
3175
 
3103
3176
 
3104
3177
class InterBranch(InterObject):
3161
3234
        raise NotImplementedError(self.fetch)
3162
3235
 
3163
3236
 
3164
 
def _fix_overwrite_type(overwrite):
3165
 
    if isinstance(overwrite, bool):
3166
 
        if overwrite:
3167
 
            return ["history", "tags"]
3168
 
        else:
3169
 
            return []
3170
 
    return overwrite
3171
 
 
3172
 
 
3173
3237
class GenericInterBranch(InterBranch):
3174
3238
    """InterBranch implementation that uses public Branch functions."""
3175
3239
 
3280
3344
        if local and not bound_location:
3281
3345
            raise errors.LocalRequiresBoundBranch()
3282
3346
        master_branch = None
3283
 
        source_is_master = False
3284
 
        if bound_location:
3285
 
            # bound_location comes from a config file, some care has to be
3286
 
            # taken to relate it to source.user_url
3287
 
            normalized = urlutils.normalize_url(bound_location)
3288
 
            try:
3289
 
                relpath = self.source.user_transport.relpath(normalized)
3290
 
                source_is_master = (relpath == '')
3291
 
            except (errors.PathNotChild, errors.InvalidURL):
3292
 
                source_is_master = False
 
3347
        source_is_master = (self.source.user_url == bound_location)
3293
3348
        if not local and bound_location and not source_is_master:
3294
3349
            # not pulling from master, so we need to update master.
3295
3350
            master_branch = self.target.get_master_branch(possible_transports)
3340
3395
        result.target_branch = self.target
3341
3396
        result.old_revno, result.old_revid = self.target.last_revision_info()
3342
3397
        self.source.update_references(self.target)
3343
 
        overwrite = _fix_overwrite_type(overwrite)
3344
3398
        if result.old_revid != stop_revision:
3345
3399
            # We assume that during 'push' this repository is closer than
3346
3400
            # the target.
3347
3401
            graph = self.source.repository.get_graph(self.target.repository)
3348
 
            self._update_revisions(stop_revision,
3349
 
                overwrite=("history" in overwrite),
3350
 
                graph=graph)
 
3402
            self._update_revisions(stop_revision, overwrite=overwrite,
 
3403
                    graph=graph)
3351
3404
        if self.source._push_should_merge_tags():
3352
 
            result.tag_updates, result.tag_conflicts = (
3353
 
                self.source.tags.merge_to(
3354
 
                self.target.tags, "tags" in overwrite))
 
3405
            result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
 
3406
                overwrite)
3355
3407
        result.new_revno, result.new_revid = self.target.last_revision_info()
3356
3408
        return result
3357
3409
 
3435
3487
            # -- JRV20090506
3436
3488
            result.old_revno, result.old_revid = \
3437
3489
                self.target.last_revision_info()
3438
 
            overwrite = _fix_overwrite_type(overwrite)
3439
 
            self._update_revisions(stop_revision,
3440
 
                overwrite=("history" in overwrite),
 
3490
            self._update_revisions(stop_revision, overwrite=overwrite,
3441
3491
                graph=graph)
3442
3492
            # TODO: The old revid should be specified when merging tags, 
3443
3493
            # so a tags implementation that versions tags can only 
3444
3494
            # pull in the most recent changes. -- JRV20090506
3445
 
            result.tag_updates, result.tag_conflicts = (
3446
 
                self.source.tags.merge_to(self.target.tags,
3447
 
                    "tags" in overwrite,
3448
 
                    ignore_master=not merge_tags_to_master))
 
3495
            result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
 
3496
                overwrite, ignore_master=not merge_tags_to_master)
3449
3497
            result.new_revno, result.new_revid = self.target.last_revision_info()
3450
3498
            if _hook_master:
3451
3499
                result.master_branch = _hook_master