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
18
17
This contains functionality for installing bundles into repositories
21
from __future__ import absolute_import
24
from bzrlib.i18n import gettext
21
from bzrlib.progress import ProgressPhase
25
22
from bzrlib.merge import Merger
26
from bzrlib.progress import ProgressPhase
23
from bzrlib.repository import install_revision
27
24
from bzrlib.trace import note
28
from bzrlib.vf_repository import install_revision
31
27
def install_bundle(repository, bundle_reader):
32
custom_install = getattr(bundle_reader, 'install', None)
33
if custom_install is not None:
34
return custom_install(repository)
35
pb = ui.ui_factory.nested_progress_bar()
28
pb = bzrlib.ui.ui_factory.nested_progress_bar()
36
29
repository.lock_write()
38
31
real_revisions = bundle_reader.real_revisions
39
32
for i, revision in enumerate(reversed(real_revisions)):
40
pb.update(gettext("Install revisions"),i, len(real_revisions))
33
pb.update("Install revisions",i, len(real_revisions))
41
34
if repository.has_revision(revision.revision_id):
43
36
cset_tree = bundle_reader.revision_tree(repository,
51
def merge_bundle(reader, tree, check_clean, merge_type,
44
def merge_bundle(reader, tree, check_clean, merge_type,
52
45
reprocess, show_base, change_reporter=None):
53
46
"""Merge a revision bundle into the current tree."""
54
pb = ui.ui_factory.nested_progress_bar()
47
pb = bzrlib.ui.ui_factory.nested_progress_bar()
56
49
pp = ProgressPhase("Merge phase", 6, pb)
58
51
install_bundle(tree.branch.repository, reader)
59
merger = Merger(tree.branch, this_tree=tree,
52
merger = Merger(tree.branch, this_tree=tree, pb=pb,
60
53
change_reporter=change_reporter)
62
55
merger.pp.next_phase()
63
if check_clean and tree.has_changes():
64
raise errors.UncommittedChanges(self)
56
merger.check_basis(check_clean, require_commits=False)
65
57
merger.other_rev_id = reader.target
66
58
merger.other_tree = merger.revision_tree(reader.target)
67
59
merger.other_basis = reader.target
68
60
merger.pp.next_phase()
70
62
if merger.base_rev_id == merger.other_rev_id:
71
note(gettext("Nothing to do."))
63
note("Nothing to do.")
73
65
merger.merge_type = merge_type
74
66
merger.show_base = show_base