~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/ui/__init__.py

  • Committer: Patch Queue Manager
  • Date: 2016-04-21 05:06:57 UTC
  • mfrom: (6603.4.1 bzr)
  • Revision ID: pqm@pqm.ubuntu.com-20160421050657-ygnzfybewvudf1j9
(richard-wilbur) Use initial_comment as commit_message for lp_propose.(Shawn
 Wang) (Shawn Wang)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005-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
41
41
    back to working through the terminal.
42
42
"""
43
43
 
 
44
from __future__ import absolute_import
44
45
 
45
 
import os
46
 
import sys
47
46
import warnings
48
47
 
49
48
from bzrlib.lazy_import import lazy_import
50
49
lazy_import(globals(), """
51
 
import getpass
52
 
 
53
50
from bzrlib import (
54
 
    errors,
 
51
    config,
55
52
    osutils,
56
53
    progress,
57
54
    trace,
58
55
    )
59
56
""")
60
 
from bzrlib.symbol_versioning import (
61
 
    deprecated_function,
62
 
    deprecated_in,
63
 
    deprecated_method,
64
 
    )
65
57
 
66
58
 
67
59
_valid_boolean_strings = dict(yes=True, no=False,
100
92
    return val
101
93
 
102
94
 
 
95
class ConfirmationUserInterfacePolicy(object):
 
96
    """Wrapper for a UIFactory that allows or denies all confirmed actions."""
 
97
 
 
98
    def __init__(self, wrapped_ui, default_answer, specific_answers):
 
99
        """Generate a proxy UI that does no confirmations.
 
100
 
 
101
        :param wrapped_ui: Underlying UIFactory.
 
102
        :param default_answer: Bool for whether requests for
 
103
            confirmation from the user should be noninteractively accepted or
 
104
            denied.
 
105
        :param specific_answers: Map from confirmation_id to bool answer.
 
106
        """
 
107
        self.wrapped_ui = wrapped_ui
 
108
        self.default_answer = default_answer
 
109
        self.specific_answers = specific_answers
 
110
 
 
111
    def __getattr__(self, name):
 
112
        return getattr(self.wrapped_ui, name)
 
113
 
 
114
    def __repr__(self):
 
115
        return '%s(%r, %r, %r)' % (
 
116
            self.__class__.__name__,
 
117
            self.wrapped_ui,
 
118
            self.default_answer, 
 
119
            self.specific_answers)
 
120
 
 
121
    def confirm_action(self, prompt, confirmation_id, prompt_kwargs):
 
122
        if confirmation_id in self.specific_answers:
 
123
            return self.specific_answers[confirmation_id]
 
124
        elif self.default_answer is not None:
 
125
            return self.default_answer
 
126
        else:
 
127
            return self.wrapped_ui.confirm_action(
 
128
                prompt, confirmation_id, prompt_kwargs)
 
129
 
 
130
 
103
131
class UIFactory(object):
104
132
    """UI abstraction.
105
133
 
106
134
    This tells the library how to display things to the user.  Through this
107
135
    layer different applications can choose the style of UI.
 
136
 
 
137
    UI Factories are also context managers, for some syntactic sugar some users
 
138
    need.
 
139
 
 
140
    :ivar suppressed_warnings: Identifiers for user warnings that should 
 
141
        no be emitted.
108
142
    """
109
143
 
 
144
    _user_warning_templates = dict(
 
145
        cross_format_fetch=("Doing on-the-fly conversion from "
 
146
            "%(from_format)s to %(to_format)s.\n"
 
147
            "This may take some time. Upgrade the repositories to the "
 
148
            "same format for better performance."
 
149
            ),
 
150
        experimental_format_fetch=("Fetching into experimental format "
 
151
            "%(to_format)s.\n"
 
152
            "This format may be unreliable or change in the future "
 
153
            "without an upgrade path.\n"),
 
154
        deprecated_command=(
 
155
            "The command 'bzr %(deprecated_name)s' "
 
156
            "has been deprecated in bzr %(deprecated_in_version)s. "
 
157
            "Please use 'bzr %(recommended_name)s' instead."),
 
158
        deprecated_command_option=(
 
159
            "The option '%(deprecated_name)s' to 'bzr %(command)s' "
 
160
            "has been deprecated in bzr %(deprecated_in_version)s. "
 
161
            "Please use '%(recommended_name)s' instead."),
 
162
        recommend_upgrade=("%(current_format_name)s is deprecated "
 
163
            "and a better format is available.\n"
 
164
            "It is recommended that you upgrade by "
 
165
            "running the command\n"
 
166
            "  bzr upgrade %(basedir)s"),
 
167
        locks_steal_dead=(
 
168
            u"Stole dead lock %(lock_url)s %(other_holder_info)s."),
 
169
        not_checking_ssl_cert=(
 
170
            u"Not checking SSL certificate for %(host)s."),
 
171
        )
 
172
 
110
173
    def __init__(self):
111
174
        self._task_stack = []
 
175
        self.suppressed_warnings = set()
112
176
        self._quiet = False
113
177
 
 
178
    def __enter__(self):
 
179
        """Context manager entry support.
 
