/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
19
18
"""GTK UI
20
19
"""
21
20
 
22
 
from gi.repository import Gtk
 
21
import gtk
 
22
import sys
23
23
 
 
24
import bzrlib.progress
24
25
from bzrlib.ui import UIFactory
25
26
 
26
27
 
27
 
class PromptDialog(Gtk.Dialog):
28
 
    """Prompt the user for a yes/no answer."""
29
 
 
 
28
class PromptDialog(gtk.Dialog):
 
29
    """ Prompt the user for a yes/no answer. """
30
30
    def __init__(self, prompt):
31
 
        GObject.GObject.__init__(self)
32
 
 
33
 
        label = Gtk.Label(label=prompt)
34
 
        self.vbox.pack_start(label, True, True, 10)
 
31
        gtk.Dialog.__init__(self)
 
32
        
 
33
        label = gtk.Label(prompt)
 
34
        self.vbox.pack_start(label, padding=10)
 
35
        
35
36
        self.vbox.show_all()
36
37
 
37
 
        self.add_buttons(Gtk.STOCK_YES, Gtk.ResponseType.YES, Gtk.STOCK_NO,
38
 
                         Gtk.ResponseType.NO)
39
 
 
40
 
 
41
 
class GtkProgressBar(Gtk.ProgressBar):
42
 
 
43
 
    def __init__(self):
44
 
        GObject.GObject.__init__(self)
 
38
        self.add_buttons(gtk.STOCK_YES, gtk.RESPONSE_YES, gtk.STOCK_NO, gtk.RESPONSE_NO)
 
39
 
 
40
 
 
41
class GtkProgressBar(gtk.ProgressBar):
 
42
    def __init__(self, stack):
 
43
        super(GtkProgressBar, self).__init__()
45
44
        self.set_fraction(0.0)
46
 
        self.current = None
47
 
        self.total = None
 
45
        self._stack = stack
 
46
 
 
47
    def finished(self):
 
48
        self._stack.remove(self)
 
49
 
 
50
    def clear(self):
 
51
        pass
48
52
 
49
53
    def tick(self):
50
 
        self.show()
51
54
        self.pulse()
52
55
 
53
 
    def update(self, msg=None, current_cnt=None, total_cnt=None):
54
 
        self.show()
55
 
        if current_cnt is not None:
56
 
            self.current = current_cnt
57
 
        if total_cnt is not None:
58
 
            self.total = total_cnt
 
56
    def update(self, msg=None, current=None, total=None):
59
57
        if msg is not None:
60
58
            self.set_text(msg)
61
 
        if None not in (self.current, self.total):
62
 
            self.fraction = float(self.current) / self.total
63
 
            if self.fraction < 0.0 or self.fraction > 1.0:
64
 
                raise AssertionError
65
 
            self.set_fraction(self.fraction)
66
 
        while Gtk.events_pending():
67
 
            Gtk.main_iteration()
68
 
 
69
 
    def finished(self):
70
 
        self.hide()
71
 
 
72
 
    def clear(self):
73
 
        self.hide()
74
 
 
75
 
 
76
 
class ProgressBarWindow(Gtk.Window):
77
 
 
 
59
        self.set_fraction(1.0 * current / total)
 
60
        while gtk.events_pending():
 
61
            gtk.main_iteration()
 
62
 
 
63
 
 
64
class GtkProgressBarStack(gtk.Window):
78
65
    def __init__(self):
79
 
        super(ProgressBarWindow, self).__init__(type=Gtk.WindowType.TOPLEVEL)
 
66
        super(GtkProgressBarStack, self).__init__(type=gtk.WINDOW_TOPLEVEL)
80
67
        self.set_border_width(0)
81
68
        self.set_title("Progress")
82
 
        self.set_position(Gtk.WindowPosition.CENTER_ALWAYS)
83
 
        self.pb = GtkProgressBar()
84
 
        self.add(self.pb)
85
 
        self.resize(250, 15)
 
69
        self.set_position(gtk.WIN_POS_CENTER_ALWAYS)
 
70
        self.vbox = gtk.VBox()
 
71
        self.add(self.vbox)
86
72
        self.set_resizable(False)
87
73
 
88
 
    def tick(self, *args, **kwargs):
89
 
        self.show_all()
90
 
        self.pb.tick(*args, **kwargs)
91
 
 
92
 
    def update(self, *args, **kwargs):
93
 
        self.show_all()
94
 
        self.pb.update(*args, **kwargs)
95
 
 
96
 
    def finished(self):
97
 
        self.pb.finished()
98
 
        self.hide_all()
99
 
        self.destroy()
100
 
 
101
 
    def clear(self):
102
 
        self.pb.clear()
