~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/ui/__init__.py

  • Committer: John Arbash Meinel
  • Date: 2009-08-17 15:38:52 UTC
  • mfrom: (4615.1.1 2.0b1-merge-sort)
  • mto: This revision was merged to the branch mainline in revision 4629.
  • Revision ID: john@arbash-meinel.com-20090817153852-9127xf3aabmcoegt
Bring in the changes to merge_sort.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2011 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2007, 2008, 2009 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
22
22
Several levels are supported, and you can also register new factories such as
23
23
for a GUI.
24
24
 
25
 
bzrlib.ui.UIFactory
 
25
UIFactory
26
26
    Semi-abstract base class
27
27
 
28
 
bzrlib.ui.SilentUIFactory
 
28
SilentUIFactory
29
29
    Produces no output and cannot take any input; useful for programs using
30
30
    bzrlib in batch mode or for programs such as loggerhead.
31
31
 
32
 
bzrlib.ui.CannedInputUIFactory
 
32
CannedInputUIFactory
33
33
    For use in testing; the input values to be returned are provided 
34
34
    at construction.
35
35
 
36
 
bzrlib.ui.text.TextUIFactory
 
36
TextUIFactory
37
37
    Standard text command-line interface, with stdin, stdout, stderr.
38
38
    May make more or less advanced use of them, eg in drawing progress bars,
39
39
    depending on the detected capabilities of the terminal.
42
42
"""
43
43
 
44
44
 
 
45
import os
 
46
import sys
45
47
import warnings
46
48
 
47
49
from bzrlib.lazy_import import lazy_import
48
50
lazy_import(globals(), """
 
51
import getpass
 
52
 
49
53
from bzrlib import (
 
54
    errors,
50
55
    osutils,
51
56
    progress,
52
57
    trace,
53
58
    )
54
59
""")
 
60
from bzrlib.symbol_versioning import (
 
61
    deprecated_function,
 
62
    deprecated_in,
 
63
    deprecated_method,
 
64
    )
55
65
 
56
66
 
57
67
_valid_boolean_strings = dict(yes=True, no=False,
90
100
    return val
91
101
 
92
102
 
93
 
class ConfirmationUserInterfacePolicy(object):
94
 
    """Wrapper for a UIFactory that allows or denies all confirmed actions."""
95
 
 
96
 
    def __init__(self, wrapped_ui, default_answer, specific_answers):
97
 
        """Generate a proxy UI that does no confirmations.
98
 
 
99
 
        :param wrapped_ui: Underlying UIFactory.
100
 
        :param default_answer: Bool for whether requests for
101
 
            confirmation from the user should be noninteractively accepted or
102
 
            denied.
103
 
        :param specific_answers: Map from confirmation_id to bool answer.
104
 
        """
105
 
        self.wrapped_ui = wrapped_ui
106
 
        self.default_answer = default_answer
107
 
        self.specific_answers = specific_answers
108
 
 
109
 
    def __getattr__(self, name):
110
 
        return getattr(self.wrapped_ui, name)
111
 
 
112
 
    def __repr__(self):
113
 
        return '%s(%r, %r, %r)' % (
114
 
            self.__class__.__name__,
115
 
            self.wrapped_ui,
116
 
            self.default_answer, 
117
 
            self.specific_answers)
118
 
 
119
 
    def confirm_action(self, prompt, confirmation_id, prompt_kwargs):
120
 
        if confirmation_id in self.specific_answers:
121
 
            return self.specific_answers[confirmation_id]
122
 
        elif self.default_answer is not None:
123
 
            return self.default_answer
124
 
        else:
125
 
            return self.wrapped_ui.confirm_action(
126
 
                prompt, confirmation_id, prompt_kwargs)
127
 
 
128
 
 
129
103
class UIFactory(object):
130
104
    """UI abstraction.
131
105
 
132
106
    This tells the library how to display things to the user.  Through this
133
107
    layer different applications can choose the style of UI.
134
 
 
135
 
    UI Factories are also context managers, for some syntactic sugar some users
136
 
    need.
137
 
 
138
 
    :ivar suppressed_warnings: Identifiers for user warnings that should 
139
 
        no be emitted.