180
 
 
181
        Override in a concrete factory class if initialisation before use is
 
182
        needed.
 
183
        """
 
184
        return self # This is bound to the 'as' clause in a with statement.
 
185
 
 
186
    def __exit__(self, exc_type, exc_val, exc_tb):
 
187
        """Context manager exit support.
 
188
 
 
189
        Override in a concrete factory class if more cleanup than a simple
 
190
        self.clear_term() is needed when the UIFactory is finished with.
 
191
        """
 
192
        self.clear_term()
 
193
        return False # propogate exceptions.
 
194
 
114
195
    def be_quiet(self, state):
115
196
        """Tell the UI to be more quiet, or not.
116
197
 
119
200
        """
120
201
        self._quiet = state
121
202
 
122
 
    def get_password(self, prompt='', **kwargs):
 
203
    def confirm_action(self, prompt, confirmation_id, prompt_kwargs):
 
204
        """Seek user confirmation for an action.
 
205
 
 
206
        If the UI is noninteractive, or the user does not want to be asked
 
207
        about this action, True is returned, indicating bzr should just
 
208
        proceed.
 
209
 
 
210
        The confirmation id allows the user to configure certain actions to
 
211
        always be confirmed or always denied, and for UIs to specialize the
 
212
        display of particular confirmations.
 
213
 
 
214
        :param prompt: Suggested text to display to the user.
 
215
        :param prompt_kwargs: A dictionary of arguments that can be
 
216
            string-interpolated into the prompt.
 
217
        :param confirmation_id: Unique string identifier for the confirmation.
 
218
        """
 
219
        return self.get_boolean(prompt % prompt_kwargs)
 
220
 
 
221
    def get_password(self, prompt=u'', **kwargs):
123
222
        """Prompt the user for a password.
124
223
 
125
 
        :param prompt: The prompt to present the user
 
224
        :param prompt: The prompt to present the user (must be unicode)
126
225
        :param kwargs: Arguments which will be expanded into the prompt.
127
226
                       This lets front ends display different things if
128
227
                       they so choose.
146
245
        version of stdout, but in a GUI it might be appropriate to send it to a 
147
246
        window displaying the text.
148
247
     
149
 
        :param encoding: Unicode encoding for output; default is the 
150
 
            terminal encoding, which may be different from the user encoding.
 
248
        :param encoding: Unicode encoding for output; if not specified 
 
249
            uses the configured 'output_encoding' if any; otherwise the 
 
250
            terminal encoding. 
151
251
            (See get_terminal_encoding.)
152
252
 
153
253
        :param encoding_type: How to handle encoding errors:
155
255
        """
156
256
        # XXX: is the caller supposed to close the resulting object?
157
257
        if encoding is None:
158
 
            encoding = osutils.get_terminal_encoding()
 
258
            encoding = config.GlobalStack().get('output_encoding')
 
259
        if encoding is None:
 
260
            encoding = osutils.get_terminal_encoding(trace=True)
159
261
        if encoding_type is None:
160
262
            encoding_type = 'replace'
161
263
        out_stream = self._make_output_stream_explicit(encoding, encoding_type)
210
312
        """
211
313
        pass
212
314
 
 
315
    def format_user_warning(self, warning_id, message_args):
 
316
        try:
 
317
            template = self._user_warning_templates[warning_id]
 
318
        except KeyError:
 
319
            fail = "bzr warning: %r, %r" % (warning_id, message_args)
 
320
            warnings.warn("no template for warning: " + fail)   # so tests will fail etc
 
321
            return fail
 
322
        try:
 
323
            return template % message_args
 
324
        except ValueError, e:
 
325
            fail = "bzr unprintable warning: %r, %r, %s" % (
 
326
                warning_id, message_args, e)
 
327
            warnings.warn(fail)   # so tests will fail etc
 
328
            return fail
 
329
 
 
330
    def choose(self, msg, choices, default=None):
 
331
        """Prompt the user for a list of alternatives.
 
332
 
 
333
        :param msg: message to be shown as part of the prompt.
 
334
 
 
335
        :param choices: list of choices, with the individual choices separated
 
336
            by '\n', e.g.: choose("Save changes?", "&Yes\n&No\n&Cancel"). The
 
337
            letter after the '&' is the shortcut key for that choice. Thus you
 
338
            can type 'c' to select "Cancel".  Shorcuts are case insensitive.
 
339
            The shortcut does not need to be the first letter. If a shorcut key
 
340
            is not provided, the first letter for the choice will be used.
 
341
 
 
342
        :param default: default choice (index), returned for example when enter
 
343
            is pressed for the console version.
 
344
 
 
345
        :return: the index fo the user choice (so '0', '1' or '2' for
 
346
            respectively yes/no/cancel in the previous example).
 
347
        """
 
348
        raise NotImplementedError(self.choose)
 
349
 
213
350
    def get_boolean(self, prompt):
