1
# Copyright (C) 2005, 2006, 2007, 2009, 2010, 2011 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
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
18
This contains functionality for installing bundles into repositories
21
from __future__ import absolute_import
24
from ..i18n import gettext
25
from ..merge import Merger
26
from ..progress import ProgressPhase
27
from ..trace import note
28
from ..bzr.vf_repository import install_revision
31
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
with repository.lock_write(), ui.ui_factory.nested_progress_bar() as pb:
36
real_revisions = bundle_reader.real_revisions
37
for i, revision in enumerate(reversed(real_revisions)):
38
pb.update(gettext("Install revisions"), i, len(real_revisions))
39
if repository.has_revision(revision.revision_id):
41
cset_tree = bundle_reader.revision_tree(repository,
43
install_revision(repository, revision, cset_tree)
46
def merge_bundle(reader, tree, check_clean, merge_type,
47
reprocess, show_base, change_reporter=None):
48
"""Merge a revision bundle into the current tree."""
49
with ui.ui_factory.nested_progress_bar() as pb:
50
pp = ProgressPhase("Merge phase", 6, pb)
52
install_bundle(tree.branch.repository, reader)
53
merger = Merger(tree.branch, this_tree=tree,
54
change_reporter=change_reporter)
56
merger.pp.next_phase()
57
if check_clean and tree.has_changes():
58
raise errors.UncommittedChanges(self)
59
merger.other_rev_id = reader.target
60
merger.other_tree = merger.revision_tree(reader.target)
61
merger.other_basis = reader.target
62
merger.pp.next_phase()
64
if merger.base_rev_id == merger.other_rev_id:
65
note(gettext("Nothing to do."))
67
merger.merge_type = merge_type
68
merger.show_base = show_base
69
merger.reprocess = reprocess
70
conflicts = merger.do_merge()