← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/conflicts.py

merge trunk

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/conflicts.py
1
 
# Copyright (C) 2005, 2006, 2007, 2009, 2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2007, 2009, 2010, 2011 Canonical Ltd
2
2
#
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
117
117
            ResolveActionOption(),
118
118
            ]
119
119
    _see_also = ['conflicts']
120
 
    def run(self, file_list=None, all=False, action=None, directory=u'.'):
 
120
    def run(self, file_list=None, all=False, action=None, directory=None):
121
121
        if all:
122
122
            if file_list:
123
123
                raise errors.BzrCommandError("If --all is specified,"
124
124
                                             " no FILE may be provided")
 
125
            if directory is None:
 
126
                directory = u'.'
125
127
            tree = workingtree.WorkingTree.open_containing(directory)[0]
126
128
            if action is None:
127
129
                action = 'done'
128
130
        else:
129
131
            tree, file_list = workingtree.WorkingTree.open_containing_paths(
130
 
                file_list)
 
132
                file_list, directory)
131
133
            if file_list is None:
132
134
                if action is None:
133
135
                    # FIXME: There is a special case here related to the option
157
159
                # conflict.auto(tree) --vila 091242
158
160
                pass
159
161
        else:
160
 
            resolve(tree, file_list, action=action)
 
162
            before, after = resolve(tree, file_list, action=action)
 
163
            trace.note('%d conflict(s) resolved, %d remaining'
 
164
                       % (before - after, after))
161
165
 
162
166
 
163
167
def resolve(tree, paths=None, ignore_misses=False, recursive=False,
176
180
    :param action: How the conflict should be resolved,
177
181
    """
178
182
    tree.lock_tree_write()
 
183
    nb_conflicts_after = None
179
184
    try:
180
185
        tree_conflicts = tree.conflicts()
 
186
        nb_conflicts_before = len(tree_conflicts)
181
187
        if paths is None:
182
188
            new_conflicts = ConflictList()
183
189
            to_process = tree_conflicts
191
197
            except NotImplementedError:
192
198
                new_conflicts.append(conflict)
193
199
        try:
 
200
            nb_conflicts_after = len(new_conflicts)
194
201
            tree.set_conflicts(new_conflicts)
195
202
        except errors.UnsupportedOperation:
196
203
            pass
197
204
    finally:
198
205
        tree.unlock()
 
206
    if nb_conflicts_after is None:
 
207
        nb_conflicts_after = nb_conflicts_before
 
208
    return nb_conflicts_before, nb_conflicts_after
199
209
 
200
210
 
201
211
def restore(filename):
504
514
        # Adjust the path for the retained file id
505
515
        tid = tt.trans_id_file_id(file_id)
506
516
        parent_tid = tt.get_tree_parent(tid)
507
 
        tt.adjust_path(path, parent_tid, tid)
 
517
        tt.adjust_path(osutils.basename(path), parent_tid, tid)
508
518
        tt.apply()
509
519
 
510
520
    def _revision_tree(self, tree, revid):
590
600
        # 'item.suffix_to_remove' has been deleted, this is a no-op)
591
601
        this_tid = tt.trans_id_file_id(self.file_id)
592
602
        parent_tid = tt.get_tree_parent(this_tid)
593
 
        tt.adjust_path(self.path, parent_tid, this_tid)
 
603
        tt.adjust_path(osutils.basename(self.path), parent_tid, this_tid)
594
604
        tt.apply()
595
605
 
596
606
    def action_take_this(self, tree):
600
610
        self._resolve_with_cleanups(tree, 'THIS')
601
611
 
602
612
 
603
 
# FIXME: TextConflict is about a single file-id, there never is a conflict_path
604
 
# attribute so we shouldn't inherit from PathConflict but simply from Conflict
605
 
 
606
613
# TODO: There should be a base revid attribute to better inform the user about
607
614
# how the conflicts were generated.
608
 
class TextConflict(PathConflict):
 
615
class TextConflict(Conflict):
609
616
    """The merge algorithm could not resolve all differences encountered."""
610
617
 
611
618
    has_files = True
614
621
 
615
622
    format = 'Text conflict in %(path)s'
616
623
 
 
624
    rformat = '%(class)s(%(path)r, %(file_id)r)'
 
625
 
617
626
    def associated_filenames(self):
618
627
        return [self.path + suffix for suffix in CONFLICT_SUFFIXES]
619
628
 
 
629
    def _resolve(self, tt, winner_suffix):
 
630
        """Resolve the conflict by copying one of .THIS or .OTHER into file.
 
631
 
 
632
        :param tt: The TreeTransform where the conflict is resolved.
 
633
        :param winner_suffix: Either 'THIS' or 'OTHER'
 
634
 
 
635
        The resolution is symmetric, when taking THIS, item.THIS is renamed
 
636
        into item and vice-versa. This takes one of the files as a whole
 
637
        ignoring every difference that could have been merged cleanly.
 
638
        """
 
639
        # To avoid useless copies, we switch item and item.winner_suffix, only
 
640
        # item will exist after the conflict has been resolved anyway.
 
641
        item_tid = tt.trans_id_file_id(self.file_id)
 
642
        item_parent_tid = tt.get_tree_parent(item_tid)
 
643
        winner_path = self.path + '.' + winner_suffix
 
644
        winner_tid = tt.trans_id_tree_path(winner_path)
 
645
        winner_parent_tid = tt.get_tree_parent(winner_tid)
 
646
        # Switch the paths to preserve the content
 
647
        tt.adjust_path(osutils.basename(self.path),
 
648
                       winner_parent_tid, winner_tid)
 
649
        tt.adjust_path(osutils.basename(winner_path), item_parent_tid, item_tid)
 
650
        # Associate the file_id to the right content
 
651
        tt.unversion_file(item_tid)
 
652
        tt.version_file(self.file_id, winner_tid)
 
653
        tt.apply()
 
654
 
 
655
    def action_take_this(self, tree):
 
656
        self._resolve_with_cleanups(tree, 'THIS')
 
657
 
 
658
    def action_take_other(self, tree):
 
659
        self._resolve_with_cleanups(tree, 'OTHER')
 
660
 
620
661
 
621
662
class HandledConflict(Conflict):
622
663
    """A path problem that has been provisionally resolved.
715
756
        pass
716
757
 
717
758
    def action_take_other(self, tree):
718
 
        # FIXME: We shouldn't have to manipulate so many paths here (and there
719
 
        # is probably a bug or two...)
720
 
        base_path = osutils.basename(self.path)
721
 
        conflict_base_path = osutils.basename(self.conflict_path)
722
759
        tt = transform.TreeTransform(tree)
723
760
        try:
724
761
            p_tid = tt.trans_id_file_id(self.file_id)
725
762
            parent_tid = tt.get_tree_parent(p_tid)
726
763
            cp_tid = tt.trans_id_file_id(self.conflict_file_id)
727
764
            cparent_tid = tt.get_tree_parent(cp_tid)
728
 
            tt.adjust_path(base_path, cparent_tid, cp_tid)
729
 
            tt.adjust_path(conflict_base_path, parent_tid, p_tid)
 
765
            tt.adjust_path(osutils.basename(self.path), cparent_tid, cp_tid)
 
766
            tt.adjust_path(osutils.basename(self.conflict_path),
 
767
                           parent_tid, p_tid)
730
768
            tt.apply()
731
769
        finally:
732
770
            tt.finalize()