140
108
    """
141
109
 
142
 
    _user_warning_templates = dict(
143
 
        cross_format_fetch=("Doing on-the-fly conversion from "
144
 
            "%(from_format)s to %(to_format)s.\n"
145
 
            "This may take some time. Upgrade the repositories to the "
146
 
            "same format for better performance."
147
 
            ),
148
 
        deprecated_command=(
149
 
            "The command 'bzr %(deprecated_name)s' "
150
 
            "has been deprecated in bzr %(deprecated_in_version)s. "
151
 
            "Please use 'bzr %(recommended_name)s' instead."),
152
 
        recommend_upgrade=("%(current_format_name)s is deprecated "
153
 
            "and a better format is available.\n"
154
 
            "It is recommended that you upgrade by "
155
 
            "running the command\n"
156
 
            "  bzr upgrade %(basedir)s"),
157
 
        locks_steal_dead=(
158
 
            u"Stole dead lock %(lock_url)s %(other_holder_info)s."),
159
 
        )
160
 
 
161
110
    def __init__(self):
162
111
        self._task_stack = []
163
 
        self.suppressed_warnings = set()
164
 
        self._quiet = False
165
 
 
166
 
    def __enter__(self):
167
 
        """Context manager entry support.
168
 
 
169
 
        Override in a concrete factory class if initialisation before use is
170
 
        needed.
171
 
        """
172
 
        return self # This is bound to the 'as' clause in a with statement.
173
 
 
174
 
    def __exit__(self, exc_type, exc_val, exc_tb):
175
 
        """Context manager exit support.
176
 
 
177
 
        Override in a concrete factory class if more cleanup than a simple
178
 
        self.clear_term() is needed when the UIFactory is finished with.
179
 
        """
180
 
        self.clear_term()
181
 
        return False # propogate exceptions.
182
 
 
183
 
    def be_quiet(self, state):
184
 
        """Tell the UI to be more quiet, or not.
185
 
 
186
 
        Typically this suppresses progress bars; the application may also look
187
 
        at ui_factory.is_quiet().
188
 
        """
189
 
        self._quiet = state
190
 
 
191
 
    def confirm_action(self, prompt, confirmation_id, prompt_kwargs):
192
 
        """Seek user confirmation for an action.
193
 
 
194
 
        If the UI is noninteractive, or the user does not want to be asked
195
 
        about this action, True is returned, indicating bzr should just
196
 
        proceed.
197
 
 
198
 
        The confirmation id allows the user to configure certain actions to
199
 
        always be confirmed or always denied, and for UIs to specialize the
200
 
        display of particular confirmations.
201
 
 
202
 
        :param prompt: Suggested text to display to the user.
203
 
        :param prompt_kwargs: A dictionary of arguments that can be
204
 
            string-interpolated into the prompt.
205
 
        :param confirmation_id: Unique string identifier for the confirmation.
206
 
        """
207
 
        return self.get_boolean(prompt % prompt_kwargs)
208
 
 
209
 
    def get_password(self, prompt=u'', **kwargs):
 
112
 
 
113
    def get_password(self, prompt='', **kwargs):
210
114
        """Prompt the user for a password.
211
115
 
212
 
        :param prompt: The prompt to present the user (must be unicode)
 
116
        :param prompt: The prompt to present the user
213
117
        :param kwargs: Arguments which will be expanded into the prompt.
214
118
                       This lets front ends display different things if
215
119
                       they so choose.
221
125
        """
222
126
        raise NotImplementedError(self.get_password)
223
127
 
224
 
    def is_quiet(self):
225
 
        return self._quiet
226
 
 
227
 
    def make_output_stream(self, encoding=None, encoding_type=None):
228
 
        """Get a stream for sending out bulk text data.
229
 
 
230
 
        This is used for commands that produce bulk text, such as log or diff
231
 
        output, as opposed to user interaction.  This should work even for
232
 
        non-interactive user interfaces.  Typically this goes to a decorated
233
 
        version of stdout, but in a GUI it might be appropriate to send it to a 
234
 
        window displaying the text.
235
 
     
236
 
        :param encoding: Unicode encoding for output; if not specified 
237
 
            uses the configured 'output_encoding' if any; otherwise the 
238
 
            terminal encoding. 
239
 
            (See get_terminal_encoding.)
240
 
 
241
 
        :param encoding_type: How to handle encoding errors:
242
 
            replace/strict/escape/exact.  Default is replace.
243
 
        """
244
 
        # XXX: is the caller supposed to close the resulting object?
245
 
        if encoding is None:
246
 
            from bzrlib import config
247
 
            encoding = config.GlobalConfig().get_user_option(
248
 
                'output_encoding')
249
 
        if encoding is None:
250
 
            encoding = osutils.get_terminal_encoding(trace=True)
251
 
        if encoding_type is None:
252
 
            encoding_type = 'replace'
253
 
        out_stream = self._make_output_stream_explicit(encoding, encoding_type)
254
 
        return out_stream
255
 
 
256
 
    def _make_output_stream_explicit(self, encoding, encoding_type):
257
 
        raise NotImplementedError("%s doesn't support make_output_stream"
258
 
            % (self.__class__.__name__))
259
 
 
260
128
    def nested_progress_bar(self):
261
129
        """Return a nested progress bar.
