← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/shelf_ui.py

  • Committer: Vincent Ladeuil
  • Date: 2011-06-15 11:36:05 UTC
  • mto: This revision was merged to the branch mainline in revision 5975.
  • Revision ID: v.ladeuil+lp@free.fr-20110615113605-p7zyyfry9wy1hquc
https://launchpad.net/bugs/788000
Make ContentConflict resolution more robust

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/shelf_ui.py
16
16
 
17
17
 
18
18
from cStringIO import StringIO
19
 
import os
20
19
import shutil
21
20
import sys
22
21
import tempfile
35
34
    ui,
36
35
    workingtree,
37
36
)
38
 
from bzrlib.i18n import gettext
 
37
 
39
38
 
40
39
class UseEditor(Exception):
41
40
    """Use an editor instead of selecting hunks."""
43
42
 
44
43
class ShelfReporter(object):
45
44
 
46
 
    vocab = {'add file': gettext('Shelve adding file "%(path)s"?'),
47
 
             'binary': gettext('Shelve binary changes?'),
48
 
             'change kind': gettext('Shelve changing "%s" from %(other)s'
49
 
             ' to %(this)s?'),
50
 
             'delete file': gettext('Shelve removing file "%(path)s"?'),
51
 
             'final': gettext('Shelve %d change(s)?'),
52
 
             'hunk': gettext('Shelve?'),
53
 
             'modify target': gettext('Shelve changing target of'
54
 
             ' "%(path)s" from "%(other)s" to "%(this)s"?'),
55
 
             'rename': gettext('Shelve renaming "%(other)s" =>'
56
 
                        ' "%(this)s"?')
 
45
    vocab = {'add file': 'Shelve adding file "%(path)s"?',
 
46
             'binary': 'Shelve binary changes?',
 
47
             'change kind': 'Shelve changing "%s" from %(other)s'
 
48
             ' to %(this)s?',
 
49
             'delete file': 'Shelve removing file "%(path)s"?',
 
50
             'final': 'Shelve %d change(s)?',
 
51
             'hunk': 'Shelve?',
 
52
             'modify target': 'Shelve changing target of'
 
53
             ' "%(path)s" from "%(other)s" to "%(this)s"?',
 
54
             'rename': 'Shelve renaming "%(other)s" =>'
 
55
                        ' "%(this)s"?'
57
56
             }
58
57
 
59
58
    invert_diff = False
67
66
 
68
67
    def shelved_id(self, shelf_id):
69
68
        """Report the id changes were shelved to."""
70
 
        trace.note(gettext('Changes shelved with id "%d".') % shelf_id)
 
69
        trace.note('Changes shelved with id "%d".' % shelf_id)
71
70
 
72
71
    def changes_destroyed(self):
73
72
        """Report that changes were made without shelving."""
74
 
        trace.note(gettext('Selected changes destroyed.'))
 
73
        trace.note('Selected changes destroyed.')
75
74
 
76
75
    def selected_changes(self, transform):
77
76
        """Report the changes that were selected."""
78
 
        trace.note(gettext("Selected changes:"))
 
77
        trace.note("Selected changes:")
79
78
        changes = transform.iter_changes()
80
79
        delta.report_changes(changes, self.delta_reporter)
81
80
 
95
94
 
96
95
class ApplyReporter(ShelfReporter):
97
96
 
98
 
    vocab = {'add file': gettext('Delete file "%(path)s"?'),
99
 
             'binary': gettext('Apply binary changes?'),
100
 
             'change kind': gettext('Change "%(path)s" from %(this)s'
101
 
             ' to %(other)s?'),
102
 
             'delete file': gettext('Add file "%(path)s"?'),
103
 
             'final': gettext('Apply %d change(s)?'),
104
 
             'hunk': gettext('Apply change?'),
105
 
             'modify target': gettext('Change target of'
106
 
             ' "%(path)s" from "%(this)s" to "%(other)s"?'),
107
 
             'rename': gettext('Rename "%(this)s" => "%(other)s"?'),
 
97
    vocab = {'add file': 'Delete file "%(path)s"?',
 
98
             'binary': 'Apply binary changes?',
 
99
             'change kind': 'Change "%(path)s" from %(this)s'
 
100
             ' to %(other)s?',
 
101
             'delete file': 'Add file "%(path)s"?',
 
102
             'final': 'Apply %d change(s)?',
 
103
             'hunk': 'Apply change?',
 
104
             'modify target': 'Change target of'
 
105
             ' "%(path)s" from "%(this)s" to "%(other)s"?',
 
106
             'rename': 'Rename "%(this)s" => "%(other)s"?',
108
107
             }
109
108
 
110
109
    invert_diff = True
252
251
        diff_file.seek(0)
253
252
        return patches.parse_patch(diff_file)
254
253
 
255
 
    def prompt(self, message, choices, default):
256
 
        return ui.ui_factory.choose(message, choices, default=default)
257
 
 
258
 
    def prompt_bool(self, question, allow_editor=False):
 
254
    def prompt(self, message):
 
255
        """Prompt the user for a character.
 
256
 
 
257
        :param message: The message to prompt a user with.
 
258
        :return: A character.
 
259
        """
 
260
        if not sys.stdin.isatty():
 
261
            # Since there is no controlling terminal we will hang when trying
 
262
            # to prompt the user, better abort now.  See
 
263
            # https://code.launchpad.net/~bialix/bzr/shelve-no-tty/+merge/14905
 
264
            # for more context.
 
265
            raise errors.BzrError("You need a controlling terminal.")
 
266
        sys.stdout.write(message)
 
267
        char = osutils.getchar()
 
268
        sys.stdout.write("\r" + ' ' * len(message) + '\r')
 
269
        sys.stdout.flush()
 
270
        return char
 
271
 
 
272
    def prompt_bool(self, question, long=False, allow_editor=False):
259
273
        """Prompt the user with a yes/no question.
260
274
 
261
275
        This may be overridden by self.auto.  It may also *set* self.auto.  It
265
279
        """
266
280
        if self.auto:
267
281
            return True
268
 
        alternatives_chars = 'yn'
269
 
        alternatives = '&yes\n&No'
270
 
        if allow_editor:
271
 
            alternatives_chars += 'e'
272
 
            alternatives += '\n&edit manually'
273
 
        alternatives_chars += 'fq'
274
 
        alternatives += '\n&finish\n&quit'
275
 
        choice = self.prompt(question, alternatives, 1)
276
 
        if choice is None:
277
 
            # EOF.
278
 
            char = 'n'
 
282
        editor_string = ''
 
283
        if long:
 
284
            if allow_editor:
 
285
                editor_string = '(E)dit manually, '
 
286
            prompt = ' [(y)es, (N)o, %s(f)inish, or (q)uit]' % editor_string
279
287
        else:
280
 
            char = alternatives_chars[choice]
 
288
            if allow_editor:
 
289
                editor_string = 'e'
 
290
            prompt = ' [yN%sfq?]' % editor_string
 
291
        char = self.prompt(question + prompt)
281
292
        if char == 'y':
282
293
            return True
283
294
        elif char == 'e' and allow_editor:
285
296
        elif char == 'f':
286
297
            self.auto = True
287
298
            return True
 
299
        elif char == '?':
 
300
            return self.prompt_bool(question, long=True)
288
301
        if char == 'q':
289
302
            raise errors.UserAbort()
290
303
        else:
399
412
            else:
400
413
                shelf_id = manager.last_shelf()
401
414
                if shelf_id is None:
402
 
                    raise errors.BzrCommandError(gettext('No changes are shelved.'))
 
415
                    raise errors.BzrCommandError('No changes are shelved.')
403
416
            apply_changes = True
404
417
            delete_shelf = True
405
418
            read_shelf = True
457
470
        cleanups = [self.tree.unlock]
458
471
        try:
459
472
            if self.read_shelf:
460
 
                trace.note(gettext('Using changes with id "%d".') % self.shelf_id)
 
473
                trace.note('Using changes with id "%d".' % self.shelf_id)
461
474
                unshelver = self.manager.get_unshelver(self.shelf_id)
462
475
                cleanups.append(unshelver.finalize)
463
476
                if unshelver.message is not None:
464
 
                    trace.note(gettext('Message: %s') % unshelver.message)
 
477
                    trace.note('Message: %s' % unshelver.message)
465
478
                change_reporter = delta._ChangeReporter()
466
479
                merger = unshelver.make_merger(None)
467
480
                merger.change_reporter = change_reporter
473
486
                    self.show_changes(merger)
474
487
            if self.delete_shelf:
475
488
                self.manager.delete_shelf(self.shelf_id)
476
 
                trace.note(gettext('Deleted changes with id "%d".') % self.shelf_id)
 
489
                trace.note('Deleted changes with id "%d".' % self.shelf_id)
477
490
        finally:
478
491
            for cleanup in reversed(cleanups):
479
492
                cleanup()