~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/ui/__init__.py

  • Committer: Martin Pool
  • Date: 2006-03-21 12:26:54 UTC
  • mto: This revision was merged to the branch mainline in revision 1621.
  • Revision ID: mbp@sourcefrog.net-20060321122654-514047ed65795a17
New developer commands 'weave-list' and 'weave-join'.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2007 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
28
28
displays no output.
29
29
"""
30
30
 
31
 
import sys
32
 
 
33
 
from bzrlib.lazy_import import lazy_import
34
 
lazy_import(globals(), """
35
 
import getpass
36
 
 
37
 
from bzrlib import (
38
 
    osutils,
39
 
    progress,
40
 
    trace,
41
 
    )
42
 
""")
43
 
 
44
 
from bzrlib.symbol_versioning import (deprecated_method, zero_eight)
 
31
 
 
32
import bzrlib.progress
 
33
from bzrlib.symbol_versioning import *
45
34
 
46
35
 
47
36
class UIFactory(object):
67
56
        :param kwargs: Arguments which will be expanded into the prompt.
68
57
                       This lets front ends display different things if
69
58
                       they so choose.
70
 
 
71
 
        :return: The password string, return None if the user canceled the
72
 
                 request. Note that we do not touch the encoding, users may
73
 
                 have whatever they see fit and the password should be
74
 
                 transported as is.
 
59
        :return: The password string, return None if the user 
 
60
                 canceled the request.
75
61
        """
76
62
        raise NotImplementedError(self.get_password)
77
 
 
 
63
        
78
64
    def nested_progress_bar(self):
79
65
        """Return a nested progress bar.
80
66
 
81
 
        When the bar has been finished with, it should be released by calling
 
67
        When the bar has been finished with, it should be released bu calling
82
68
        bar.finished().
83
69
        """
84
70
        raise NotImplementedError(self.nested_progress_bar)
85
71
 
86
 
    def clear_term(self):
87
 
        """Prepare the terminal for output.
88
 
 
89
 
        This will, for example, clear text progress bars, and leave the
90
 
        cursor at the leftmost position."""
91
 
        raise NotImplementedError(self.clear_term)
92
 
 
93
 
    def get_boolean(self, prompt):
94
 
        """Get a boolean question answered from the user. 
95
 
 
96
 
        :param prompt: a message to prompt the user with. Should be a single
97
 
        line without terminating \n.
98
 
        :return: True or False for y/yes or n/no.
99
 
        """
100
 
        raise NotImplementedError(self.get_boolean)
101
 
 
102
 
    def recommend_upgrade(self,
103
 
        current_format_name,
104
 
        basedir):
105
 
        # this should perhaps be in the TextUIFactory and the default can do
106
 
        # nothing
107
 
        trace.warning("%s is deprecated "
108
 
            "and a better format is available.\n"
109
 
            "It is recommended that you upgrade by "
110
 
            "running the command\n"
111
 
            "  bzr upgrade %s",
112
 
            current_format_name,
113
 
            basedir)
114
 
 
115
 
 
116
 
class CLIUIFactory(UIFactory):
117
 
    """Common behaviour for command line UI factories."""
118
 
 
119
 
    def __init__(self):
120
 
        super(CLIUIFactory, self).__init__()
121
 
        self.stdin = sys.stdin
122
 
 
123
 
    def get_boolean(self, prompt):
124
 
        self.clear_term()
125
 
        # FIXME: make a regexp and handle case variations as well.
126
 
        while True:
127
 
            self.prompt(prompt + "? [y/n]: ")
128
 
            line = self.stdin.readline()
129
 
            if line in ('y\n', 'yes\n'):
130
 
                return True
131
 
            if line in ('n\n', 'no\n'):
132
 
                return False
133
 
 
134
 
    def get_non_echoed_password(self, prompt):
135
 
        encoding = osutils.get_terminal_encoding()
136
 
        return getpass.getpass(prompt.encode(encoding, 'replace'))
137
 
 
138
 
    def get_password(self, prompt='', **kwargs):
139
 
        """Prompt the user for a password.
140
 
 
141
 
        :param prompt: The prompt to present the user
142
 
        :param kwargs: Arguments which will be expanded into the prompt.
143
 
                       This lets front ends display different things if
144
 
                       they so choose.
145
 
        :return: The password string, return None if the user 
146
 
                 canceled the request.
147
 
        """
148
 
        prompt += ': '
149
 
        prompt = (prompt % kwargs)
150
 
        # There's currently no way to say 'i decline to enter a password'
151
 
        # as opposed to 'my password is empty' -- does it matter?
152
 
        return self.get_non_echoed_password(prompt)
153
 
 
154
 
    def prompt(self, prompt):
155
 
        """Emit prompt on the CLI."""
156
 
 
157
 
 
158
 
class SilentUIFactory(CLIUIFactory):
 
72
 
 
73
class SilentUIFactory(UIFactory):
159
74
    """A UI Factory which never prints anything.
160
75
 
161
76
    This is the default UI, if another one is never registered.
164
79
    @deprecated_method(zero_eight)
165
80
    def progress_bar(self):
166
81
        """See UIFactory.nested_progress_bar()."""
167
 
        return progress.DummyProgress()
 
82
        return bzrlib.progress.DummyProgress()
168
83
 
169
84
    def get_password(self, prompt='', **kwargs):
170
85
        return None
171
86
 
172
87
    def nested_progress_bar(self):
173
88
        if self._progress_bar_stack is None:
174
 
            self._progress_bar_stack = progress.ProgressBarStack(
175
 
                klass=progress.DummyProgress)
 
89
            self._progress_bar_stack = bzrlib.progress.ProgressBarStack(
 
90
                klass=bzrlib.progress.DummyProgress)
176
91
        return self._progress_bar_stack.get_nested()
177
92
 
178
 
    def clear_term(self):
179
 
        pass
180
 
 
181
 
    def recommend_upgrade(self, *args):
182
 
        pass
183
 
 
184
 
 
185
 
def clear_decorator(func, *args, **kwargs):
186
 
    """Decorator that clears the term"""
187
 
    ui_factory.clear_term()
188
 
    func(*args, **kwargs)
189
 
 
190
93
 
191
94
ui_factory = SilentUIFactory()
192
95
"""IMPORTANT: never import this symbol directly. ONLY ever access it as