/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar
3948.2.2 by Martin Pool
Corrections to finishing progress bars
1
# Copyright (C) 2005, 2006, 2007, 2008, 2009 Canonical Ltd
1887.1.1 by Adeodato Simó
Do not separate paragraphs in the copyright statement with blank lines,
2
#
1185.1.29 by Robert Collins
merge merge tweaks from aaron, which includes latest .dev
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
1887.1.1 by Adeodato Simó
Do not separate paragraphs in the copyright statement with blank lines,
7
#
1185.1.29 by Robert Collins
merge merge tweaks from aaron, which includes latest .dev
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
# GNU General Public License for more details.
1887.1.1 by Adeodato Simó
Do not separate paragraphs in the copyright statement with blank lines,
12
#
1185.1.29 by Robert Collins
merge merge tweaks from aaron, which includes latest .dev
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
4183.7.1 by Sabin Iacob
update FSF mailing address
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
1185.1.29 by Robert Collins
merge merge tweaks from aaron, which includes latest .dev
16
4449.3.41 by Martin Pool
UI doc updates suggested by Robert
17
"""Abstraction for interacting with the user.
1185.1.29 by Robert Collins
merge merge tweaks from aaron, which includes latest .dev
18
4449.3.41 by Martin Pool
UI doc updates suggested by Robert
19
Applications can choose different types of UI, and they deal with displaying
20
messages or progress to the user, and with gathering different types of input.
1185.1.29 by Robert Collins
merge merge tweaks from aaron, which includes latest .dev
21
4449.3.6 by Martin Pool
Updated bzrlib.ui module docs
22
Several levels are supported, and you can also register new factories such as
23
for a GUI.
24
4711.1.3 by Martin Pool
Clearer class names in bzrlib.ui docstring
25
bzrlib.ui.UIFactory
4449.3.15 by Martin Pool
Move NullProgressView and make_progress_view up to UIFactory base class
26
    Semi-abstract base class
4449.3.6 by Martin Pool
Updated bzrlib.ui module docs
27
4711.1.3 by Martin Pool
Clearer class names in bzrlib.ui docstring
28
bzrlib.ui.SilentUIFactory
4449.3.6 by Martin Pool
Updated bzrlib.ui module docs
29
    Produces no output and cannot take any input; useful for programs using
30
    bzrlib in batch mode or for programs such as loggerhead.
4449.3.15 by Martin Pool
Move NullProgressView and make_progress_view up to UIFactory base class
31
4711.1.3 by Martin Pool
Clearer class names in bzrlib.ui docstring
32
bzrlib.ui.CannedInputUIFactory
4449.3.41 by Martin Pool
UI doc updates suggested by Robert
33
    For use in testing; the input values to be returned are provided 
34
    at construction.
35
4711.1.3 by Martin Pool
Clearer class names in bzrlib.ui docstring
36
bzrlib.ui.text.TextUIFactory
4449.3.15 by Martin Pool
Move NullProgressView and make_progress_view up to UIFactory base class
37
    Standard text command-line interface, with stdin, stdout, stderr.
38
    May make more or less advanced use of them, eg in drawing progress bars,
39
    depending on the detected capabilities of the terminal.
4449.3.41 by Martin Pool
UI doc updates suggested by Robert
40
    GUIs may choose to subclass this so that unimplemented methods fall
41
    back to working through the terminal.
1185.1.29 by Robert Collins
merge merge tweaks from aaron, which includes latest .dev
42
"""
43
4449.3.16 by Martin Pool
Commentary on UIs
44
3882.8.11 by Martin Pool
Choose the UIFactory class depending on the terminal capabilities
45
import os
1687.1.4 by Robert Collins
Add bzrlib.ui.ui_factory.get_boolean().
46
import sys
3882.8.12 by Martin Pool
Give a warning, not an error, if a progress bar is not finished in order
47
import warnings
1594.1.1 by Robert Collins
Introduce new bzr progress bar api. ui_factory.nested_progress_bar.
48
1996.3.32 by John Arbash Meinel
from bzrlib.ui lazy import progress, and make progress import lazily
49
from bzrlib.lazy_import import lazy_import
50
lazy_import(globals(), """
2294.4.4 by Vincent Ladeuil
Provide a better implementation for testing passwords.
51
import getpass
52
1996.3.32 by John Arbash Meinel
from bzrlib.ui lazy import progress, and make progress import lazily
53
from bzrlib import (
3260.2.1 by Alexander Belchenko
Don't ask a password if there is no real terminal. (#69851)
54
    errors,
2461.1.2 by Vincent Ladeuil
Take jam's remark into account.
55
    osutils,
1996.3.32 by John Arbash Meinel
from bzrlib.ui lazy import progress, and make progress import lazily
56
    progress,
2323.6.2 by Martin Pool
Move responsibility for suggesting upgrades to ui object
57
    trace,
1996.3.32 by John Arbash Meinel
from bzrlib.ui lazy import progress, and make progress import lazily
58
    )
59
""")
4449.3.18 by Martin Pool
Fuse CLIUIFactory and TextUIFactory and deprecate the old name
60
from bzrlib.symbol_versioning import (
61
    deprecated_function,
62
    deprecated_in,
63
    deprecated_method,
64
    )
