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