/b-gtk/fix-viz

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/b-gtk/fix-viz

« back to all changes in this revision

Viewing changes to ui.py

  • Committer: Jelmer Vernooij
  • Date: 2007-04-03 17:16:07 UTC
  • mfrom: (187 trunk)
  • mto: This revision was merged to the branch mainline in revision 188.
  • Revision ID: jelmer@samba.org-20070403171607-0zaskazouokrm4cq
Tags: bzr-gtk-0.15.2
Prepare for 0.15.2

Show diffs side-by-side

added added

removed removed

Lines of Context:
ui.py
1
1
# Copyright (C) 2006 Szilveszter Farkas <szilveszter.farkas@gmail.com>
2
 
# Copyright (C) 2007 Jelmer Vernooij <jelmer@samba.org>
3
2
 
4
3
# This program is free software; you can redistribute it and/or modify
5
4
# it under the terms of the GNU General Public License as published by
22
21
import gtk
23
22
import sys
24
23
 
25
 
from bzrlib import progress
 
24
import bzrlib.progress
26
25
from bzrlib.ui import UIFactory
27
26
 
28
27
 
30
29
    """ Prompt the user for a yes/no answer. """
31
30
    def __init__(self, prompt):
32
31
        gtk.Dialog.__init__(self)
33
 
 
 
32
        
34
33
        label = gtk.Label(prompt)
35
34
        self.vbox.pack_start(label, padding=10)
 
35
        
36
36
        self.vbox.show_all()
37
37
 
38
 
        self.add_buttons(gtk.STOCK_YES, gtk.RESPONSE_YES, gtk.STOCK_NO,
39
 
                         gtk.RESPONSE_NO)
 
38
        self.add_buttons(gtk.STOCK_YES, gtk.RESPONSE_YES, gtk.STOCK_NO, gtk.RESPONSE_NO)
40
39
 
41
40
 
42
41
class GtkProgressBar(gtk.ProgressBar):
43
 
 
44
 
    def __init__(self):
45
 
        gtk.ProgressBar.__init__(self)
 
42
    def __init__(self, stack):
 
43
        super(GtkProgressBar, self).__init__()
46
44
        self.set_fraction(0.0)
47
 
        self.current = None
48
 
        self.total = None
 
45
        self._stack = stack
 
46
 
 
47
    def finished(self):
 
48
        self._stack.remove(self)
 
49
 
 
50
    def clear(self):
 
51
        pass
49
52
 
50
53
    def tick(self):
51
 
        self.show()
52
54
        self.pulse()
53
55
 
54
 
    def update(self, msg=None, current_cnt=None, total_cnt=None):
55
 
        self.show()
56
 
        if current_cnt is not None:
57
 
            self.current = current_cnt
58
 
        if total_cnt is not None:
59
 
            self.total = total_cnt
 
56
    def update(self, msg=None, current=None, total=None):
60
57
        if msg is not None:
61
58
            self.set_text(msg)
62
 
        if None not in (self.current, self.total):
63
 
            self.fraction = float(self.current) / self.total
64
 
            if self.fraction < 0.0 or self.fraction > 1.0:
65
 
                raise AssertionError
66
 
            self.set_fraction(self.fraction)
 
59
        self.set_fraction(1.0 * current / total)
67
60
        while gtk.events_pending():
68
61
            gtk.main_iteration()
69
62
 
70
 
    def finished(self):
71
 
        self.hide()
72
 
 
73
 
    def clear(self):
74
 
        self.hide()
75
 
 
76
 
 
77
 
class ProgressBarWindow(gtk.Window):
78
 
 
 
63
 
 
64
class GtkProgressBarStack(gtk.Window):
79
65
    def __init__(self):
80
 
        super(ProgressBarWindow, self).__init__(type=gtk.WINDOW_TOPLEVEL)
 
66
        super(GtkProgressBarStack, self).__init__(type=gtk.WINDOW_TOPLEVEL)
81
67
        self.set_border_width(0)
82
68
        self.set_title("Progress")
83
69
        self.set_position(gtk.WIN_POS_CENTER_ALWAYS)
84
 
        self.pb = GtkProgressBar()
85
 
        self.add(self.pb)
86
 
        self.resize(250, 15)
 
70
        self.vbox = gtk.VBox()
 
71
        self.add(self.vbox)
87
72
        self.set_resizable(False)
88
73
 
89
 
    def tick(self, *args, **kwargs):
90
 
        self.show_all()
91
 
        self.pb.tick(*args, **kwargs)
92
 
 
93
 
    def update(self, *args, **kwargs):
94
 
        self.show_all()
95
 
        self.pb.update(*args, **kwargs)
96
 
 
97
 
    def finished(self):
98
 
        self.pb.finished()
99
 
        self.hide_all()
100
 
        self.destroy()
101
 
 
102
 
    def clear(self):
103
 
        self.pb.clear()
104
 
        self.hide_all()
105
 
 
106
 
 
107
 
class ProgressPanel(gtk.HBox):
108
 
 
109
 
    def __init__(self):
110
 
        super(ProgressPanel, self).__init__()