262
130
 
275
143
        if not self._task_stack:
276
144
            warnings.warn("%r finished but nothing is active"
277
145
                % (task,))
278
 
        if task in self._task_stack:
279
 
            self._task_stack.remove(task)
 
146
        elif task != self._task_stack[-1]:
 
147
            warnings.warn("%r is not the active task %r"
 
148
                % (task, self._task_stack[-1]))
280
149
        else:
281
 
            warnings.warn("%r is not in active stack %r"
282
 
                % (task, self._task_stack))
 
150
            del self._task_stack[-1]
283
151
        if not self._task_stack:
284
152
            self._progress_all_finished()
285
153
 
302
170
        """
303
171
        pass
304
172
 
305
 
    def format_user_warning(self, warning_id, message_args):
306
 
        try:
307
 
            template = self._user_warning_templates[warning_id]
308
 
        except KeyError:
309
 
            fail = "bzr warning: %r, %r" % (warning_id, message_args)
310
 
            warnings.warn("no template for warning: " + fail)   # so tests will fail etc
311
 
            return fail
312
 
        try:
313
 
            return template % message_args
314
 
        except ValueError, e:
315
 
            fail = "bzr unprintable warning: %r, %r, %s" % (
316
 
                warning_id, message_args, e)
317
 
            warnings.warn(fail)   # so tests will fail etc
318
 
            return fail
319
 
 
320
173
    def get_boolean(self, prompt):
321
174
        """Get a boolean question answered from the user.
322
175
 
323
176
        :param prompt: a message to prompt the user with. Should be a single
324
 
            line without terminating \\n.
 
177
        line without terminating \n.
325
178
        :return: True or False for y/yes or n/no.
326
179
        """
327
180
        raise NotImplementedError(self.get_boolean)
328
181
 
329
 
    def get_integer(self, prompt):
330
 
        """Get an integer from the user.
331
 
 
332
 
        :param prompt: a message to prompt the user with. Could be a multi-line
333
 
            prompt but without a terminating \\n.
334
 
 
335
 
        :return: A signed integer.
336
 
        """
337
 
        raise NotImplementedError(self.get_integer)
338
 
 
339
182
    def make_progress_view(self):
340
183
        """Construct a new ProgressView object for this UI.
341
184
 
344
187
        """
345
188
        return NullProgressView()
346
189
 
347
 
    def recommend_upgrade(self, current_format_name, basedir):
348
 
        """Recommend the user upgrade a control directory.
349
 
 
350
 
        :param current_format_name: Description of the current format
351
 
        :param basedir: Location of the control dir
352
 
        """
353
 
        self.show_user_warning('recommend_upgrade',
354
 
            current_format_name=current_format_name, basedir=basedir)
 
190
    def recommend_upgrade(self,
 
191
        current_format_name,
 
192
        basedir):
 
193
        # this should perhaps be in the TextUIFactory and the default can do
 
194
        # nothing
 
195
        trace.warning("%s is deprecated "
 
196
            "and a better format is available.\n"
 
197
            "It is recommended that you upgrade by "
 
198
            "running the command\n"
 
199
            "  bzr upgrade %s",
 
200
            current_format_name,
 
201
            basedir)
355
202
 
356
203
    def report_transport_activity(self, transport, byte_count, direction):
357
204
        """Called by transports as they do IO.
361
208
        """
362
209
        pass
363
210
 
364
 
    def log_transport_activity(self, display=False):
365
 
        """Write out whatever transport activity has been measured.
366
 
 
367
 
        Implementations are allowed to do nothing, but it is useful if they can
368
 
        write a line to the log file.
369
 
 
370
 
        :param display: If False, only log to disk, if True also try to display
371
 
            a message to the user.
372
 
        :return: None
373
 
        """
374
 
        # Default implementation just does nothing
375
 
        pass
376
 
 
377
 
    def show_user_warning(self, warning_id, **message_args):
378
 
        """Show a warning to the user.
379
 
 
380
 
        This is specifically for things that are under the user's control (eg
381
 
        outdated formats), not for internal program warnings like deprecated
382
 
        APIs.
