1
# Copyright (C) 2007, 2009, 2010 Canonical Ltd.
1
# Copyright (C) 2007, 2009-2012 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
19
19
# Original author: David Allouche
21
from bzrlib import errors, merge, revision
22
27
from bzrlib.branch import Branch
23
28
from bzrlib.i18n import gettext
24
29
from bzrlib.trace import note
35
def switch(control_dir, to_branch, force=False, quiet=False, revision_id=None):
40
def switch(control_dir, to_branch, force=False, quiet=False, revision_id=None,
41
store_uncommitted=False):
36
42
"""Switch the branch associated with a checkout.
38
44
:param control_dir: ControlDir of the checkout to change
39
45
:param to_branch: branch that the checkout is to reference
40
46
:param force: skip the check for local commits in a heavy checkout
41
47
:param revision_id: revision ID to switch to.
48
:param store_uncommitted: If True, store uncommitted changes in the
43
51
_check_pending_merges(control_dir, force)
45
53
source_repository = control_dir.open_branch().repository
46
54
except errors.NotBranchError:
47
55
source_repository = to_branch.repository
57
with lock.write_locked(control_dir.open_workingtree()) as tree:
58
tree.store_uncommitted()
48
59
to_branch.lock_read()
50
61
_set_branch_location(control_dir, to_branch, force)
53
64
tree = control_dir.open_workingtree()
54
_update(tree, source_repository, quiet, revision_id)
65
_update(tree, source_repository, quiet, revision_id, store_uncommitted)
55
66
_run_post_switch_hooks(control_dir, to_branch, force, revision_id)
57
68
def _check_pending_merges(control, force=False):
105
116
'Unable to connect to current master branch %(target)s: '
106
117
'%(error)s To switch anyway, use --force.') %
108
b.set_bound_location(None)
109
b.pull(to_branch, overwrite=True,
110
possible_transports=possible_transports)
111
b.set_bound_location(to_branch.base)
112
b.set_parent(b.get_master_branch().get_parent())
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())
114
129
# If this is a standalone tree and the new branch
115
130
# is derived from this one, create a lightweight checkout.
150
def _update(tree, source_repository, quiet=False, revision_id=None):
165
def _update(tree, source_repository, quiet=False, revision_id=None,
166
restore_uncommitted=False):
151
167
"""Update a working tree to the latest revision of its branch.
153
169
:param tree: the working tree
154
170
:param source_repository: repository holding the revisions
171
:param restore_uncommitted: restore any uncommitted changes in the branch.
156
tree.lock_tree_write()
173
if restore_uncommitted:
176
tree.lock_tree_write()
158
178
to_branch = tree.branch
159
179
if revision_id is None:
161
181
if tree.last_revision() == revision_id:
163
183
note(gettext("Tree is up to date at revision %d."), to_branch.revno())
165
base_tree = source_repository.revision_tree(tree.last_revision())
166
merge.Merge3Merger(tree, tree, base_tree, to_branch.repository.revision_tree(revision_id))
167
tree.set_last_revision(to_branch.last_revision())
169
note(gettext('Updated to 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()