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
17
from gi.repository import Gtk
18
from gi.repository import GdkPixbuf
19
from bzrlib.plugins.gtk import _i18n
20
20
from bzrlib.config import parse_username
22
from bzrlib.plugins.gtk.i18n import _i18n
23
from bzrlib.plugins.gtk.avatarproviders import (
24
AvatarProviderGravatar,
25
AvatarDownloaderWorker,
29
class Avatar(Gtk.HBox):
21
from bzrlib.plugins.gtk.avatarproviders import AvatarProviderGravatar, AvatarDownloaderWorker
24
class Avatar(gtk.Box):
30
25
""" Author or committer avatar """
32
27
def __init__(self, apparent_username):
33
28
""" Constructor """
34
super(Avatar, self).__init__()
29
gtk.Box.__init__(self)
36
31
self.apparent_username = apparent_username
37
32
self.username, self.email = parse_username(apparent_username)
40
35
def __eq__(self, other):
41
return (self.apparent_username == other.apparent_username and
42
self.username == other.username and
43
self.email == other.email)
37
Return True if attributes of the given avatar
38
match to current object attributes otherwise return False
40
return self.apparent_username == other.apparent_username and \
41
self.name == other.name and \
42
self.email == other.email
44
# ~~~~~ Public methods ~~~~~
45
45
def show_spinner(self):
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
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
49
the translatable 'No email' text.
52
tooltip = _i18n("Retrieving avatar for %s...") % self.email
53
spinner = Gtk.Spinner()
51
if not self.email is "":
52
spinner = gtk.Spinner()
55
self.pack_start(spinner, False, True, 0)
56
spinner.set_tooltip_text(tooltip)
54
self.pack_start(spinner, False)
55
spinner.set_tooltip_text(_i18n("Retrieving avatar for %s...") % self.email)
57
56
spinner.set_size_request(20, 20)
60
no_email = Gtk.Label(label=_i18n("No email"))
61
self.pack_start(no_email, True, True, 0)
59
no_email = gtk.Label(_i18n("No email"))
60
self.pack_start(no_email)
62
61
self.set_tooltip_text(self.apparent_username)
65
64
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()
70
self.remove(children[0])
71
self.pack_start(self.image, True, True, 0)
65
""" Replace the current content of the Avatar with the gtk.Image """
66
if not self.email is "" and self.image:
67
self.remove(self.get_children()[0])
68
self.pack_start(self.image)
72
69
self.image.set_tooltip_text(self.apparent_username)
76
class AvatarBox(Gtk.HBox):
77
"""HBox showing an avatar."""
73
class AvatarBox(gtk.HBox):
74
""" Improved gtk.HBox """
79
76
def __init__(self, homogeneous=False, spacing=0):
80
super(AvatarBox, self).__init__(
81
homogeneous=homogeneous, spacing=spacing)
78
gtk.HBox.__init__(self, homogeneous, spacing)
82
79
self.__avatars = {}
84
81
self.__displaying = None
84
# ~~~~~ Public methods ~~~~~
86
85
def reset_view(self):
87
"""Remove current avatars from the gtk box."""
86
""" Remove current avatars from the gtk box """
88
87
for child in self.get_children():
90
89
self.__displaying = None
92
91
def have_avatar(self, avatar):
93
"""Return True if this box has the specified avatar.
93
Return True if this box has registered given avatar,
94
otherwise return False
95
96
return avatar.email in self.__avatars
97
98
def showing(self, avatar):
98
"""Return True if the displaying avatar matches the specified one.
100
Return True if the displaying avatar is the same
101
as the given one otherwise return False
100
103
return self.__displaying and self.__displaying == avatar
102
105
def append_avatars_with(self, avatar):
104
107
Append avatars collection with the given one if not already registed
106
109
When an avatar is added this method True, otherwise, if the avatar
107
110
was in the collection, return False.
109
if avatar.email and not avatar.email in self.__avatars:
112
if not avatar.email is "" and not avatar.email in self.__avatars:
110
113
self.__avatars[avatar.email] = avatar
111
114
self._new_cell_for_avatar(avatar)
114
117
self.and_avatar_email(avatar.email).come_back_to_gui()
117
120
def and_avatar_email(self, email):
119
122
Select the avatar from avatars collection
120
123
in order to apply future actions
122
125
self.avatar = None
123
if email and email in self.__avatars:
126
if not email is "" and email in self.__avatars:
124
127
self.avatar = self.__avatars[email]
126
129
self.avatar = None
129
132
def update_avatar_image_from_pixbuf_loader(self, loader):
130
"""Replace the Gtk.Spinner with the given loader."""
133
""" Replace the gtk.Spinner with the given loader """
132
self.avatar.image = Gtk.Image()
135
self.avatar.image = gtk.Image()
133
136
self.avatar.image.set_from_pixbuf(loader.get_pixbuf())
134
137
self.avatar.show_image()
135
138
self.__displaying = self.avatar
137
140
def come_back_to_gui(self):
138
"""Render back avatar in the GUI."""
141
""" Render back avatar in the GUI """
140
self.pack_start(self.avatar, True, True, 0)
143
self.pack_start(self.avatar)
141
144
self.__displaying = self.avatar
143
146
self._show_no_email()
149
# ~~~~~ Private methods ~~~~~~
145
150
def _new_cell_for_avatar(self, avatar):
146
"""Create a new cell in this box with a Gtk.Spinner."""
151
""" Create a new cell in this box with a gtk.Spinner """
147
152
avatar.show_spinner()
148
self.pack_start(avatar, True, True, 0)
153
self.pack_start(avatar)
150
155
self.__displaying = avatar
152
157
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)
158
""" Show a gtk.Label with test 'No email' """
159
no_email = gtk.Label(_i18n("No email"))
160
self.pack_start(no_email)
159
class AvatarsBox(Gtk.HBox):
160
"""GTK container for author and committer avatars."""
164
class AvatarsBox(gtk.HBox):
165
""" GTK container for authors and committers avatars """
162
167
def __init__(self):
163
super(AvatarsBox, self).__init__(homogeneous=False, spacing=10)
169
gtk.HBox.__init__(self, False, 10)
164
171
self.__committer_box = None
165
172
self.__authors_box = None
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.
175
# If more later you want to implement more avatar providers, to it like this:
176
# Create a new class named AvatarProvider + provider_name that inherit from
177
# the AvatarProvider class.
178
# Implement a method that return url to use in the request.
173
179
# For example, with Gravatar, the method return the complete url
174
# with MD5 hash of the email address and put the value in a
176
# Then create a new worker (manage them in a python dictionary).
180
# with MD5 hash of the email address and put the value in a gravatar_id field.
181
# Then create a new worker (manage them in a python dictionnary).
177
182
provider = AvatarProviderGravatar()
178
183
self.__worker = AvatarDownloaderWorker(
179
184
provider.gravatar_id_for_email
181
# This callback method should be fired by all workers when a request
186
# This callback method should be fired bt all workers when a request is done.
183
187
self.__worker.set_callback_method(self._update_avatar_from_response)
184
self.connect('destroy', self.on_destroy)
186
def on_destroy(self, widget):
188
if self.__worker.is_alive():
188
self.__worker.start()
191
# ~~~~~ Public methods ~~~~~
191
192
def add(self, username, role):
192
"""Add the given username in the role box and add in the worker queue.
194
Add the given username in the right role box and add in the worker queue.
195
Here again: If you want to implement more providers, you should add the
196
avatar request in all workers queue.
194
198
avatar = Avatar(username)
195
is_shown = self._role_box_for("committer").showing(avatar)
196
if (role == "author" and not is_shown) or role == "committer":
199
if role is "author" and not self._role_box_for("committer").showing(avatar) or role is "committer":
197
200
if self._role_box_for(role).append_avatars_with(avatar):
198
201
self.__worker.queue(avatar.email)
200
203
def merge(self, usernames, role):
201
"""Add avatars from a list"""
204
""" Add avatars from a list """
202
205
for username in usernames:
203
206
self.add(username, role)
207
210
Request a reset view for all boxes in order to show only avatars
208
211
of the selected line in the revision view screen.
210
for role in ("committer", "author"):
211
self._role_box_for(role).reset_view()
213
[self._role_box_for(role).reset_view() for role in ["committer", "author"]]
216
# ~~~~~ Private methods ~~~~~
213
217
def _init_gui(self):
214
"""Create boxes where avatars will be displayed."""
215
# 2 Gtk.HBox: One for the committer and one for authors
218
""" Create boxes where avatars will be displayed """
219
# 2 gtk.HBox: One for the committer and one for authors
217
221
self.__committer_box = AvatarBox()
218
222
self.__committer_box.set_size_request(80, 80)
219
self.pack_end(self.__committer_box, False, True, 0)
223
self.pack_end(self.__committer_box, False)
220
224
self.__committer_box.show()
222
226
self.__authors_box = AvatarBox()
223
self.pack_end(self.__authors_box, False, True, 0)
227
self.pack_end(self.__authors_box, False)
224
228
self.__authors_box.set_spacing(10)
225
229
self.__authors_box.show()
227
231
def _update_avatar_from_response(self, response, email):
228
"""Callback method fired by avatar worker when finished.
230
:param response: a urllib2.urlopen() return value.
231
:param email: used to identify item from self.__avatars.
234
# Convert downloaded image from provider to Gtk.Image
235
loader = GdkPixbuf.PixbufLoader()
233
Callback method fired by avatar worker when finished.
235
response is a urllib2.urlopen() return value.
236
email is used to identify item from self.__avatars.
239
# Convert downloaded image from provider to gtk.Image
240
loader = gtk.gdk.PixbufLoader()
236
241
loader.write(response.read())
239
244
for role in ["committer", "author"]:
240
self._role_box_for(role).and_avatar_email(
241
email).update_avatar_image_from_pixbuf_loader(loader)
245
self._role_box_for(role).and_avatar_email(email).update_avatar_image_from_pixbuf_loader(loader)
243
247
def _role_box_for(self, role):
244
""" Return the Gtk.HBox for the given role """
245
if role == "committer":
246
return self.__committer_box
248
return self.__authors_box
248
""" Return the gtk.HBox for the given role """
249
return self.__committer_box if role is "committer" else self.__authors_box