1
# Copyright (C) 2007, 2009-2012 Canonical Ltd.
1
# Copyright (C) 2007, 2009, 2010 Canonical Ltd.
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
17
from __future__ import absolute_import
19
17
# Original author: David Allouche
19
from bzrlib import errors, merge, revision
27
20
from bzrlib.branch import Branch
28
from bzrlib.i18n import gettext
29
21
from bzrlib.trace import note
31
24
def _run_post_switch_hooks(control_dir, to_branch, force, revision_id):
32
25
from bzrlib.branch import SwitchHookParams
33
26
hooks = Branch.hooks['post_switch']
40
def switch(control_dir, to_branch, force=False, quiet=False, revision_id=None,
41
store_uncommitted=False):
33
def switch(control_dir, to_branch, force=False, quiet=False, revision_id=None):
42
34
"""Switch the branch associated with a checkout.
44
:param control_dir: ControlDir of the checkout to change
36
:param control_dir: BzrDir of the checkout to change
45
37
:param to_branch: branch that the checkout is to reference
46
38
:param force: skip the check for local commits in a heavy checkout
47
39
:param revision_id: revision ID to switch to.
48
:param store_uncommitted: If True, store uncommitted changes in the
51
41
_check_pending_merges(control_dir, force)
53
43
source_repository = control_dir.open_branch().repository
54
44
except errors.NotBranchError:
55
45
source_repository = to_branch.repository
57
with lock.write_locked(control_dir.open_workingtree()) as tree:
58
tree.store_uncommitted()
61
_set_branch_location(control_dir, to_branch, force)
46
_set_branch_location(control_dir, to_branch, force)
64
47
tree = control_dir.open_workingtree()
65
_update(tree, source_repository, quiet, revision_id, store_uncommitted)
48
_update(tree, source_repository, quiet, revision_id)
66
49
_run_post_switch_hooks(control_dir, to_branch, force, revision_id)
68
51
def _check_pending_merges(control, force=False):
69
52
"""Check that there are no outstanding pending merges before switching.
71
:param control: ControlDir of the branch to check
54
:param control: BzrDir of the branch to check
74
57
tree = control.open_workingtree()
81
64
# XXX: Should the tree be locked for get_parent_ids?
82
65
existing_pending_merges = tree.get_parent_ids()[1:]
83
66
if len(existing_pending_merges) > 0:
84
raise errors.BzrCommandError(gettext('Pending merges must be '
85
'committed or reverted before using switch.'))
67
raise errors.BzrCommandError('Pending merges must be '
68
'committed or reverted before using switch.')
88
71
def _set_branch_location(control, to_branch, force=False):
89
72
"""Set location value of a branch reference.
91
:param control: ControlDir of the checkout to change
74
:param control: BzrDir of the checkout to change
92
75
:param to_branch: branch that the checkout is to reference
93
76
:param force: skip the check for local commits in a heavy checkout
107
90
possible_transports = []
109
92
if not force and _any_local_commits(b, possible_transports):
110
raise errors.BzrCommandError(gettext(
93
raise errors.BzrCommandError(
111
94
'Cannot switch as local commits found in the checkout. '
112
95
'Commit these to the bound branch or use --force to '
114
97
except errors.BoundBranchConnectionFailure, e:
115
raise errors.BzrCommandError(gettext(
98
raise errors.BzrCommandError(
116
99
'Unable to connect to current master branch %(target)s: '
117
'%(error)s To switch anyway, use --force.') %
100
'%(error)s To switch anyway, use --force.' %
121
b.set_bound_location(None)
122
b.pull(to_branch, overwrite=True,
123
possible_transports=possible_transports)
124
b.set_bound_location(to_branch.base)
125
b.set_parent(b.get_master_branch().get_parent())
102
b.set_bound_location(None)
103
b.pull(to_branch, overwrite=True,
104
possible_transports=possible_transports)
105
b.set_bound_location(to_branch.base)
106
b.set_parent(b.get_master_branch().get_parent())
129
# If this is a standalone tree and the new branch
130
# is derived from this one, create a lightweight checkout.
133
graph = b.repository.get_graph(to_branch.repository)
134
if (b.bzrdir._format.colocated_branches and
135
(force or graph.is_ancestor(b.last_revision(),
136
to_branch.last_revision()))):
137
b.bzrdir.destroy_branch()
138
b.bzrdir.set_branch_reference(to_branch, name="")
140
raise errors.BzrCommandError(gettext('Cannot switch a branch, '
108
raise errors.BzrCommandError('Cannot switch a branch, '
146
112
def _any_local_commits(this_branch, possible_transports):
165
def _update(tree, source_repository, quiet=False, revision_id=None,
166
restore_uncommitted=False):
131
def _update(tree, source_repository, quiet=False, revision_id=None):
167
132
"""Update a working tree to the latest revision of its branch.
169
134
:param tree: the working tree
170
135
:param source_repository: repository holding the revisions
171
:param restore_uncommitted: restore any uncommitted changes in the branch.
173
if restore_uncommitted:
176
tree.lock_tree_write()
137
tree.lock_tree_write()
178
139
to_branch = tree.branch
179
140
if revision_id is None:
180
141
revision_id = to_branch.last_revision()
181
142
if tree.last_revision() == revision_id:
183
note(gettext("Tree is up to date at revision %d."), to_branch.revno())
185
base_tree = source_repository.revision_tree(tree.last_revision())
186
merge.Merge3Merger(tree, tree, base_tree,
187
to_branch.repository.revision_tree(revision_id))
188
tree.set_last_revision(to_branch.last_revision())
190
note(gettext('Updated to revision %d.') % to_branch.revno())
191
if restore_uncommitted:
192
tree.restore_uncommitted()
144
note("Tree is up to date at revision %d.", to_branch.revno())
146
base_tree = source_repository.revision_tree(tree.last_revision())
147
merge.Merge3Merger(tree, tree, base_tree, to_branch.repository.revision_tree(revision_id))
148
tree.set_last_revision(to_branch.last_revision())
150
note('Updated to revision %d.' % to_branch.revno())