/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 avatarsbox.py

  • Committer: Jelmer Vernooij
  • Date: 2012-03-29 13:15:14 UTC
  • mfrom: (786.1.1 register-lazy)
  • Revision ID: jelmer@samba.org-20120329131514-knrl1w2wzntx89rv
Use lazy registration.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2011 by Guillaume Hain (zedtux) <zedtux@zedroot.org>
 
2
#
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
#
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
#
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
from gi.repository import Gtk
 
18
from gi.repository import GdkPixbuf
 
19
 
 
20
from bzrlib.config import parse_username
 
21
 
 
22
from bzrlib.plugins.gtk.i18n import _i18n
 
23
from bzrlib.plugins.gtk.avatarproviders import (
 
24
    AvatarProviderGravatar,
 
25
    AvatarDownloaderWorker,
 
26
    )
 
27
 
 
28
 
 
29
class Avatar(Gtk.HBox):
 
30
    """ Author or committer avatar """
 
31
 
 
32
    def __init__(self, apparent_username):
 
33
        """ Constructor """
 
34
        super(Avatar, self).__init__()
 
35
 
 
36
        self.apparent_username = apparent_username
 
37
        self.username, self.email = parse_username(apparent_username)
 
38
        self.image = None
 
39
 
 
40
    def __eq__(self, other):
 
41
        return (self.apparent_username == other.apparent_username and
 
42
                self.username == other.username and
 
43
                self.email == other.email)
 
44
 
 
45
    def show_spinner(self):
 
46
        """
 
47
        Replace the current content of the Avatar with a Gtk.Spinner
 
48
        if an email address has been parsed. If not, show an Gtk.Label with
 
49
        the translatable 'No email' text.
 
50
        """
 
51
        if self.email:
 
52
            tooltip = _i18n("Retrieving avatar for %s...") % self.email
 
53
            spinner = Gtk.Spinner()
 
54
            spinner.start()
 
55
            self.pack_start(spinner, False, True, 0)
 
56
            spinner.set_tooltip_text(tooltip)
 
57
            spinner.set_size_request(20, 20)
 
58
            spinner.show()
 
59
        else:
 
60
            no_email = Gtk.Label(label=_i18n("No email"))
 
61
            self.pack_start(no_email, True, True, 0)
 
62
            self.set_tooltip_text(self.apparent_username)
 
63
            no_email.show()
 
64
 
 
65
    def show_image(self):
 
66
        """Replace the current content of the Avatar with the Gtk.Image """
 
67
        if self.email and self.image:
 
68
            children = self.get_children()
 
69
            if children != []:
 
70
                self.remove(children[0])
 
71
            self.pack_start(self.image, True, True, 0)
 
72
            self.image.set_tooltip_text(self.apparent_username)
 
73
            self.image.show()
 
74
 
 
75
 
 
76
class AvatarBox(Gtk.HBox):
 
77
    """HBox showing an avatar."""
 
78
 
 
79
    def __init__(self, homogeneous=False, spacing=0):
 
80
        super(AvatarBox, self).__init__(
 
81
            homogeneous=homogeneous, spacing=spacing)
 
82
        self.__avatars = {}
 
83
        self.avatar = None
 
84
        self.__displaying = None
 
85
 
 
86
    def reset_view(self):
 
87
        """Remove current avatars from the gtk box."""
 
88
        for child in self.get_children():
 
89
            self.remove(child)
 
90
        self.__displaying = None
 
91
 
 
92
    def have_avatar(self, avatar):
 
93
        """Return True if this box has the specified avatar.
 
94
        """
 
95
        return avatar.email in self.__avatars
 
96
 
 
97
    def showing(self, avatar):
 
98
        """Return True if the displaying avatar matches the specified one.
 
99
        """
 
100
        return self.__displaying and self.__displaying == avatar
 
101
 
 
102
    def append_avatars_with(self, avatar):
 
103
        """
 
104
        Append avatars collection with the given one if not already registed
 
105
        otherwise render it back.
 
106
        When an avatar is added this method True, otherwise, if the avatar
 
107
        was in the collection, return False.
 
108
        """
 
109
        if avatar.email and not avatar.email in self.__avatars:
 
110
            self.__avatars[avatar.email] = avatar
 
111
            self._new_cell_for_avatar(avatar)
 
112
            return True
 
113
        else:
 
114
            self.and_avatar_email(avatar.email).come_back_to_gui()
 
115
        return False
 
116
 
 
117
    def and_avatar_email(self, email):
 
118
        """
 
119
        Select the avatar from avatars collection
 
120
        in order to apply future actions
 
121
        """
 
122
        self.avatar = None
 
123
        if email and email in self.__avatars:
 
124
            self.avatar = self.__avatars[email]
 
125
        else:
 
126
            self.avatar = None
 
127
        return self
 
128
 
 
129
    def update_avatar_image_from_pixbuf_loader(self, loader):
 
130
        """Replace the Gtk.Spinner with the given loader."""
 
131
        if self.avatar:
 
132
            self.avatar.image = Gtk.Image()
 