1996.3.32 by John Arbash Meinel
from bzrlib.ui lazy import progress, and make progress import lazily
65
1185.1.29 by Robert Collins
merge merge tweaks from aaron, which includes latest .dev
66
4503.2.1 by Vincent Ladeuil
Get a bool from a string.
67
_valid_boolean_strings = dict(yes=True, no=False,
68
                              y=True, n=False,
69
                              on=True, off=False,
70
                              true=True, false=False)
71
_valid_boolean_strings['1'] = True
72
_valid_boolean_strings['0'] = False
73
74
75
def bool_from_string(s, accepted_values=None):
76
    """Returns a boolean if the string can be interpreted as such.
77
78
    Interpret case insensitive strings as booleans. The default values
79
    includes: 'yes', 'no, 'y', 'n', 'true', 'false', '0', '1', 'on',
80
    'off'. Alternative values can be provided with the 'accepted_values'
81
    parameter.
82
83
    :param s: A string that should be interpreted as a boolean. It should be of
84
        type string or unicode.
85
86
    :param accepted_values: An optional dict with accepted strings as keys and
87
        True/False as values. The strings will be tested against a lowered
88
        version of 's'.
89
90
    :return: True or False for accepted strings, None otherwise.
91
    """
92
    if accepted_values is None:
93
        accepted_values = _valid_boolean_strings
94
    val = None
95
    if type(s) in (str, unicode):
96
        try:
97
            val = accepted_values[s.lower()]
98
        except KeyError:
99
            pass
100
    return val
101
102
1185.49.21 by John Arbash Meinel
Refactored bzrlib/ui.py into a module with the possibility for multiple ui forms.
103
class UIFactory(object):
104
    """UI abstraction.
105
106
    This tells the library how to display things to the user.  Through this
107
    layer different applications can choose the style of UI.
108
    """
1594.1.1 by Robert Collins
Introduce new bzr progress bar api. ui_factory.nested_progress_bar.
109
1594.1.3 by Robert Collins
Fixup pb usage to use nested_progress_bar.
110
    def __init__(self):
3882.7.7 by Martin Pool
Change progress bars to a more MVC style
111
        self._task_stack = []
1594.1.3 by Robert Collins
Fixup pb usage to use nested_progress_bar.
112
1185.49.22 by John Arbash Meinel
Added get_password to the UIFactory, using it inside of sftp.py
113
    def get_password(self, prompt='', **kwargs):
114
        """Prompt the user for a password.
115
116
        :param prompt: The prompt to present the user
117
        :param kwargs: Arguments which will be expanded into the prompt.
118
                       This lets front ends display different things if
119
                       they so choose.
2294.4.1 by Vincent Ladeuil
Add a UIFactory.get_login method, fix tests.
120
121
        :return: The password string, return None if the user canceled the
122
                 request. Note that we do not touch the encoding, users may
123
                 have whatever they see fit and the password should be
124
                 transported as is.
1185.49.22 by John Arbash Meinel
Added get_password to the UIFactory, using it inside of sftp.py
125
        """