214
351
        """Get a boolean question answered from the user.
215
352
 
216
353
        :param prompt: a message to prompt the user with. Should be a single
217
 
        line without terminating \n.
 
354
            line without terminating \\n.
218
355
        :return: True or False for y/yes or n/no.
219
356
        """
220
 
        raise NotImplementedError(self.get_boolean)
 
357
        choice = self.choose(prompt + '?', '&yes\n&no', default=None)
 
358
        return 0 == choice
221
359
 
222
360
    def get_integer(self, prompt):
223
361
        """Get an integer from the user.
224
362
 
225
363
        :param prompt: a message to prompt the user with. Could be a multi-line
226
 
            prompt but without a terminating \n.
 
364
            prompt but without a terminating \\n.
227
365
 
228
366
        :return: A signed integer.
229
367
        """
237
375
        """
238
376
        return NullProgressView()
239
377
 
240
 
    def recommend_upgrade(self,
241
 
        current_format_name,
242
 
        basedir):
243
 
        # this should perhaps be in the TextUIFactory and the default can do
244
 
        # nothing
245
 
        trace.warning("%s is deprecated "
246
 
            "and a better format is available.\n"
247
 
            "It is recommended that you upgrade by "
248
 
            "running the command\n"
249
 
            "  bzr upgrade %s",
250
 
            current_format_name,
251
 
            basedir)
 
378
    def recommend_upgrade(self, current_format_name, basedir):
 
379
        """Recommend the user upgrade a control directory.
 
380
 
 
381
        :param current_format_name: Description of the current format
 
382
        :param basedir: Location of the control dir
 
383
        """
 
384
        self.show_user_warning('recommend_upgrade',
 
385
            current_format_name=current_format_name, basedir=basedir)
252
386
 
253
387
    def report_transport_activity(self, transport, byte_count, direction):
254
388
        """Called by transports as they do IO.
271
405
        # Default implementation just does nothing
272
406
        pass
273
407
 
 
408
    def show_user_warning(self, warning_id, **message_args):
 
409
        """Show a warning to the user.
 
410
 
 
411
        This is specifically for things that are under the user's control (eg
 
412
        outdated formats), not for internal program warnings like deprecated
 
413
        APIs.
 
414
 
 
415
        This can be overridden by UIFactory subclasses to show it in some 
 
416
        appropriate way; the default UIFactory is noninteractive and does
 
417
        nothing.  format_user_warning maps it to a string, though other
 
418
        presentations can be used for particular UIs.
 
419
 
 
420
        :param warning_id: An identifier like 'cross_format_fetch' used to 
 
421
            check if the message is suppressed and to look up the string.
 
422
        :param message_args: Arguments to be interpolated into the message.
 
423
        """
 
424
        pass
 
425
 
274
426
    def show_error(self, msg):
275
427
        """Show an error message (not an exception) to the user.
276
428
        
287
439
        """Show a warning to the user."""
288
440
        raise NotImplementedError(self.show_warning)
289
441
 
290
 
    def warn_cross_format_fetch(self, from_format, to_format):
291
 
        """Warn about a potentially slow cross-format transfer"""
292
 
        # See <https://launchpad.net/bugs/456077> asking for a warning here
293
 
        trace.warning("Doing on-the-fly conversion from %s to %s.\n"
294
 
            "This may take some time. Upgrade the repositories to the "
295
 
            "same format for better performance.\n" %
296
 
            (from_format, to_format))
297
 
 
298
 
    def warn_experimental_format_fetch(self, inter):
299
 
        """Warn about fetching into experimental repository formats."""
300
 
        if inter.target._format.experimental:
301
 
            trace.warning("Fetching into experimental format %s.\n"
302
 
                "This format may be unreliable or change in the future "
303
 
                "without an upgrade path.\n" % (inter.target._format,))
304
 
 
305
 
 
306
 
 
307
 
class SilentUIFactory(UIFactory):
 
442
 
 
443
class NoninteractiveUIFactory(UIFactory):
 
444
    """Base class for UIs with no user."""
 
445
 
 
446
    def confirm_action(self, prompt, confirmation_id, prompt_kwargs):
 
447
        return True
 
448
 
 
449
    def __repr__(self):
 
450
        return '%s()' % (self.__class__.__name__, )
 
451
 
 
452
 
 
453
class SilentUIFactory(NoninteractiveUIFactory):
308
454
    """A UI Factory which never prints anything.
309
455
 
310
456
    This is the default UI, if another one is never registered by a program
345
491
    def __repr__(self):
346
492
        return "%s(%r)" % (self.__class__.__name__, self.responses)
347
493
 
 
494
    def confirm_action(self, prompt, confirmation_id, args):
 
495
        return self.get_boolean(prompt % args)
 
496
 
348
497
    def get_boolean(self, prompt):
349
498
        return self.responses.pop(0)
350
499
 
351
500
    def get_integer(self, prompt):
352
501
        return self.responses.pop(0)
353
502
 
354
 
    def get_password(self, prompt='', **kwargs):
 
503
    def get_password(self, prompt=u'', **kwargs):
355
504
        return self.responses.pop(0)
356
505
 
357
506
    def get_username(self, prompt, **kwargs):