1
# Copyright (C) 2005, 2006, 2007, 2009, 2010, 2011 Canonical Ltd
1
# Copyright (C) 2006 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
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
17
This contains functionality for installing bundles into repositories
21
from bzrlib.i18n import gettext
21
from bzrlib.progress import ProgressPhase
22
22
from bzrlib.merge import Merger
23
from bzrlib.progress import ProgressPhase
23
from bzrlib.repository import install_revision
24
24
from bzrlib.trace import note
25
from bzrlib.vf_repository import install_revision
28
27
def install_bundle(repository, bundle_reader):
29
28
custom_install = getattr(bundle_reader, 'install', None)
30
29
if custom_install is not None:
31
30
return custom_install(repository)
32
pb = ui.ui_factory.nested_progress_bar()
31
pb = bzrlib.ui.ui_factory.nested_progress_bar()
33
32
repository.lock_write()
35
34
real_revisions = bundle_reader.real_revisions
36
35
for i, revision in enumerate(reversed(real_revisions)):
37
pb.update(gettext("Install revisions"),i, len(real_revisions))
36
pb.update("Install revisions",i, len(real_revisions))
38
37
if repository.has_revision(revision.revision_id):
40
39
cset_tree = bundle_reader.revision_tree(repository,
48
def merge_bundle(reader, tree, check_clean, merge_type,
47
def merge_bundle(reader, tree, check_clean, merge_type,
49
48
reprocess, show_base, change_reporter=None):
50
49
"""Merge a revision bundle into the current tree."""
51
pb = ui.ui_factory.nested_progress_bar()
50
pb = bzrlib.ui.ui_factory.nested_progress_bar()
53
52
pp = ProgressPhase("Merge phase", 6, pb)
55
54
install_bundle(tree.branch.repository, reader)
56
merger = Merger(tree.branch, this_tree=tree,
55
merger = Merger(tree.branch, this_tree=tree, pb=pb,
57
56
change_reporter=change_reporter)
59
58
merger.pp.next_phase()
60
if check_clean and tree.has_changes():
61
raise errors.UncommittedChanges(self)
59
merger.check_basis(check_clean, require_commits=False)
62
60
merger.other_rev_id = reader.target
63
61
merger.other_tree = merger.revision_tree(reader.target)
64
62
merger.other_basis = reader.target
65
63
merger.pp.next_phase()
67
65
if merger.base_rev_id == merger.other_rev_id:
68
note(gettext("Nothing to do."))
66
note("Nothing to do.")
70
68
merger.merge_type = merge_type
71
69
merger.show_base = show_base