383
 
 
384
 
        This can be overridden by UIFactory subclasses to show it in some 
385
 
        appropriate way; the default UIFactory is noninteractive and does
386
 
        nothing.  format_user_warning maps it to a string, though other
387
 
        presentations can be used for particular UIs.
388
 
 
389
 
        :param warning_id: An identifier like 'cross_format_fetch' used to 
390
 
            check if the message is suppressed and to look up the string.
391
 
        :param message_args: Arguments to be interpolated into the message.
392
 
        """
393
 
        pass
394
 
 
395
 
    def show_error(self, msg):
396
 
        """Show an error message (not an exception) to the user.
397
 
        
398
 
        The message should not have an error prefix or trailing newline.  That
399
 
        will be added by the factory if appropriate.
400
 
        """
401
 
        raise NotImplementedError(self.show_error)
402
 
 
403
 
    def show_message(self, msg):
404
 
        """Show a message to the user."""
405
 
        raise NotImplementedError(self.show_message)
406
 
 
407
 
    def show_warning(self, msg):
408
 
        """Show a warning to the user."""
409
 
        raise NotImplementedError(self.show_warning)
410
 
 
411
 
    def warn_cross_format_fetch(self, from_format, to_format):
412
 
        """Warn about a potentially slow cross-format transfer.
413
 
        
414
 
        This is deprecated in favor of show_user_warning, but retained for api
415
 
        compatibility in 2.0 and 2.1.
416
 
        """
417
 
        self.show_user_warning('cross_format_fetch', from_format=from_format,
418
 
            to_format=to_format)
419
 
 
420
 
    def warn_experimental_format_fetch(self, inter):
421
 
        """Warn about fetching into experimental repository formats."""
422
 
        if inter.target._format.experimental:
423
 
            trace.warning("Fetching into experimental format %s.\n"
424
 
                "This format may be unreliable or change in the future "
425
 
                "without an upgrade path.\n" % (inter.target._format,))
426
 
 
427
 
 
428
 
class NoninteractiveUIFactory(UIFactory):
429
 
    """Base class for UIs with no user."""
430
 
 
431
 
    def confirm_action(self, prompt, confirmation_id, prompt_kwargs):
432
 
        return True
433
 
 
434
 
    def __repr__(self):
435
 
        return '%s()' % (self.__class__.__name__, )
436
 
 
437
 
 
438
 
class SilentUIFactory(NoninteractiveUIFactory):
 
211
 
 
212
 
 
213
class CLIUIFactory(UIFactory):
 
214
    """Deprecated in favor of TextUIFactory."""
 
215
 
 
216
    @deprecated_method(deprecated_in((1, 18, 0)))
 
217
    def __init__(self, stdin=None, stdout=None, stderr=None):
 
218
        UIFactory.__init__(self)
 
219
        self.stdin = stdin or sys.stdin
 
220
        self.stdout = stdout or sys.stdout
 
221
        self.stderr = stderr or sys.stderr
 
222
 
 
223
    _accepted_boolean_strings = dict(y=True, n=False, yes=True, no=False)
 
224
 
 
225
    def get_boolean(self, prompt):
 
226
        while True:
 
227
            self.prompt(prompt + "? [y/n]: ")
 
228
            line = self.stdin.readline()
 
229
            line = line.rstrip('\n')
 
230
            val = bool_from_string(line, self._accepted_boolean_strings)
 
231
            if val is not None:
 
232
                return val
 
233
 
 
234
    def get_non_echoed_password(self):
 
235
        isatty = getattr(self.stdin, 'isatty', None)
 
236
        if isatty is not None and isatty():
 
237
            # getpass() ensure the password is not echoed and other
 
238
            # cross-platform niceties
 
239
            password = getpass.getpass('')
 
240
        else:
 
241
            # echo doesn't make sense without a terminal
 
242
            password = self.stdin.readline()
 
243
            if not password:
 
244
                password = None
 
245
            elif password[-1] == '\n':
 
246
                password = password[:-1]
 
247
        return password
 
248
 
 
249
    def get_password(self, prompt='', **kwargs):
 
250
        """Prompt the user for a password.
 
251
 
 
252
        :param prompt: The prompt to present the user
 
253
        :param kwargs: Arguments which will be expanded into the prompt.
 
254
                       This lets front ends display different things if
 
255
                       they so choose.
 
256
        :return: The password string, return None if the user
 
257
                 canceled the request.
 
258
        """
 
259
        prompt += ': '
 
260
        self.prompt(prompt, **kwargs)
 
261
        # There's currently no way to say 'i decline to enter a password'
 
262
        # as opposed to 'my password is empty' -- does it matter?
 
263
        return self.get_non_echoed_password()
 
264
 
 
265
    def get_username(self, prompt, **kwargs):
 
266
        """Prompt the user for a username.
 
267
 
 
268
        :param prompt: The prompt to present the user
 
269
        :param kwargs: Arguments which will be expanded into the prompt.
 
270
                       This lets front ends display different things if
 
271
                       they so choose.
 
272
        :return: The username string, return None if the user
 
273
                 canceled the request.
 
274
        """
 
275
        prompt += ': '
 
276
        self.prompt(prompt, **kwargs)
 
277
        username = self.stdin.readline()
 
278
        if not username:
 
279
            username = None
 
280
        elif username[-1] == '\n':
 
281
            username = username[:-1]
 
282
        return username
 
283
 
 
284
    def prompt(self, prompt, **kwargs):
 
285
        """Emit prompt on the CLI.
 
286
        
 
287
        :param kwargs: Dictionary of arguments to insert into the prompt,
 
288
            to allow UIs to reformat the prompt.
 
289
        """
 
290
        if kwargs:
 
291
            # See <https://launchpad.net/bugs/365891>
 
292
            prompt = prompt % kwargs
 
293
        prompt = prompt.encode(osutils.get_terminal_encoding(), 'replace')
 
294
        self.clear_term()
 
295
        self.stderr.write(prompt)
 
296
 
 
297
    def note(self, msg):
 
298
        """Write an already-formatted message."""
 
299
        self.stdout.write(msg + '\n')
 
300
 
 
301
 
 
302
class SilentUIFactory(UIFactory):
439
303
    """A UI Factory which never prints anything.
440
304
 
441
305
    This is the default UI, if another one is never registered by a program
454
318
    def get_username(self, prompt, **kwargs):
455
319
        return None
456
320
 
457
 
    def _make_output_stream_explicit(self, encoding, encoding_type):
458
 
        return NullOutputStream(encoding)
459
 
 
460
 
    def show_error(self, msg):
461
 
        pass
462
 
 
463
 
    def show_message(self, msg):
464
 
        pass
465
 
 
466
 
    def show_warning(self, msg):
467
 
        pass
468
 
 
469
321
 
470
322
class CannedInputUIFactory(SilentUIFactory):
471
323
    """A silent UI that return canned input."""
476
328
    def __repr__(self):
477
329
        return "%s(%r)" % (self.__class__.__name__, self.responses)
478
330
 
479
 
    def confirm_action(self, prompt, confirmation_id, args):
480
 
        return self.get_boolean(prompt % args)
481
 
 
482
331
    def get_boolean(self, prompt):
483
332
        return self.responses.pop(0)
484
333
 
485
 
    def get_integer(self, prompt):
486
 
        return self.responses.pop(0)
487
 
 
488
 
    def get_password(self, prompt=u'', **kwargs):
 
334
    def get_password(self, prompt='', **kwargs):
489
335
        return self.responses.pop(0)
490
336
 
491
337
    def get_username(self, prompt, **kwargs):
492
338
        return self.responses.pop(0)
493
 
 
 
339
    
494
340
    def assert_all_input_consumed(self):
495
341
        if self.responses:
496
342
            raise AssertionError("expected all input in %r to be consumed"
497
343
                % (self,))
498
344
 
499
345
 
 
346
@deprecated_function(deprecated_in((1, 18, 0)))
 
347
def clear_decorator(func, *args, **kwargs):
 
348
    """Decorator that clears the term"""
 
349
    ui_factory.clear_term()
 
350
    func(*args, **kwargs)
 
351
 
 
352
 
500
353
ui_factory = SilentUIFactory()
501
354
# IMPORTANT: never import this symbol directly. ONLY ever access it as
502
355
# ui.ui_factory, so that you refer to the current value.
522
375
 
523
376
    def show_transport_activity(self, transport, direction, byte_count):
524
377
        pass
525
 
 
526
 
    def log_transport_activity(self, display=False):
527
 
        pass
528
 
 
529
 
 
530
 
class NullOutputStream(object):
531
 
    """Acts like a file, but discard all output."""
532
 
 
533
 
    def __init__(self, encoding):
534
 
        self.encoding = encoding
535
 
 
536
 
    def write(self, data):
537
 
        pass
538
 
 
539
 
    def writelines(self, data):
540
 
        pass
541
 
 
542
 
    def close(self):
543
 
        pass