~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/merge.py

  • Committer: Martin Pool
  • Date: 2005-08-25 00:49:37 UTC
  • Revision ID: mbp@sourcefrog.net-20050825004937-82d58b07d948022e
- when terminated by an exception, say what it was

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005 Canonical Ltd
2
 
 
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.
7
 
 
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.
12
 
 
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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
17
 
 
18
 
import os.path
19
 
import tempfile
20
 
import shutil
21
 
import errno
22
 
from fetch import greedy_fetch
23
 
 
24
 
import bzrlib.osutils
25
 
import bzrlib.revision
26
1
from bzrlib.merge_core import merge_flex, ApplyMerge3, BackupBeforeChange
27
2
from bzrlib.changeset import generate_changeset, ExceptionConflictHandler
28
3
from bzrlib.changeset import Inventory, Diff3Merge
29
 
from bzrlib.branch import find_branch
 
4
from bzrlib import find_branch
 
5
import bzrlib.osutils
30
6
from bzrlib.errors import BzrCommandError, UnrelatedBranches
31
7
from bzrlib.delta import compare_trees
32
 
from bzrlib.trace import mutter, warning
33
 
from bzrlib.fetch import greedy_fetch
34
 
from bzrlib.revision import is_ancestor
 
8
from trace import mutter, warning
 
9
import os.path
 
10
import tempfile
 
11
import shutil
 
12
import errno
 
13
 
35
14
 
36
15
# comments from abentley on irc: merge happens in two stages, each
37
16
# of which generates a changeset object
131
110
        if not self.ignore_zero:
132
111
            print "%d conflicts encountered.\n" % self.conflicts
133
112
            
134
 
def get_tree(treespec, temp_root, label, local_branch=None):
 
113
def get_tree(treespec, temp_root, label):
135
114
    location, revno = treespec
136
115
    branch = find_branch(location)
137
116
    if revno is None:
138
 
        revision = None
 
117
        base_tree = branch.working_tree()
139
118
    elif revno == -1:
140
 
        revision = branch.last_patch()
141
 
    else:
142
 
        revision = branch.lookup_revision(revno)
143
 
    return branch, get_revid_tree(branch, revision, temp_root, label,
144
 
                                  local_branch)
145
 
 
146
 
def get_revid_tree(branch, revision, temp_root, label, local_branch):
147
 
    if revision is None:
148
 
        base_tree = branch.working_tree()
149
 
    else:
150
 
        if local_branch is not None:
151
 
            greedy_fetch(local_branch, branch, revision)
152
 
            base_tree = local_branch.revision_tree(revision)
153
 
        else:
154
 
            base_tree = branch.revision_tree(revision)
 
119
        base_tree = branch.basis_tree()
 
120
    else:
 
121
        base_tree = branch.revision_tree(branch.lookup_revision(revno))
155
122
    temp_path = os.path.join(temp_root, label)
156
123
    os.mkdir(temp_path)
157
 
    return MergeTree(base_tree, temp_path)
 
124
    return branch, MergeTree(base_tree, temp_path)
158
125
 
159
126
 
160
127
def file_exists(tree, file_id):
196
163
            return True
197
164
        return self.tree.inventory.has_id(file_id)
198
165
 
199
 
    def has_or_had_id(self, file_id):
200
 
        if file_id == self.tree.inventory.root.file_id:
201
 
            return True
202
 
        return self.tree.inventory.has_id(file_id)
203
 
 
204
166
    def readonly_path(self, id):
205
167
        if id not in self.tree:
206
168
            return None
226
188
    """Merge changes into a tree.
227
189
 
228
190
    base_revision
229
 
        tuple(path, revision) Base for three-way merge.
 
191
        Base for three-way merge.
230
192
    other_revision
231
 
        tuple(path, revision) Other revision for three-way merge.
 
193
        Other revision for three-way merge.
232
194
    this_dir
233
195
        Directory to merge changes into; '.' by default.
234
196
    check_clean
235
197
        If true, this_dir must have no uncommitted changes before the
236
198
        merge begins.
237
 
    all available ancestors of other_revision and base_revision are
238
 
    automatically pulled into the branch.
239
199
    """
240
 
    from bzrlib.revision import common_ancestor, MultipleRevisionSources
241
 
    from bzrlib.errors import NoSuchRevision
242
200
    tempdir = tempfile.mkdtemp(prefix="bzr-")
243
201
    try:
244
202
        if this_dir is None:
245
203
            this_dir = '.'
246
204
        this_branch = find_branch(this_dir)
247
 
        this_rev_id = this_branch.last_patch()
248
 
        if this_rev_id is None:
249
 
            raise BzrCommandError("This branch has no commits")
250
205
        if check_clean:
251
206
            changes = compare_trees(this_branch.working_tree(), 
252
207
                                    this_branch.basis_tree(), False)
253
208
            if changes.has_changed():
254
209
                raise BzrCommandError("Working tree has uncommitted changes.")
255
 
        other_branch, other_tree = get_tree(other_revision, tempdir, "other",
256
 
                                            this_branch)
257
 
        if other_revision[1] == -1:
258
 
            other_rev_id = other_branch.last_patch()
259
 
            other_basis = other_rev_id
260
 
        elif other_revision[1] is not None:
261
 
            other_rev_id = other_branch.lookup_revision(other_revision[1])
262
 
            other_basis = other_rev_id
263
 
        else:
264
 
            other_rev_id = None
265
 
            other_basis = other_branch.last_patch()
 
210
        other_branch, other_tree = get_tree(other_revision, tempdir, "other")
266
211
        if base_revision == [None, None]:
267
 
            base_rev_id = common_ancestor(this_rev_id, other_basis, 
268
 
                                          this_branch)
269
 
            if base_rev_id is None:
 
212
            if other_revision[1] == -1:
 
213
                o_revno = None
 
214
            else:
 
215
                o_revno = other_revision[1]
 
216
            base_revno = this_branch.common_ancestor(other_branch, 
 
217
                                                     other_revno=o_revno)[0]
 
218
            if base_revno is None:
270
219
                raise UnrelatedBranches()
271
 
            base_tree = get_revid_tree(this_branch, base_rev_id, tempdir, 
272
 
                                       "base", None)
273
 
            base_is_ancestor = True
274
 
        else:
275
 
            base_branch, base_tree = get_tree(base_revision, tempdir, "base")
276
 
            if base_revision[1] == -1:
277
 
                base_rev_id = base_branch.last_patch()
278
 
            elif base_revision[1] is None:
279
 
                base_rev_id = None
280
 
            else:
281
 
                base_rev_id = base_branch.lookup_revision(base_revision[1])
282
 
            if base_rev_id is not None:
283
 
                base_is_ancestor = is_ancestor(this_rev_id, base_rev_id, 
284
 
                                               MultipleRevisionSources(this_branch, 
285
 
                                                                       base_branch))
286
 
            else:
287
 
                base_is_ancestor = False
 
220
            base_revision = ['.', base_revno]
 
221
        base_branch, base_tree = get_tree(base_revision, tempdir, "base")
288
222
        if file_list is None:
289
223
            interesting_ids = None
290
224
        else:
304
238
        merge_inner(this_branch, other_tree, base_tree, tempdir, 
305
239
                    ignore_zero=ignore_zero, backup_files=backup_files, 
306
240
                    merge_type=merge_type, interesting_ids=interesting_ids)
307
 
        if base_is_ancestor and other_rev_id is not None\
308
 
            and other_rev_id not in this_branch.revision_history():
309
 
            this_branch.add_pending_merge(other_rev_id)
310
241
    finally:
311
242
        shutil.rmtree(tempdir)
312
243