1594.1.1 by Robert Collins
Introduce new bzr progress bar api. ui_factory.nested_progress_bar.
126
        raise NotImplementedError(self.get_password)
2294.4.1 by Vincent Ladeuil
Add a UIFactory.get_login method, fix tests.
127
1594.1.1 by Robert Collins
Introduce new bzr progress bar api. ui_factory.nested_progress_bar.
128
    def nested_progress_bar(self):
129
        """Return a nested progress bar.
130
2095.4.5 by mbp at sourcefrog
Use regular progress-bar classes, not a special mechanism
131
        When the bar has been finished with, it should be released by calling
1594.1.1 by Robert Collins
Introduce new bzr progress bar api. ui_factory.nested_progress_bar.
132
        bar.finished().
133
        """
3882.7.7 by Martin Pool
Change progress bars to a more MVC style
134
        if self._task_stack:
3882.8.3 by Martin Pool
Move display of transport throughput into TextProgressView
135
            t = progress.ProgressTask(self._task_stack[-1], self)
3882.7.7 by Martin Pool
Change progress bars to a more MVC style
136
        else:
3882.8.3 by Martin Pool
Move display of transport throughput into TextProgressView
137
            t = progress.ProgressTask(None, self)
3882.7.7 by Martin Pool
Change progress bars to a more MVC style
138
        self._task_stack.append(t)
139
        return t
140
3948.2.3 by Martin Pool
Make the interface from ProgressTask to ui more private
141
    def _progress_finished(self, task):
142
        """Called by the ProgressTask when it finishes"""
143
        if not self._task_stack:
144
            warnings.warn("%r finished but nothing is active"
145
                % (task,))
146
        elif task != self._task_stack[-1]:
4032.1.2 by John Arbash Meinel
Track down a few more files that have trailing whitespace.
147
            warnings.warn("%r is not the active task %r"
3948.2.2 by Martin Pool
Corrections to finishing progress bars
148
                % (task, self._task_stack[-1]))
3882.8.12 by Martin Pool
Give a warning, not an error, if a progress bar is not finished in order
149
        else:
150
            del self._task_stack[-1]
3948.2.3 by Martin Pool
Make the interface from ProgressTask to ui more private
151
        if not self._task_stack:
3948.2.5 by Martin Pool
rename to _progress_all_finished
152
            self._progress_all_finished()
3948.2.3 by Martin Pool
Make the interface from ProgressTask to ui more private
153
3948.2.5 by Martin Pool
rename to _progress_all_finished
154
    def _progress_all_finished(self):
3948.2.3 by Martin Pool
Make the interface from ProgressTask to ui more private
155
        """Called when the top-level progress task finished"""
156
        pass
157
158
    def _progress_updated(self, task):
159
        """Called by the ProgressTask when it changes.
4032.1.2 by John Arbash Meinel
Track down a few more files that have trailing whitespace.
160
3948.2.7 by Martin Pool
pep8
161
        Should be specialized to draw the progress.
162
        """
3948.2.3 by Martin Pool
Make the interface from ProgressTask to ui more private
163
        pass
1594.1.1 by Robert Collins
Introduce new bzr progress bar api. ui_factory.nested_progress_bar.
164
1558.8.1 by Aaron Bentley
Fix overall progress bar's interaction with 'note' and 'warning'
165
    def clear_term(self):
166
        """Prepare the terminal for output.
167
168
        This will, for example, clear text progress bars, and leave the
3948.2.7 by Martin Pool
pep8
169
        cursor at the leftmost position.
170
        """
3948.2.2 by Martin Pool
Corrections to finishing progress bars
171
        pass
1558.8.1 by Aaron Bentley
Fix overall progress bar's interaction with 'note' and 'warning'
172
1687.1.4 by Robert Collins
Add bzrlib.ui.ui_factory.get_boolean().
173
    def get_boolean(self, prompt):
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
174
        """Get a boolean question answered from the user.
1687.1.4 by Robert Collins
Add bzrlib.ui.ui_factory.get_boolean().
175
176
        :param prompt: a message to prompt the user with. Should be a single
177
        line without terminating \n.
178
        :return: True or False for y/yes or n/no.
179
        """
