/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: 2017-09-01 07:15:43 UTC
  • mfrom: (6770.3.2 py3_test_cleanup)
  • Revision ID: jelmer@jelmer.uk-20170901071543-1t83321xkog9qrxh
Merge lp:~gz/brz/py3_test_cleanup

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