111
 
        image_loading = gtk.image_new_from_stock(gtk.STOCK_REFRESH,
112
 
                                                 gtk.ICON_SIZE_BUTTON)
113
 
        image_loading.show()
114
 
 
115
 
        self.pb = GtkProgressBar()
116
 
        self.set_spacing(5)
117
 
        self.set_border_width(5)
118
 
        self.pack_start(image_loading, False, False)
119
 
        self.pack_start(self.pb, True, True)
120
 
 
121
 
    def tick(self, *args, **kwargs):
122
 
        self.show_all()
123
 
        self.pb.tick(*args, **kwargs)
124
 
 
125
 
    def update(self, *args, **kwargs):
126
 
        self.show_all()
127
 
        self.pb.update(*args, **kwargs)
128
 
 
129
 
    def finished(self):
130
 
        self.pb.finished()
131
 
        self.hide_all()
132
 
 
133
 
    def clear(self):
134
 
        self.pb.clear()
135
 
        self.hide_all()
 
74
    def _adapt_size(self):
 
75
        self.resize(250, 15 * len(self.vbox.get_children()))
 
76
 
 
77
    def get_nested(self):
 
78
        nested = GtkProgressBar(self)
 
79
        self.vbox.pack_start(nested)
 
80
        self._adapt_size()
 
81
        self.show_all()
 
82
        return nested
 
83
 
 
84
    def remove(self, pb):
 
85
        self.vbox.remove(pb)
136
86
 
137
87
 
138
88
class PasswordDialog(gtk.Dialog):
139
89
    """ Prompt the user for a password. """
140
 
 
141
90
    def __init__(self, prompt):
142
91
        gtk.Dialog.__init__(self)
143
 
 
 
92
        
144
93
        label = gtk.Label(prompt)
145
94
        self.vbox.pack_start(label, padding=10)
146
 
 
 
95
        
147
96
        self.entry = gtk.Entry()
148
97
        self.entry.set_visibility(False)
149
98
        self.vbox.pack_end(self.entry, padding=10)
150
 
 
 
99
        
151
100
        self.vbox.show_all()
152
 
 
153
 
        self.add_buttons(gtk.STOCK_OK, gtk.RESPONSE_OK,
154
 
                         gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL)
155
 
 
 
101
        
 
102
        self.add_buttons(gtk.STOCK_OK, gtk.RESPONSE_OK, gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL)
 
103
    
156
104
    def _get_passwd(self):
157
105
        return self.entry.get_text()
158
106
 
162
110
class GtkUIFactory(UIFactory):
163
111
    """A UI factory for GTK user interfaces."""
164
112
 
165
 
    def __init__(self):
166
 
        """Create a GtkUIFactory"""
 
113
    def __init__(self,
 
114
                 stdout=None,
 
115
                 stderr=None):
 
116
        """Create a GtkUIFactory.
 
117
 
 
118
        """
167
119
        super(GtkUIFactory, self).__init__()
168
 
        self.set_progress_bar_widget(None)
169
 
 
170
 
    def set_progress_bar_widget(self, widget):
171
 
        self._progress_bar_widget = widget
 
120
        if stdout is None:
 
121
            self.stdout = sys.stdout
 
122
        else:
 
123
            self.stdout = stdout
 
124
        if stderr is None:
 
125
            self.stderr = sys.stderr
 
126
        else:
 
127
            self.stderr = stderr
 
128
        self._progress_bar_stack = None
172
129
 
173
130
    def get_boolean(self, prompt):
174
131
        """GtkDialog with yes/no answers"""
176
133
        response = dialog.run()
177
134
        dialog.destroy()
178
135
        return (response == gtk.RESPONSE_YES)
179
 
 
 
136
        
180
137
    def get_password(self, prompt='', **kwargs):
181
138
        """Prompt the user for a password.
182
139
 
196
153
        else:
197
154
            return None
198
155
 
199
 
    def _progress_all_finished(self):
200
 
        """See UIFactory._progress_all_finished"""
201
 
        pbw = self._progress_bar_widget
202
 
        if pbw:
203
 
            pbw.finished()
204
 
 
205
 
    def _progress_updated(self, task):
206
 
        """See UIFactory._progress_updated"""
207
 
        if self._progress_bar_widget is None:
208
 
            # Default to a window since nobody gave us a better mean to report
209
 
            # progress.
210
 
            self.set_progress_bar_widget(ProgressBarWindow())
211
 
        self._progress_bar_widget.update(task.msg,
212
 
                                         task.current_cnt, task.total_cnt)
213
 
 
 
156
    def nested_progress_bar(self):
 
157
        """Return a nested progress bar.
 
158
        """
 
159
        if self._progress_bar_stack is None:
 
160
            self._progress_bar_stack = GtkProgressBarStack()
 
161
        return self._progress_bar_stack.get_nested()
 
162
 
 
163
    def set_progress_bar_vbox(self, vbox):
 
164
        """Change the vbox to put progress bars in.
 
165
        """
 
166
        self._progress_bar_stack = vbox
 
167
 
 
168
    def clear_term(self):
 
169
        """Prepare the terminal for output.
 
170
 
 
171
        It has no sense when talking about GTK."""
 
172
        pass