1
# Copyright (C) 2005-2010 Canonical Ltd
1
# Copyright (C) 2005-2011 Canonical Ltd
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
49
47
from bzrlib.lazy_import import lazy_import
50
48
lazy_import(globals(), """
53
49
from bzrlib import (
60
from bzrlib.symbol_versioning import (
67
58
_valid_boolean_strings = dict(yes=True, no=False,
94
class ConfirmationUserInterfacePolicy(object):
95
"""Wrapper for a UIFactory that allows or denies all confirmed actions."""
97
def __init__(self, wrapped_ui, default_answer, specific_answers):
98
"""Generate a proxy UI that does no confirmations.
100
:param wrapped_ui: Underlying UIFactory.
101
:param default_answer: Bool for whether requests for
102
confirmation from the user should be noninteractively accepted or
104
:param specific_answers: Map from confirmation_id to bool answer.
106
self.wrapped_ui = wrapped_ui
107
self.default_answer = default_answer
108
self.specific_answers = specific_answers
110
def __getattr__(self, name):
111
return getattr(self.wrapped_ui, name)
114
return '%s(%r, %r, %r)' % (
115
self.__class__.__name__,
118
self.specific_answers)
120
def confirm_action(self, prompt, confirmation_id, prompt_kwargs):
121
if confirmation_id in self.specific_answers:
122
return self.specific_answers[confirmation_id]
123
elif self.default_answer is not None:
124
return self.default_answer
126
return self.wrapped_ui.confirm_action(
127
prompt, confirmation_id, prompt_kwargs)
103
130
class UIFactory(object):
104
131
"""UI abstraction.
106
133
This tells the library how to display things to the user. Through this
107
134
layer different applications can choose the style of UI.
136
UI Factories are also context managers, for some syntactic sugar some users
109
139
:ivar suppressed_warnings: Identifiers for user warnings that should
115
145
"%(from_format)s to %(to_format)s.\n"
116
146
"This may take some time. Upgrade the repositories to the "
117
147
"same format for better performance."
149
experimental_format_fetch=("Fetching into experimental format "
151
"This format may be unreliable or change in the future "
152
"without an upgrade path.\n"),
154
"The command 'bzr %(deprecated_name)s' "
155
"has been deprecated in bzr %(deprecated_in_version)s. "
156
"Please use 'bzr %(recommended_name)s' instead."),
157
recommend_upgrade=("%(current_format_name)s is deprecated "
158
"and a better format is available.\n"
159
"It is recommended that you upgrade by "
160
"running the command\n"
161
" bzr upgrade %(basedir)s"),
163
u"Stole dead lock %(lock_url)s %(other_holder_info)s."),
121
166
def __init__(self):
123
168
self.suppressed_warnings = set()
124
169
self._quiet = False
172
"""Context manager entry support.
174
Override in a concrete factory class if initialisation before use is
177
return self # This is bound to the 'as' clause in a with statement.
179
def __exit__(self, exc_type, exc_val, exc_tb):
180
"""Context manager exit support.
182
Override in a concrete factory class if more cleanup than a simple
183
self.clear_term() is needed when the UIFactory is finished with.
186
return False # propogate exceptions.
126
188
def be_quiet(self, state):
127
189
"""Tell the UI to be more quiet, or not.
132
194
self._quiet = state
134
def get_password(self, prompt='', **kwargs):
196
def confirm_action(self, prompt, confirmation_id, prompt_kwargs):
197
"""Seek user confirmation for an action.
199
If the UI is noninteractive, or the user does not want to be asked
200
about this action, True is returned, indicating bzr should just
203
The confirmation id allows the user to configure certain actions to
204
always be confirmed or always denied, and for UIs to specialize the
205
display of particular confirmations.
207
:param prompt: Suggested text to display to the user.
208
:param prompt_kwargs: A dictionary of arguments that can be
209
string-interpolated into the prompt.
210
:param confirmation_id: Unique string identifier for the confirmation.
212
return self.get_boolean(prompt % prompt_kwargs)
214
def get_password(self, prompt=u'', **kwargs):
135
215
"""Prompt the user for a password.
137
:param prompt: The prompt to present the user
217
:param prompt: The prompt to present the user (must be unicode)
138
218
:param kwargs: Arguments which will be expanded into the prompt.
139
219
This lets front ends display different things if
158
238
version of stdout, but in a GUI it might be appropriate to send it to a
159
239
window displaying the text.
161
:param encoding: Unicode encoding for output; default is the
162
terminal encoding, which may be different from the user encoding.
241
:param encoding: Unicode encoding for output; if not specified
242
uses the configured 'output_encoding' if any; otherwise the
163
244
(See get_terminal_encoding.)
165
246
:param encoding_type: How to handle encoding errors:
168
249
# XXX: is the caller supposed to close the resulting object?
169
250
if encoding is None:
170
encoding = osutils.get_terminal_encoding()
251
encoding = config.GlobalStack().get('output_encoding')
253
encoding = osutils.get_terminal_encoding(trace=True)
171
254
if encoding_type is None:
172
255
encoding_type = 'replace'
173
256
out_stream = self._make_output_stream_explicit(encoding, encoding_type)
227
310
template = self._user_warning_templates[warning_id]
229
fail = "failed to format warning %r, %r" % (warning_id, message_args)
230
warnings.warn(fail) # so tests will fail etc
312
fail = "bzr warning: %r, %r" % (warning_id, message_args)
313
warnings.warn("no template for warning: " + fail) # so tests will fail etc
233
316
return template % message_args
234
317
except ValueError, e:
235
fail = "failed to format warning %r, %r: %s" % (
318
fail = "bzr unprintable warning: %r, %r, %s" % (
236
319
warning_id, message_args, e)
237
320
warnings.warn(fail) # so tests will fail etc
241
324
"""Get a boolean question answered from the user.
243
326
:param prompt: a message to prompt the user with. Should be a single
244
line without terminating \n.
327
line without terminating \\n.
245
328
:return: True or False for y/yes or n/no.
247
330
raise NotImplementedError(self.get_boolean)
250
333
"""Get an integer from the user.
252
335
:param prompt: a message to prompt the user with. Could be a multi-line
253
prompt but without a terminating \n.
336
prompt but without a terminating \\n.
255
338
:return: A signed integer.
265
348
return NullProgressView()
267
def recommend_upgrade(self,
270
# XXX: this should perhaps be in the TextUIFactory and the default can do
273
# XXX: Change to show_user_warning - that will accomplish the previous
274
# xxx. -- mbp 2010-02-25
275
trace.warning("%s is deprecated "
276
"and a better format is available.\n"
277
"It is recommended that you upgrade by "
278
"running the command\n"
350
def recommend_upgrade(self, current_format_name, basedir):
351
"""Recommend the user upgrade a control directory.
353
:param current_format_name: Description of the current format
354
:param basedir: Location of the control dir
356
self.show_user_warning('recommend_upgrade',
357
current_format_name=current_format_name, basedir=basedir)
283
359
def report_transport_activity(self, transport, byte_count, direction):
284
360
"""Called by transports as they do IO.
335
411
"""Show a warning to the user."""
336
412
raise NotImplementedError(self.show_warning)
338
def warn_cross_format_fetch(self, from_format, to_format):
339
"""Warn about a potentially slow cross-format transfer.
341
This is deprecated in favor of show_user_warning, but retained for api
342
compatibility in 2.0 and 2.1.
344
self.show_user_warning('cross_format_fetch', from_format=from_format,
347
def warn_experimental_format_fetch(self, inter):
348
"""Warn about fetching into experimental repository formats."""
349
if inter.target._format.experimental:
350
trace.warning("Fetching into experimental format %s.\n"
351
"This format may be unreliable or change in the future "
352
"without an upgrade path.\n" % (inter.target._format,))
356
class SilentUIFactory(UIFactory):
415
class NoninteractiveUIFactory(UIFactory):
416
"""Base class for UIs with no user."""
418
def confirm_action(self, prompt, confirmation_id, prompt_kwargs):
422
return '%s()' % (self.__class__.__name__, )
425
class SilentUIFactory(NoninteractiveUIFactory):
357
426
"""A UI Factory which never prints anything.
359
428
This is the default UI, if another one is never registered by a program
394
463
def __repr__(self):
395
464
return "%s(%r)" % (self.__class__.__name__, self.responses)
466
def confirm_action(self, prompt, confirmation_id, args):
467
return self.get_boolean(prompt % args)
397
469
def get_boolean(self, prompt):
398
470
return self.responses.pop(0)
400
472
def get_integer(self, prompt):
401
473
return self.responses.pop(0)
403
def get_password(self, prompt='', **kwargs):
475
def get_password(self, prompt=u'', **kwargs):
404
476
return self.responses.pop(0)
406
478
def get_username(self, prompt, **kwargs):