← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/ui.py

  • Committer: Martin Pool
  • Date: 2005-09-02 02:05:26 UTC
  • Revision ID: mbp@sourcefrog.net-20050902020526-0ab28bd5a998df70
- fix off-by-one in 'bzr log -r'

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/ui.py
1
 
# Copyright (C) 2005, 2006, 2007, 2008 Canonical Ltd
2
 
#
 
1
# Copyright (C) 2005 Canonical Ltd
 
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
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
#
 
7
 
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
12
 
#
 
12
 
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
 
17
 
 
18
 
17
19
"""UI abstraction.
18
20
 
19
21
This tells the library how to display things to the user.  Through this
26
28
displays no output.
27
29
"""
28
30
 
29
 
import sys
30
 
 
31
 
from bzrlib.lazy_import import lazy_import
32
 
lazy_import(globals(), """
33
 
import getpass
34
 
 
35
 
from bzrlib import (
36
 
    errors,
37
 
    osutils,
38
 
    progress,
39
 
    trace,
40
 
    )
41
 
""")
42
 
 
43
 
 
44
 
class UIFactory(object):
45
 
    """UI abstraction.
46
 
 
47
 
    This tells the library how to display things to the user.  Through this
48
 
    layer different applications can choose the style of UI.
49
 
    """
50
 
 
51
 
    def __init__(self):
52
 
        super(UIFactory, self).__init__()
53
 
        self._progress_bar_stack = None
54
 
 
55
 
    def get_password(self, prompt='', **kwargs):
56
 
        """Prompt the user for a password.
57
 
 
58
 
        :param prompt: The prompt to present the user
59
 
        :param kwargs: Arguments which will be expanded into the prompt.
60
 
                       This lets front ends display different things if
61
 
                       they so choose.
62
 
 
63
 
        :return: The password string, return None if the user canceled the
64
 
                 request. Note that we do not touch the encoding, users may
65
 
                 have whatever they see fit and the password should be
66
 
                 transported as is.
67
 
        """
68
 
        raise NotImplementedError(self.get_password)
69
 
 
70
 
    def nested_progress_bar(self):
71
 
        """Return a nested progress bar.
72
 
 
73
 
        When the bar has been finished with, it should be released by calling
74
 
        bar.finished().
75
 
        """
76
 
        raise NotImplementedError(self.nested_progress_bar)
77
 
 
78
 
    def clear_term(self):
79
 
        """Prepare the terminal for output.
80
 
 
81
 
        This will, for example, clear text progress bars, and leave the
82
 
        cursor at the leftmost position."""
83
 
        raise NotImplementedError(self.clear_term)
84
 
 
85
 
    def get_boolean(self, prompt):
86
 
        """Get a boolean question answered from the user. 
87
 
 
88
 
        :param prompt: a message to prompt the user with. Should be a single
89
 
        line without terminating \n.
90
 
        :return: True or False for y/yes or n/no.
91
 
        """
92
 
        raise NotImplementedError(self.get_boolean)
93
 
 
94
 
    def recommend_upgrade(self,
95
 
        current_format_name,
96
 
        basedir):
97
 
        # this should perhaps be in the TextUIFactory and the default can do
98
 
        # nothing
99
 
        trace.warning("%s is deprecated "
100
 
            "and a better format is available.\n"
101
 
            "It is recommended that you upgrade by "
102
 
            "running the command\n"
103
 
            "  bzr upgrade %s",
104
 
            current_format_name,
105
 
            basedir)
106
 
 
107
 
 
108
 
class CLIUIFactory(UIFactory):
109
 
    """Common behaviour for command line UI factories."""
110
 
 
111
 
    def __init__(self):
112
 
        super(CLIUIFactory, self).__init__()
113
 
        self.stdin = sys.stdin
114
 
 
115
 
    def get_boolean(self, prompt):
116
 
        self.clear_term()
117
 
        # FIXME: make a regexp and handle case variations as well.
118
 
        while True:
119
 
            self.prompt(prompt + "? [y/n]: ")
120
 
            line = self.stdin.readline()
121
 
            if line in ('y\n', 'yes\n'):
122
 
                return True
123
 
            if line in ('n\n', 'no\n'):
124
 
                return False
125
 
 
126
 
    def get_non_echoed_password(self, prompt):
127
 
        if not sys.stdin.isatty():
128
 
            raise errors.NotATerminal()
129
 
        encoding = osutils.get_terminal_encoding()
130
 
        return getpass.getpass(prompt.encode(encoding, 'replace'))
131
 
 
132
 
    def get_password(self, prompt='', **kwargs):
133
 
        """Prompt the user for a password.
134
 
 
135
 
        :param prompt: The prompt to present the user
136
 
        :param kwargs: Arguments which will be expanded into the prompt.
137
 
                       This lets front ends display different things if
138
 
                       they so choose.
139
 
        :return: The password string, return None if the user 
140
 
                 canceled the request.
141
 
        """
142
 
        prompt += ': '
143
 
        prompt = (prompt % kwargs)
144
 
        # There's currently no way to say 'i decline to enter a password'
145
 
        # as opposed to 'my password is empty' -- does it matter?
146
 
        return self.get_non_echoed_password(prompt)
147
 
 
148
 
    def prompt(self, prompt):
149
 
        """Emit prompt on the CLI."""
150
 
 
151
 
 
152
 
class SilentUIFactory(CLIUIFactory):
153
 
    """A UI Factory which never prints anything.
154
 
 
155
 
    This is the default UI, if another one is never registered.
156
 
    """
157
 
 
158
 
    def get_password(self, prompt='', **kwargs):
159
 
        return None
160
 
 
161
 
    def nested_progress_bar(self):
162
 
        if self._progress_bar_stack is None:
163
 
            self._progress_bar_stack = progress.ProgressBarStack(
164
 
                klass=progress.DummyProgress)
165
 
        return self._progress_bar_stack.get_nested()
166
 
 
167
 
    def clear_term(self):
168
 
        pass
169
 
 
170
 
    def recommend_upgrade(self, *args):
171
 
        pass
172
 
 
173
 
 
174
 
def clear_decorator(func, *args, **kwargs):
175
 
    """Decorator that clears the term"""
176
 
    ui_factory.clear_term()
177
 
    func(*args, **kwargs)
 
31
 
 
32
 
 
33
 
 
34
 
 
35
import bzrlib.progress
 
36
 
 
37
 
 
38
class TextUIFactory(object):
 
39
    def progress_bar(self):
 
40
 
 
41
        # this in turn is abstract, and creates either a tty or dots
 
42
        # bar depending on what we think of the terminal
 
43
        return bzrlib.progress.ProgressBar()
 
44
 
 
45
 
 
46
class SilentUIFactory(object):
 
47
    def progress_bar(self):
 
48
        return bzrlib.progress.DummyProgress()
178
49
 
179
50
 
180
51
ui_factory = SilentUIFactory()
181
 
"""IMPORTANT: never import this symbol directly. ONLY ever access it as 
182
 
ui.ui_factory."""