← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/shelf_ui.py

  • Committer: Patch Queue Manager
  • Date: 2011-10-06 08:34:03 UTC
  • mfrom: (6191.2.1 843900-url-nameerror)
  • Revision ID: pqm@pqm.ubuntu.com-20111006083403-jnsw0exlirg01aed
(mbp) error message without traceback on invalid ubuntu/debian url (bug
 843900) (Martin Pool)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/shelf_ui.py
252
252
        diff_file.seek(0)
253
253
        return patches.parse_patch(diff_file)
254
254
 
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):
 
255
    def _char_based(self):
 
256
        # FIXME: A bit hackish to use INSIDE_EMACS here, but there is another
 
257
        # work in progress moving this method (and more importantly prompt()
 
258
        # below) into the ui area and address the issue in better ways.
 
259
        # -- vila 2011-09-28
 
260
        return os.environ.get('INSIDE_EMACS', None) is None
 
261
 
 
262
    def prompt(self, message):
 
263
        """Prompt the user for a character.
 
264
 
 
265
        :param message: The message to prompt a user with.
 
266
        :return: A character.
 
267
        """
 
268
        char_based = self._char_based()
 
269
        if char_based and not sys.stdin.isatty():
 
270
            # Since there is no controlling terminal we will hang when
 
271
            # trying to prompt the user, better abort now.  See
 
272
            # https://code.launchpad.net/~bialix/bzr/shelve-no-tty/+merge/14905
 
273
            # for more context.
 
274
            raise errors.BzrError(gettext("You need a controlling terminal."))
 
275
        sys.stdout.write(message)
 
276
        if char_based:
 
277
            # We peek one char at a time which requires a real term here
 
278
            char = osutils.getchar()
 
279
        else:
 
280
            # While running tests (or under emacs) the input is line buffered
 
281
            # so we must not use osutils.getchar(). Instead we switch to a mode
 
282
            # where each line is terminated by a new line
 
283
            line = sys.stdin.readline()
 
284
            if line:
 
285
                # XXX: Warn if more than one char is typed ?
 
286
                char = line[0]
 
287
            else:
 
288
                # Empty input, callers handle it as enter
 
289
                char = ''
 
290
        sys.stdout.write("\r" + ' ' * len(message) + '\r')
 
291
        sys.stdout.flush()
 
292
        return char
 
293
 
 
294
    def prompt_bool(self, question, long=False, allow_editor=False):
259
295
        """Prompt the user with a yes/no question.
260
296
 
261
297
        This may be overridden by self.auto.  It may also *set* self.auto.  It
265
301
        """
266
302
        if self.auto:
267
303
            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'
 
304
        editor_string = ''
 
305
        if long:
 
306
            if allow_editor:
 
307
                editor_string = '(E)dit manually, '
 
308
            prompt = ' [(y)es, (N)o, %s(f)inish, or (q)uit]' % editor_string
279
309
        else:
280
 
            char = alternatives_chars[choice]
 
310
            if allow_editor:
 
311
                editor_string = 'e'
 
312
            prompt = ' [yN%sfq?]' % editor_string
 
313
        char = self.prompt(question + prompt)
281
314
        if char == 'y':
282
315
            return True
283
316
        elif char == 'e' and allow_editor:
285
318
        elif char == 'f':
286
319
            self.auto = True
287
320
            return True
 
321
        elif char == '?':
 
322
            return self.prompt_bool(question, long=True)
288
323
        if char == 'q':
289
324
            raise errors.UserAbort()
290
325
        else: