/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/conflicts.py

  • Committer: Jelmer Vernooij
  • Date: 2018-03-01 23:44:14 UTC
  • mto: This revision was merged to the branch mainline in revision 6871.
  • Revision ID: jelmer@jelmer.uk-20180301234414-5btxmc1z8pz131ob
Fix context manager for RecordingUIFactory.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/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
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
 
# TODO: 'bzr resolve' should accept a directory name and work from that
 
17
# TODO: 'brz resolve' should accept a directory name and work from that
18
18
# point down
19
19
 
 
20
from __future__ import absolute_import
 
21
 
20
22
import os
21
23
 
22
 
from bzrlib.lazy_import import lazy_import
 
24
from .lazy_import import lazy_import
23
25
lazy_import(globals(), """
24
26
import errno
25
27
 
26
 
from bzrlib import (
27
 
    builtins,
 
28
from breezy import (
28
29
    cleanup,
29
 
    commands,
30
30
    errors,
31
31
    osutils,
32
32
    rio,
34
34
    transform,
35
35
    workingtree,
36
36
    )
 
37
from breezy.i18n import gettext, ngettext
37
38
""")
38
 
from bzrlib import (
 
39
from . import (
 
40
    cache_utf8,
 
41
    commands,
39
42
    option,
40
43
    registry,
41
44
    )
 
45
from .sixish import text_type
42
46
 
43
47
 
44
48
CONFLICT_SUFFIXES = ('.THIS', '.BASE', '.OTHER')
50
54
    Merge will do its best to combine the changes in two branches, but there
51
55
    are some kinds of problems only a human can fix.  When it encounters those,
52
56
    it will mark a conflict.  A conflict means that you need to fix something,
53
 
    before you should commit.
 
57
    before you can commit.
54
58
 
55
59
    Conflicts normally are listed as short, human-readable messages.  If --text
56
60
    is supplied, the pathnames of files with text conflicts are listed,
57
61
    instead.  (This is useful for editing all files with text conflicts.)
58
62
 
59
 
    Use bzr resolve when you have fixed a problem.
 
63
    Use brz resolve when you have fixed a problem.
60
64
    """
61
65
    takes_options = [
 
66
            'directory',
62
67
            option.Option('text',
63
68
                          help='List paths of files with text conflicts.'),
64
69
        ]
65
70
    _see_also = ['resolve', 'conflict-types']
66
71
 
67
 
    def run(self, text=False):
68
 
        wt = workingtree.WorkingTree.open_containing(u'.')[0]
 
72
    def run(self, text=False, directory=u'.'):
 
73
        wt = workingtree.WorkingTree.open_containing(directory)[0]
69
74
        for conflict in wt.conflicts():
70
75
            if text:
71
76
                if conflict.typestring != 'text conflict':
72
77
                    continue
73
78
                self.outf.write(conflict.path + '\n')
74
79
            else:
75
 
                self.outf.write(str(conflict) + '\n')
 
80
                self.outf.write(text_type(conflict) + '\n')
76
81
 
77
82
 
78
83
resolve_action_registry = registry.Registry()
79
84
 
80
85
 
81
86
resolve_action_registry.register(
82
 
    'done', 'done', 'Marks the conflict as resolved' )
 
87
    'done', 'done', 'Marks the conflict as resolved.')
83
88
resolve_action_registry.register(
84
89
    'take-this', 'take_this',
85
 
    'Resolve the conflict preserving the version in the working tree' )
 
90
    'Resolve the conflict preserving the version in the working tree.')
86
91
resolve_action_registry.register(
87
92
    'take-other', 'take_other',
88
 
    'Resolve the conflict taking the merged version into account' )
 
93
    'Resolve the conflict taking the merged version into account.')
89
94
resolve_action_registry.default_key = 'done'
90
95
 
91
96
class ResolveActionOption(option.RegistryOption):
103
108
    Merge will do its best to combine the changes in two branches, but there
104
109
    are some kinds of problems only a human can fix.  When it encounters those,
105
110
    it will mark a conflict.  A conflict means that you need to fix something,
106
 
    before you should commit.
 
111
    before you can commit.
107
112
 
108
 
    Once you have fixed a problem, use "bzr resolve" to automatically mark
109
 
    text conflicts as fixed, "bzr resolve FILE" to mark a specific conflict as
110
 
    resolved, or "bzr resolve --all" to mark all conflicts as resolved.
 
113
    Once you have fixed a problem, use "brz resolve" to automatically mark
 
114
    text conflicts as fixed, "brz resolve FILE" to mark a specific conflict as
 
115
    resolved, or "brz resolve --all" to mark all conflicts as resolved.
111
116
    """
112
117
    aliases = ['resolved']
113
118
    takes_args = ['file*']
114
119
    takes_options = [
 
120
            'directory',
115
121
            option.Option('all', help='Resolve all conflicts in this tree.'),
116
122
            ResolveActionOption(),
117
123
            ]
118
124
    _see_also = ['conflicts']
119
 
    def run(self, file_list=None, all=False, action=None):
 
125
    def run(self, file_list=None, all=False, action=None, directory=None):
120
126
        if all:
121
127
            if file_list:
122
 
                raise errors.BzrCommandError("If --all is specified,"
123
 
                                             " no FILE may be provided")
124
 
            tree = workingtree.WorkingTree.open_containing('.')[0]
 
128
                raise errors.BzrCommandError(gettext("If --all is specified,"
 
129
                                             " no FILE may be provided"))
 
130
            if directory is None:
 
131
                directory = u'.'
 
132
            tree = workingtree.WorkingTree.open_containing(directory)[0]
125
133
            if action is None:
126
134
                action = 'done'
127
135
        else:
128
 
            tree, file_list = builtins.tree_files(file_list)
 
136
            tree, file_list = workingtree.WorkingTree.open_containing_paths(
 
137
                file_list, directory)
129
138
            if file_list is None:
130
139
                if action is None:
131
140
                    # FIXME: There is a special case here related to the option
141
150
            if file_list is None:
142
151
                un_resolved, resolved = tree.auto_resolve()
143
152
                if len(un_resolved) > 0:
144
 
                    trace.note('%d conflict(s) auto-resolved.', len(resolved))
145
 
                    trace.note('Remaining conflicts:')
 
153
                    trace.note(ngettext('%d conflict auto-resolved.',
 
154
                        '%d conflicts auto-resolved.', len(resolved)),
 
155
                        len(resolved))
 
156
                    trace.note(gettext('Remaining conflicts:'))
146
157
                    for conflict in un_resolved:
147
 
                        trace.note(conflict)
 
158
                        trace.note(text_type(conflict))
148
159
                    return 1
149
160
                else:
150
 
                    trace.note('All conflicts resolved.')
 
161
                    trace.note(gettext('All conflicts resolved.'))
151
162
                    return 0
152
163
            else:
153
164
                # FIXME: This can never occur but the block above needs some
155
166
                # conflict.auto(tree) --vila 091242
156
167
                pass
157
168
        else:
158
 
            resolve(tree, file_list, action=action)
 
169
            before, after = resolve(tree, file_list, action=action)
 
170
            trace.note(ngettext('{0} conflict resolved, {1} remaining',
 
171
                                '{0} conflicts resolved, {1} remaining',
 
172
                                before-after).format(before - after, after))
159
173
 
160
174
 
161
175
def resolve(tree, paths=None, ignore_misses=False, recursive=False,
173
187
        paths do not have conflicts.
174
188
    :param action: How the conflict should be resolved,
175
189
    """
176
 
    tree.lock_tree_write()
177
 
    try:
 
190
    nb_conflicts_after = None
 
191
    with tree.lock_tree_write():
178
192
        tree_conflicts = tree.conflicts()
 
193
        nb_conflicts_before = len(tree_conflicts)
179
194
        if paths is None:
180
195
            new_conflicts = ConflictList()
181
196
            to_process = tree_conflicts
189
204
            except NotImplementedError:
190
205
                new_conflicts.append(conflict)
191
206
        try:
 
207
            nb_conflicts_after = len(new_conflicts)
192
208
            tree.set_conflicts(new_conflicts)
193
209
        except errors.UnsupportedOperation:
194
210
            pass
195
 
    finally:
196
 
        tree.unlock()
 
211
    if nb_conflicts_after is None:
 
212
        nb_conflicts_after = nb_conflicts_before
 
213
    return nb_conflicts_before, nb_conflicts_after
197
214
 
198
215
 
199
216
def restore(filename):
205
222
    try:
206
223
        osutils.rename(filename + ".THIS", filename)
207
224
        conflicted = True
208
 
    except OSError, e:
 
225
    except OSError as e:
209
226
        if e.errno != errno.ENOENT:
210
227
            raise
211
228
    try:
212
229
        os.unlink(filename + ".BASE")
213
230
        conflicted = True
214
 
    except OSError, e:
 
231
    except OSError as e:
215
232
        if e.errno != errno.ENOENT:
216
233
            raise
217
234
    try:
218
235
        os.unlink(filename + ".OTHER")
219
236
        conflicted = True
220
 
    except OSError, e:
 
237
    except OSError as e:
221
238
        if e.errno != errno.ENOENT:
222
239
            raise
223
240
    if not conflicted:
279
296
    def to_strings(self):
280
297
        """Generate strings for the provided conflicts"""
281
298
        for conflict in self:
282
 
            yield str(conflict)
 
299
            yield text_type(conflict)
283
300
 
284
301
    def remove_files(self, tree):
285
302
        """Remove the THIS, BASE and OTHER files for listed conflicts"""
336
353
        if ignore_misses is not True:
337
354
            for path in [p for p in paths if p not in selected_paths]:
338
355
                if not os.path.exists(tree.abspath(path)):
339
 
                    print "%s does not exist" % path
 
356
                    print("%s does not exist" % path)
340
357
                else:
341
 
                    print "%s is not conflicted" % path
 
358
                    print("%s is not conflicted" % path)
342
359
        return new_conflicts, selected_conflicts
343
360
 
344
361
 
350
367
 
351
368
    def __init__(self, path, file_id=None):
352
369
        self.path = path
353
 
        # warn turned off, because the factory blindly transfers the Stanza
354
 
        # values to __init__ and Stanza is purely a Unicode api.
355
 
        self.file_id = osutils.safe_file_id(file_id, warn=False)
 
370
        # the factory blindly transfers the Stanza values to __init__ and
 
371
        # Stanza is purely a Unicode api.
 
372
        if isinstance(file_id, text_type):
 
373
            file_id = cache_utf8.encode(file_id)
 
374
        self.file_id = osutils.safe_file_id(file_id)
356
375
 
357
376
    def as_stanza(self):
358
377
        s = rio.Stanza(type=self.typestring, path=self.path)
378
397
    def __ne__(self, other):
379
398
        return not self.__eq__(other)
380
399
 
381
 
    def __str__(self):
 
400
    def __unicode__(self):
382
401
        return self.format % self.__dict__
383
402
 
384
403
    def __repr__(self):
420
439
        for fname in self.associated_filenames():
421
440
            try:
422
441
                osutils.delete_any(tree.abspath(fname))
423
 
            except OSError, e:
 
442
            except OSError as e:
424
443
                if e.errno != errno.ENOENT:
425
444
                    raise
426
445
 
494
513
            raise AssertionError('bad winner: %r' % (winner,))
495
514
        if path_to_create is not None:
496
515
            tid = tt.trans_id_tree_path(path_to_create)
 
516
            tree = self._revision_tree(tt._tree, revid)
497
517
            transform.create_from_tree(
498
 
                tt, tt.trans_id_tree_path(path_to_create),
499
 
                self._revision_tree(tt._tree, revid), file_id)
 
518
                tt, tid, tree, tree.id2path(file_id), file_id=file_id)
500
519
            tt.version_file(file_id, tid)
501
 
 
 
520
        else:
 
521
            tid = tt.trans_id_file_id(file_id)
502
522
        # Adjust the path for the retained file id
503
 
        tid = tt.trans_id_file_id(file_id)
504
523
        parent_tid = tt.get_tree_parent(tid)
505
 
        tt.adjust_path(path, parent_tid, tid)
 
524
        tt.adjust_path(osutils.basename(path), parent_tid, tid)
506
525
        tt.apply()
507
526
 
508
527
    def _revision_tree(self, tree, revid):
571
590
        :param tt: The TreeTransform where the conflict is resolved.
572
591
        :param suffix_to_remove: Either 'THIS' or 'OTHER'
573
592
 
574
 
        The resolution is symmetric, when taking THIS, OTHER is deleted and
 
593
        The resolution is symmetric: when taking THIS, OTHER is deleted and
575
594
        item.THIS is renamed into item and vice-versa.
576
595
        """
577
596
        try:
584
603
            # never existed or was already deleted (including the case
585
604
            # where the user deleted it)
586
605
            pass
587
 
        # Rename 'item.suffix_to_remove' (note that if
588
 
        # 'item.suffix_to_remove' has been deleted, this is a no-op)
589
 
        this_tid = tt.trans_id_file_id(self.file_id)
590
 
        parent_tid = tt.get_tree_parent(this_tid)
591
 
        tt.adjust_path(self.path, parent_tid, this_tid)
592
 
        tt.apply()
 
606
        try:
 
607
            this_path = tt._tree.id2path(self.file_id)
 
608
        except errors.NoSuchId:
 
609
            # The file is not present anymore. This may happen if the user
 
610
            # deleted the file either manually or when resolving a conflict on
 
611
            # the parent.  We may raise some exception to indicate that the
 
612
            # conflict doesn't exist anymore and as such doesn't need to be
 
613
            # resolved ? -- vila 20110615 
 
614
            this_tid = None
 
615
        else:
 
616
            this_tid = tt.trans_id_tree_path(this_path)
 
617
        if this_tid is not None:
 
618
            # Rename 'item.suffix_to_remove' (note that if
 
619
            # 'item.suffix_to_remove' has been deleted, this is a no-op)
 
620
            parent_tid = tt.get_tree_parent(this_tid)
 
621
            tt.adjust_path(osutils.basename(self.path), parent_tid, this_tid)
 
622
            tt.apply()
593
623
 
594
624
    def action_take_this(self, tree):
595
625
        self._resolve_with_cleanups(tree, 'OTHER')
598
628
        self._resolve_with_cleanups(tree, 'THIS')
599
629
 
600
630
 
601
 
# FIXME: TextConflict is about a single file-id, there never is a conflict_path
602
 
# attribute so we shouldn't inherit from PathConflict but simply from Conflict
603
 
 
604
631
# TODO: There should be a base revid attribute to better inform the user about
605
632
# how the conflicts were generated.
606
 
class TextConflict(PathConflict):
 
633
class TextConflict(Conflict):
607
634
    """The merge algorithm could not resolve all differences encountered."""
608
635
 
609
636
    has_files = True
612
639
 
613
640
    format = 'Text conflict in %(path)s'
614
641
 
 
642
    rformat = '%(class)s(%(path)r, %(file_id)r)'
 
643
 
615
644
    def associated_filenames(self):
616
645
        return [self.path + suffix for suffix in CONFLICT_SUFFIXES]
617
646
 
 
647
    def _resolve(self, tt, winner_suffix):
 
648
        """Resolve the conflict by copying one of .THIS or .OTHER into file.
 
649
 
 
650
        :param tt: The TreeTransform where the conflict is resolved.
 
651
        :param winner_suffix: Either 'THIS' or 'OTHER'
 
652
 
 
653
        The resolution is symmetric, when taking THIS, item.THIS is renamed
 
654
        into item and vice-versa. This takes one of the files as a whole
 
655
        ignoring every difference that could have been merged cleanly.
 
656
        """
 
657
        # To avoid useless copies, we switch item and item.winner_suffix, only
 
658
        # item will exist after the conflict has been resolved anyway.
 
659
        item_tid = tt.trans_id_file_id(self.file_id)
 
660
        item_parent_tid = tt.get_tree_parent(item_tid)
 
661
        winner_path = self.path + '.' + winner_suffix
 
662
        winner_tid = tt.trans_id_tree_path(winner_path)
 
663
        winner_parent_tid = tt.get_tree_parent(winner_tid)
 
664
        # Switch the paths to preserve the content
 
665
        tt.adjust_path(osutils.basename(self.path),
 
666
                       winner_parent_tid, winner_tid)
 
667
        tt.adjust_path(osutils.basename(winner_path), item_parent_tid, item_tid)
 
668
        # Associate the file_id to the right content
 
669
        tt.unversion_file(item_tid)
 
670
        tt.version_file(self.file_id, winner_tid)
 
671
        tt.apply()
 
672
 
 
673
    def action_take_this(self, tree):
 
674
        self._resolve_with_cleanups(tree, 'THIS')
 
675
 
 
676
    def action_take_other(self, tree):
 
677
        self._resolve_with_cleanups(tree, 'OTHER')
 
678
 
618
679
 
619
680
class HandledConflict(Conflict):
620
681
    """A path problem that has been provisionally resolved.
652
713
                 conflict_file_id=None):
653
714
        HandledConflict.__init__(self, action, path, file_id)
654
715
        self.conflict_path = conflict_path
655
 
        # warn turned off, because the factory blindly transfers the Stanza
656
 
        # values to __init__.
657
 
        self.conflict_file_id = osutils.safe_file_id(conflict_file_id,
658
 
                                                     warn=False)
 
716
        # the factory blindly transfers the Stanza values to __init__,
 
717
        # so they can be unicode.
 
718
        if isinstance(conflict_file_id, text_type):
 
719
            conflict_file_id = cache_utf8.encode(conflict_file_id)
 
720
        self.conflict_file_id = osutils.safe_file_id(conflict_file_id)
659
721
 
660
722
    def _cmp_list(self):
661
723
        return HandledConflict._cmp_list(self) + [self.conflict_path,
709
771
    format = 'Conflict moving %(path)s into %(conflict_path)s. %(action)s.'
710
772
 
711
773
    def action_take_this(self, tree):
712
 
        # just acccept bzr proposal
 
774
        # just acccept brz proposal
713
775
        pass
714
776
 
715
777
    def action_take_other(self, tree):
716
 
        # FIXME: We shouldn't have to manipulate so many paths here (and there
717
 
        # is probably a bug or two...)
718
 
        base_path = osutils.basename(self.path)
719
 
        conflict_base_path = osutils.basename(self.conflict_path)
720
778
        tt = transform.TreeTransform(tree)
721
779
        try:
722
780
            p_tid = tt.trans_id_file_id(self.file_id)
723
781
            parent_tid = tt.get_tree_parent(p_tid)
724
782
            cp_tid = tt.trans_id_file_id(self.conflict_file_id)
725
783
            cparent_tid = tt.get_tree_parent(cp_tid)
726
 
            tt.adjust_path(base_path, cparent_tid, cp_tid)
727
 
            tt.adjust_path(conflict_base_path, parent_tid, p_tid)
 
784
            tt.adjust_path(osutils.basename(self.path), cparent_tid, cp_tid)
 
785
            tt.adjust_path(osutils.basename(self.conflict_path),
 
786
                           parent_tid, p_tid)
728
787
            tt.apply()
729
788
        finally:
730
789
            tt.finalize()
766
825
        tree.remove([self.path], force=True, keep_files=False)
767
826
 
768
827
    def action_take_other(self, tree):
769
 
        # just acccept bzr proposal
 
828
        # just acccept brz proposal
770
829
        pass
771
830
 
772
831
 
785
844
    # MissingParent from the *user* pov.
786
845
 
787
846
    def action_take_this(self, tree):
788
 
        # just acccept bzr proposal
 
847
        # just acccept brz proposal
789
848
        pass
790
849
 
791
850
    def action_take_other(self, tree):