/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
4449.3.15 by Martin Pool
Move NullProgressView and make_progress_view up to UIFactory base class
182
    def make_progress_view(self):
183
        """Construct a new ProgressView object for this UI.
184
185
        Application code should normally not call this but instead
186
        nested_progress_bar().
187
        """
188
        return NullProgressView()
189
2323.6.2 by Martin Pool
Move responsibility for suggesting upgrades to ui object
190
    def recommend_upgrade(self,
191
        current_format_name,
192
        basedir):
2323.6.4 by Martin Pool
BzrDir._check_supported now also takes care of recommending upgrades, which
193
        # this should perhaps be in the TextUIFactory and the default can do
194
        # nothing
2323.6.2 by Martin Pool
Move responsibility for suggesting upgrades to ui object
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)
2323.6.4 by Martin Pool
BzrDir._check_supported now also takes care of recommending upgrades, which
202
3882.7.5 by Martin Pool
Further mockup of transport-based activity indicator.
203
    def report_transport_activity(self, transport, byte_count, direction):
204
        """Called by transports as they do IO.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
205
3882.7.5 by Martin Pool
Further mockup of transport-based activity indicator.
206
        This may update a progress bar, spinner, or similar display.
207
        By default it does nothing.
208
        """
209
        pass
210
4711.1.7 by Martin Pool
Add UIFactory.show_error, show_warning, show_message
211
    def show_error(self, msg):
212
        """Show an error message (not an exception) to the user.
213
        
214
        The message should not have an error prefix or trailing newline.  That
215
        will be added by the factory if appropriate. 
216
        """
217
        raise NotImplementedError(self.show_error)
218
219
    def show_message(self, msg):
220
        """Show a message to the user."""
221
        raise NotImplementedError(self.show_message)
222
223
    def show_warning(self, msg):
224
        """Show a warning to the user."""
225
        raise NotImplementedError(self.show_warning)
226
3882.7.5 by Martin Pool
Further mockup of transport-based activity indicator.
227
2461.1.1 by Vincent Ladeuil
Fix 110204 by letting TestUIFactory encode password prompt.
228
4449.3.18 by Martin Pool
Fuse CLIUIFactory and TextUIFactory and deprecate the old name
229
class SilentUIFactory(UIFactory):
1185.49.21 by John Arbash Meinel
Refactored bzrlib/ui.py into a module with the possibility for multiple ui forms.
230
    """A UI Factory which never prints anything.
231
4449.3.29 by Martin Pool
More test UI factories support get_username and get_password
232
    This is the default UI, if another one is never registered by a program
233
    using bzrlib, and it's also active for example inside 'bzr serve'.
234
235
    Methods that try to read from the user raise an error; methods that do
236
    output do nothing.
1185.49.21 by John Arbash Meinel
Refactored bzrlib/ui.py into a module with the possibility for multiple ui forms.
237
    """
1594.1.1 by Robert Collins
Introduce new bzr progress bar api. ui_factory.nested_progress_bar.
238
3882.8.11 by Martin Pool
Choose the UIFactory class depending on the terminal capabilities
239
    def __init__(self):
4449.3.18 by Martin Pool
Fuse CLIUIFactory and TextUIFactory and deprecate the old name
240
        UIFactory.__init__(self)
3882.8.11 by Martin Pool
Choose the UIFactory class depending on the terminal capabilities
241
3882.8.4 by Martin Pool
All UI factories should support note()
242
    def note(self, msg):
243
        pass
244
4449.3.29 by Martin Pool
More test UI factories support get_username and get_password
245
    def get_username(self, prompt, **kwargs):
246
        return None
247
4711.1.7 by Martin Pool
Add UIFactory.show_error, show_warning, show_message
248
    def show_error(self, msg):
249
        pass
250
4711.1.8 by Martin Pool
Add show_warning and show_message tests and implementations
251
    def show_message(self, msg):
252
        pass
253
254
    def show_warning(self, msg):
255
        pass
256
3882.8.2 by Martin Pool
ProgressTask holds a reference to the ui that displays it
257
4449.3.25 by Martin Pool
Move CannedInputUIFactory to bzrlib.ui for reuse
258
class CannedInputUIFactory(SilentUIFactory):
259
    """A silent UI that return canned input."""
260
261
    def __init__(self, responses):
262
        self.responses = responses
263
4449.3.45 by Martin Pool
Add CannedInputUIFactory.assert_all_input_consumed
264
    def __repr__(self):
265
        return "%s(%r)" % (self.__class__.__name__, self.responses)
266
4449.3.25 by Martin Pool
Move CannedInputUIFactory to bzrlib.ui for reuse
267
    def get_boolean(self, prompt):
4449.3.27 by Martin Pool
More test updates to use CannedInputUIFactory
268
        return self.responses.pop(0)
4449.3.25 by Martin Pool
Move CannedInputUIFactory to bzrlib.ui for reuse
269
4449.3.29 by Martin Pool
More test UI factories support get_username and get_password
270
    def get_password(self, prompt='', **kwargs):
271
        return self.responses.pop(0)
272
273
    def get_username(self, prompt, **kwargs):
274
        return self.responses.pop(0)
4449.3.45 by Martin Pool
Add CannedInputUIFactory.assert_all_input_consumed
275
    
276
    def assert_all_input_consumed(self):
277
        if self.responses:
278
            raise AssertionError("expected all input in %r to be consumed"
279
                % (self,))
4449.3.29 by Martin Pool
More test UI factories support get_username and get_password
280
4449.3.25 by Martin Pool
Move CannedInputUIFactory to bzrlib.ui for reuse
281
1185.1.29 by Robert Collins
merge merge tweaks from aaron, which includes latest .dev
282
ui_factory = SilentUIFactory()
4449.3.18 by Martin Pool
Fuse CLIUIFactory and TextUIFactory and deprecate the old name
283
# IMPORTANT: never import this symbol directly. ONLY ever access it as
284
# 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
285
286
287
def make_ui_for_terminal(stdin, stdout, stderr):
288
    """Construct and return a suitable UIFactory for a text mode program.
289
    """
4449.3.18 by Martin Pool
Fuse CLIUIFactory and TextUIFactory and deprecate the old name
290
    # this is now always TextUIFactory, which in turn decides whether it
291
    # should display progress bars etc
292
    from bzrlib.ui.text import TextUIFactory
293
    return TextUIFactory(stdin, stdout, stderr)
4449.3.15 by Martin Pool
Move NullProgressView and make_progress_view up to UIFactory base class
294
295
296
class NullProgressView(object):
297
    """Soak up and ignore progress information."""
298
299
    def clear(self):
300
        pass
301
302
    def show_progress(self, task):
303
        pass
304
305
    def show_transport_activity(self, transport, direction, byte_count):
306
        pass