133
            self.avatar.image.set_from_pixbuf(loader.get_pixbuf())
 
134
            self.avatar.show_image()
 
135
            self.__displaying = self.avatar
 
136
 
 
137
    def come_back_to_gui(self):
 
138
        """Render back avatar in the GUI."""
 
139
        if self.avatar:
 
140
            self.pack_start(self.avatar, True, True, 0)
 
141
            self.__displaying = self.avatar
 
142
        else:
 
143
            self._show_no_email()
 
144
 
 
145
    def _new_cell_for_avatar(self, avatar):
 
146
        """Create a new cell in this box with a Gtk.Spinner."""
 
147
        avatar.show_spinner()
 
148
        self.pack_start(avatar, True, True, 0)
 
149
        avatar.show()
 
150
        self.__displaying = avatar
 
151
 
 
152
    def _show_no_email(self):
 
153
        """Show a Gtk.Label with text 'No email'."""
 
154
        no_email = Gtk.Label(label=_i18n("No email"))
 
155
        self.pack_start(no_email, True, True, 0)
 
156
        no_email.show()
 
157
 
 
158
 
 
159
class AvatarsBox(Gtk.HBox):
 
160
    """GTK container for author and committer avatars."""
 
161
 
 
162
    def __init__(self):
 
163
        super(AvatarsBox, self).__init__(homogeneous=False, spacing=10)
 
164
        self.__committer_box = None
 
165
        self.__authors_box = None
 
166
        self._init_gui()
 
167
 
 
168
        # If more later you want to implement more avatar providers:
 
169
        # * Create a new class named AvatarProvider + provider_name that
 
170
        #   inherit from the AvatarProvider class.
 
171
        # * Implement a method that return url to use in the request.
 
172
 
 
173
        # For example, with Gravatar, the method return the complete url
 
174
        # with MD5 hash of the email address and put the value in a
 
175
        # gravatar_id field.
 
176
        # Then create a new worker (manage them in a python dictionary).
 
177
        provider = AvatarProviderGravatar()
 
178
        self.__worker = AvatarDownloaderWorker(
 
179
            provider.gravatar_id_for_email
 
180
        )
 
181
        # This callback method should be fired by all workers when a request
 
182
        # is done.
 
183
        self.__worker.set_callback_method(self._update_avatar_from_response)
 
184
        self.connect('destroy', self.on_destroy)
 
185
 
 
186
    def on_destroy(self, widget):
 
187
        self.__worker.stop()
 
188
        if self.__worker.is_alive():
 
189
            self.__worker.join()
 
190
 
 
191
    def add(self, username, role):
 
192
        """Add the given username in the role box and add in the worker queue.
 
193
        """
 
194
        avatar = Avatar(username)
 
195
        is_shown = self._role_box_for("committer").showing(avatar)
 
196
        if (role == "author" and not is_shown) or role == "committer":
 
197
            if self._role_box_for(role).append_avatars_with(avatar):
 
198
                self.__worker.queue(avatar.email)
 
199
 
 
200
    def merge(self, usernames, role):
 
201
        """Add avatars from a list"""
 
202
        for username in usernames:
 
203
            self.add(username, role)
 
204
 
 
205
    def reset(self):
 
206
        """
 
207
        Request a reset view for all boxes in order to show only avatars
 
208
        of the selected line in the revision view screen.
 
209
        """
 
210
        for role in ("committer", "author"):
 
211
            self._role_box_for(role).reset_view()
 
212
 
 
213
    def _init_gui(self):
 
214
        """Create boxes where avatars will be displayed."""
 
215
        # 2 Gtk.HBox: One for the committer and one for authors
 
216
        # Committer
 
217
        self.__committer_box = AvatarBox()
 
218
        self.__committer_box.set_size_request(80, 80)
 
219
        self.pack_end(self.__committer_box, False, True, 0)
 
220
        self.__committer_box.show()
 
221
        # Authors
 
222
        self.__authors_box = AvatarBox()
 
223
        self.pack_end(self.__authors_box, False, True, 0)
 
224
        self.__authors_box.set_spacing(10)
 
225
        self.__authors_box.show()
 
226
 
 
227
    def _update_avatar_from_response(self, response, email):
 
228
        """Callback method fired by avatar worker when finished.
 
229
 
 
230
        :param response: a urllib2.urlopen() return value.
 
231
        :param email: used to identify item from self.__avatars.
 
232
        """
 
233
        if email:
 
234
            # Convert downloaded image from provider to Gtk.Image
 
235
            loader = GdkPixbuf.PixbufLoader()
 
236
            loader.write(response.read())
 
237
            loader.close()
 
238
 
 
239
            for role in ["committer", "author"]:
 
240
                self._role_box_for(role).and_avatar_email(
 
241
                    email).update_avatar_image_from_pixbuf_loader(loader)
 
242
 
 
243
    def _role_box_for(self, role):
 
244
        """ Return the Gtk.HBox for the given role """
 
245
        if role == "committer":
 
246
            return self.__committer_box
 
247
        else:
 
248
            return self.__authors_box