← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/ui/__init__.py

  • Committer: Vincent Ladeuil
  • Date: 2008-01-29 15:16:31 UTC
  • mto: (3206.1.1 trunk)
  • mto: This revision was merged to the branch mainline in revision 3207.
  • Revision ID: v.ladeuil+lp@free.fr-20080129151631-vqjd13tb405mobx6
Fix two more leaking tmp dirs, by reworking TransformPreview lock handling.

* bzrlib/tests/test_transform.py:
(TestTransformMerge): Revert previous patch and cleanly call
preview.finalize now that we can.

* bzrlib/tests/test_merge.py:
(TestMerge.test_make_preview_transform): Catch TransformPreview
leak.

* bzrlib/builtins.py:
(cmd_merge._do_preview): Finalize the TransformPreview or the
limbodir will stay in /tmp.

* bzrlib/transform.py:
(TreeTransformBase.__init__): Create the _deletiondir since it's
reffered to by finalize.
(TreeTransformBase.finalize): Delete the dir only if _deletiondir
is set.
(TreeTransform.__init__): Use a temp var for deletiondir and set
the attribute after the base class __init__ has been called.
(TransformPreview.__init__): Read locks the tree since finalize
wants to unlock it (as suggested by Aaron).

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/ui/__init__.py
1
 
# Copyright (C) 2005 Canonical Ltd
2
 
 
 
1
# Copyright (C) 2005, 2006, 2007 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
30
30
 
31
31
import sys
32
32
 
33
 
import bzrlib.progress
34
 
from bzrlib.symbol_versioning import *
 
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
""")
35
43
 
36
44
 
37
45
class UIFactory(object):
45
53
        super(UIFactory, self).__init__()
46
54
        self._progress_bar_stack = None
47
55
 
48
 
    @deprecated_method(zero_eight)
49
 
    def progress_bar(self):
50
 
        """See UIFactory.nested_progress_bar()."""
51
 
        raise NotImplementedError(self.progress_bar)
52
 
 
53
56
    def get_password(self, prompt='', **kwargs):
54
57
        """Prompt the user for a password.
55
58
 
57
60
        :param kwargs: Arguments which will be expanded into the prompt.
58
61
                       This lets front ends display different things if
59
62
                       they so choose.
60
 
        :return: The password string, return None if the user 
61
 
                 canceled the request.
 
63
 
 
64
        :return: The password string, return None if the user canceled the
 
65
                 request. Note that we do not touch the encoding, users may
 
66
                 have whatever they see fit and the password should be
 
67
                 transported as is.
62
68
        """
63
69
        raise NotImplementedError(self.get_password)
64
 
        
 
70
 
65
71
    def nested_progress_bar(self):
66
72
        """Return a nested progress bar.
67
73
 
68
 
        When the bar has been finished with, it should be released bu calling
 
74
        When the bar has been finished with, it should be released by calling
69
75
        bar.finished().
70
76
        """
71
77
        raise NotImplementedError(self.nested_progress_bar)
86
92
        """
87
93
        raise NotImplementedError(self.get_boolean)
88
94
 
 
95
    def recommend_upgrade(self,
 
96
        current_format_name,
 
97
        basedir):
 
98
        # this should perhaps be in the TextUIFactory and the default can do
 
99
        # nothing
 
100
        trace.warning("%s is deprecated "
 
101
            "and a better format is available.\n"
 
102
            "It is recommended that you upgrade by "
 
103
            "running the command\n"
 
104
            "  bzr upgrade %s",
 
105
            current_format_name,
 
106
            basedir)
 
107
 
89
108
 
90
109
class CLIUIFactory(UIFactory):
91
110
    """Common behaviour for command line UI factories."""
98
117
        self.clear_term()
99
118
        # FIXME: make a regexp and handle case variations as well.
100
119
        while True:
101
 
            self.prompt(prompt)
 
120
            self.prompt(prompt + "? [y/n]: ")
102
121
            line = self.stdin.readline()
103
122
            if line in ('y\n', 'yes\n'):
104
123
                return True
105
124
            if line in ('n\n', 'no\n'):
106
125
                return False
107
126
 
 
127
    def get_non_echoed_password(self, prompt):
 
128
        encoding = osutils.get_terminal_encoding()
 
129
        return getpass.getpass(prompt.encode(encoding, 'replace'))
 
130
 
 
131
    def get_password(self, prompt='', **kwargs):
 
132
        """Prompt the user for a password.
 
133
 
 
134
        :param prompt: The prompt to present the user
 
135
        :param kwargs: Arguments which will be expanded into the prompt.
 
136
                       This lets front ends display different things if
 
137
                       they so choose.
 
138
        :return: The password string, return None if the user 
 
139
                 canceled the request.
 
140
        """
 
141
        prompt += ': '
 
142
        prompt = (prompt % kwargs)
 
143
        # There's currently no way to say 'i decline to enter a password'
 
144
        # as opposed to 'my password is empty' -- does it matter?
 
145
        return self.get_non_echoed_password(prompt)
 
146
 
108
147
    def prompt(self, prompt):
109
148
        """Emit prompt on the CLI."""
110
149
 
115
154
    This is the default UI, if another one is never registered.
116
155
    """
117
156
 
118
 
    @deprecated_method(zero_eight)
119
 
    def progress_bar(self):
120
 
        """See UIFactory.nested_progress_bar()."""
121
 
        return bzrlib.progress.DummyProgress()
122
 
 
123
157
    def get_password(self, prompt='', **kwargs):
124
158
        return None
125
159
 
126
160
    def nested_progress_bar(self):
127
161
        if self._progress_bar_stack is None:
128
 
            self._progress_bar_stack = bzrlib.progress.ProgressBarStack(
129
 
                klass=bzrlib.progress.DummyProgress)
 
162
            self._progress_bar_stack = progress.ProgressBarStack(
 
163
                klass=progress.DummyProgress)
130
164
        return self._progress_bar_stack.get_nested()
131
165
 
132
166
    def clear_term(self):
133
167
        pass
134
168
 
 
169
    def recommend_upgrade(self, *args):
 
170
        pass
 
171
 
135
172
 
136
173
def clear_decorator(func, *args, **kwargs):
137
174
    """Decorator that clears the term"""