180
        raise NotImplementedError(self.get_boolean)
181
4597.3.37 by Vincent Ladeuil
Allows ui factories to query users for an integer.
182
    def get_integer(self, prompt):
183
        """Get an integer from the user.
184
185
        :param prompt: a message to prompt the user with. Could be a multi-line
186
            prompt but without a terminating \n.
187
188
        :return: A signed integer.
189
        """
190
        raise NotImplementedError(self.get_integer)
191
4449.3.15 by Martin Pool
Move NullProgressView and make_progress_view up to UIFactory base class
192
    def make_progress_view(self):
193
        """Construct a new ProgressView object for this UI.
194
195
        Application code should normally not call this but instead
196
        nested_progress_bar().
197
        """
198
        return NullProgressView()
199
2323.6.2 by Martin Pool
Move responsibility for suggesting upgrades to ui object
200
    def recommend_upgrade(self,
201
        current_format_name,
202
        basedir):
2323.6.4 by Martin Pool
BzrDir._check_supported now also takes care of recommending upgrades, which
203
        # this should perhaps be in the TextUIFactory and the default can do
204
        # nothing
2323.6.2 by Martin Pool
Move responsibility for suggesting upgrades to ui object
205
        trace.warning("%s is deprecated "
206
            "and a better format is available.\n"
207
            "It is recommended that you upgrade by "
208
            "running the command\n"
209
            "  bzr upgrade %s",
210
            current_format_name,
211
            basedir)
2323.6.4 by Martin Pool
BzrDir._check_supported now also takes care of recommending upgrades, which
212
3882.7.5 by Martin Pool
Further mockup of transport-based activity indicator.
213
    def report_transport_activity(self, transport, byte_count, direction):
214
        """Called by transports as they do IO.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
215
3882.7.5 by Martin Pool
Further mockup of transport-based activity indicator.
216
        This may update a progress bar, spinner, or similar display.
217
        By default it does nothing.
218
        """
219
        pass
220
4711.1.7 by Martin Pool
Add UIFactory.show_error, show_warning, show_message
221
    def show_error(self, msg):
222
        """Show an error message (not an exception) to the user.
223
        
224
        The message should not have an error prefix or trailing newline.  That
225
        will be added by the factory if appropriate. 
226
        """
227
        raise NotImplementedError(self.show_error)
228
229
    def show_message(self, msg):
230
        """Show a message to the user."""
231
        raise NotImplementedError(self.show_message)
232
233
    def show_warning(self, msg):
234
        """Show a warning to the user."""
235
        raise NotImplementedError(self.show_warning)
236
3882.7.5 by Martin Pool
Further mockup of transport-based activity indicator.
237
2461.1.1 by Vincent Ladeuil
Fix 110204 by letting TestUIFactory encode password prompt.
238
4449.3.18 by Martin Pool
Fuse CLIUIFactory and TextUIFactory and deprecate the old name
239
class SilentUIFactory(UIFactory):
1185.49.21 by John Arbash Meinel
Refactored bzrlib/ui.py into a module with the possibility for multiple ui forms.
240
    """A UI Factory which never prints anything.
241
4449.3.29 by Martin Pool
More test UI factories support get_username and get_password
242
    This is the default UI, if another one is never registered by a program
243
    using bzrlib, and it's also active for example inside 'bzr serve'.
244
245
    Methods that try to read from the user raise an error; methods that do
246
    output do nothing.
1185.49.21 by John Arbash Meinel
Refactored bzrlib/ui.py into a module with the possibility for multiple ui forms.
247
    """
1594.1.1 by Robert Collins
Introduce new bzr progress bar api. ui_factory.nested_progress_bar.
248
3882.8.11 by Martin Pool
Choose the UIFactory class depending on the terminal capabilities
249
    def __init__(self):