103
 
        self.hide_all()
104
 
 
105
 
 
106
 
class ProgressPanel(Gtk.HBox):
107
 
 
108
 
    def __init__(self):
109
 
        super(ProgressPanel, self).__init__()
110
 
        image_loading = Gtk.Image.new_from_stock(Gtk.STOCK_REFRESH,
111
 
                                                 Gtk.IconSize.BUTTON)
112
 
        image_loading.show()
113
 
 
114
 
        self.pb = GtkProgressBar()
115
 
        self.set_spacing(5)
116
 
        self.set_border_width(5)
117
 
        self.pack_start(image_loading, False, False)
118
 
        self.pack_start(self.pb, True, True)
119
 
 
120
 
    def tick(self, *args, **kwargs):
121
 
        self.show_all()
122
 
        self.pb.tick(*args, **kwargs)
123
 
 
124
 
    def update(self, *args, **kwargs):
125
 
        self.show_all()
126
 
        self.pb.update(*args, **kwargs)
127
 
 
128
 
    def finished(self):
129
 
        self.pb.finished()
130
 
        self.hide_all()
131
 
 
132
 
    def clear(self):
133
 
        self.pb.clear()
134
 
        self.hide_all()
135
 
 
136
 
 
137
 
class PasswordDialog(Gtk.Dialog):
 
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)
 
86
 
 
87
 
 
88
class PasswordDialog(gtk.Dialog):
138
89
    """ Prompt the user for a password. """
139
 
 
140
90
    def __init__(self, prompt):
141
 
        GObject.GObject.__init__(self)
142
 
 
143
 
        label = Gtk.Label(label=prompt)
144
 
        self.vbox.pack_start(label, True, True, 10)
145
 
 
146
 
        self.entry = Gtk.Entry()
 
91
        gtk.Dialog.__init__(self)
 
92
        
 
93
        label = gtk.Label(prompt)
 
94
        self.vbox.pack_start(label, padding=10)
 
95
        
 
96
        self.entry = gtk.Entry()
147
97
        self.entry.set_visibility(False)
148
98
        self.vbox.pack_end(self.entry, padding=10)
149
 
 
 
99
        
150
100
        self.vbox.show_all()
151
 
 
152
 
        self.add_buttons(Gtk.STOCK_OK, Gtk.ResponseType.OK,
153
 
                         Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL)
154
 
 
 
101
        
 
102
        self.add_buttons(gtk.STOCK_OK, gtk.RESPONSE_OK, gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL)
 
103
    
155
104
    def _get_passwd(self):
156
105
        return self.entry.get_text()
157
106
 
161
110
class GtkUIFactory(UIFactory):
162
111
    """A UI factory for GTK user interfaces."""
163
112
 
164
 
    def __init__(self):
165
 
        """Create a GtkUIFactory"""
 
113
    def __init__(self,
 
114
                 stdout=None,
 
115
                 stderr=None):
 
116
        """Create a GtkUIFactory.
 
117
 
 
118
        """
166
119
        super(GtkUIFactory, self).__init__()
167
 
        self.set_progress_bar_widget(None)
168
 
 
169
 
    def set_progress_bar_widget(self, widget):
170
 
        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
171
129
 
172
130
    def get_boolean(self, prompt):
173
131
        """GtkDialog with yes/no answers"""
174
132
        dialog = PromptDialog(prompt)
175
133
        response = dialog.run()
176
134
        dialog.destroy()
177
 
        return (response == Gtk.ResponseType.YES)
178
 
 
 
135
        return (response == gtk.RESPONSE_YES)
 
136
        
179
137
    def get_password(self, prompt='', **kwargs):
180
138
        """Prompt the user for a password.
181
139
 
190
148
        response = dialog.run()
191
149
        passwd = dialog.passwd
192
150
        dialog.destroy()
193
 
        if response == Gtk.ResponseType.OK:
 
151
        if response == gtk.RESPONSE_OK:
194
152
            return passwd
195
153
        else:
196
154
            return None
197
155
 
198
 
    def _progress_all_finished(self):
199
 
        """See UIFactory._progress_all_finished"""
200
 
        pbw = self._progress_bar_widget
201
 
        if pbw:
202
 
            pbw.finished()
203
 
 
204
 
    def _progress_updated(self, task):
205
 
        """See UIFactory._progress_updated"""
206
 
        if self._progress_bar_widget is None:
207
 
            # Default to a window since nobody gave us a better mean to report
208
 
            # progress.
209
 
            self.set_progress_bar_widget(ProgressBarWindow())
210
 
        self._progress_bar_widget.update(task.msg,
211
 
                                         task.current_cnt, task.total_cnt)
212
 
 
 
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