4449.3.18 by Martin Pool
Fuse CLIUIFactory and TextUIFactory and deprecate the old name
250
        UIFactory.__init__(self)
3882.8.11 by Martin Pool
Choose the UIFactory class depending on the terminal capabilities
251
3882.8.4 by Martin Pool
All UI factories should support note()
252
    def note(self, msg):
253
        pass
254
4449.3.29 by Martin Pool
More test UI factories support get_username and get_password
255
    def get_username(self, prompt, **kwargs):
256
        return None
257
4711.1.7 by Martin Pool
Add UIFactory.show_error, show_warning, show_message
258
    def show_error(self, msg):
259
        pass
260
4711.1.8 by Martin Pool
Add show_warning and show_message tests and implementations
261
    def show_message(self, msg):
262
        pass
263
264
    def show_warning(self, msg):
265
        pass
266
3882.8.2 by Martin Pool
ProgressTask holds a reference to the ui that displays it
267
4449.3.25 by Martin Pool
Move CannedInputUIFactory to bzrlib.ui for reuse
268
class CannedInputUIFactory(SilentUIFactory):
269
    """A silent UI that return canned input."""
270
271
    def __init__(self, responses):
272
        self.responses = responses
273
4449.3.45 by Martin Pool
Add CannedInputUIFactory.assert_all_input_consumed
274
    def __repr__(self):
275
        return "%s(%r)" % (self.__class__.__name__, self.responses)
276
4449.3.25 by Martin Pool
Move CannedInputUIFactory to bzrlib.ui for reuse
277
    def get_boolean(self, prompt):
4449.3.27 by Martin Pool
More test updates to use CannedInputUIFactory
278
        return self.responses.pop(0)
4449.3.25 by Martin Pool
Move CannedInputUIFactory to bzrlib.ui for reuse
279
4597.3.37 by Vincent Ladeuil
Allows ui factories to query users for an integer.
280
    def get_integer(self, prompt):
281
        return self.responses.pop(0)
282
4449.3.29 by Martin Pool
More test UI factories support get_username and get_password
283
    def get_password(self, prompt='', **kwargs):
284
        return self.responses.pop(0)
285
286
    def get_username(self, prompt, **kwargs):
287
        return self.responses.pop(0)
4597.3.37 by Vincent Ladeuil
Allows ui factories to query users for an integer.
288
4449.3.45 by Martin Pool
Add CannedInputUIFactory.assert_all_input_consumed
289
    def assert_all_input_consumed(self):
290
        if self.responses:
291
            raise AssertionError("expected all input in %r to be consumed"
292
                % (self,))
4449.3.29 by Martin Pool
More test UI factories support get_username and get_password
293
4449.3.25 by Martin Pool
Move CannedInputUIFactory to bzrlib.ui for reuse
294
1185.1.29 by Robert Collins
merge merge tweaks from aaron, which includes latest .dev
295
ui_factory = SilentUIFactory()
4449.3.18 by Martin Pool
Fuse CLIUIFactory and TextUIFactory and deprecate the old name
296
# IMPORTANT: never import this symbol directly. ONLY ever access it as
297
# ui.ui_factory, so that you refer to the current value.
3882.8.11 by Martin Pool
Choose the UIFactory class depending on the terminal capabilities
298
299
300
def make_ui_for_terminal(stdin, stdout, stderr):
301
    """Construct and return a suitable UIFactory for a text mode program.
302
    """
4449.3.18 by Martin Pool
Fuse CLIUIFactory and TextUIFactory and deprecate the old name
303
    # this is now always TextUIFactory, which in turn decides whether it
304
    # should display progress bars etc
305
    from bzrlib.ui.text import TextUIFactory
306
    return TextUIFactory(stdin, stdout, stderr)
4449.3.15 by Martin Pool
Move NullProgressView and make_progress_view up to UIFactory base class
307
308
309
class NullProgressView(object):
310
    """Soak up and ignore progress information."""
311
312
    def clear(self):
313
        pass
314
315
    def show_progress(self, task):
316
        pass
317
318
    def show_transport_activity(self, transport, direction, byte_count):
319
        pass