1
# Copyright (C) 2011 by Guillaume Hain (zedtux) <zedtux@zedroot.org>
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.
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.
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
21
class AvatarDownloaderWorker(threading.Thread):
23
Threaded worker to retrieve avatar from a provider.
25
It create a persitante connection to the provider in order
26
to get avatars quickly through the same socket (urllib3).
29
def __init__(self, provider_method):
33
provider_method: Provider method that return fields
34
to send with the request.
36
threading.Thread.__init__(self)
37
self.__stop = threading.Event()
38
self.__queue = Queue.Queue()
40
self.__provider_method = provider_method
41
self.__end_thread = False
44
""" Stop this worker """
45
self.__end_thread = True
48
def set_callback_method(self, method):
49
""" Fire the given callback method when treatment is finished """
50
self.__callback_method = method
52
def queue(self, id_field):
54
Put in Queue the id_field to treat in the thread.
55
This id_field is for example with Gravatar the email address.
57
self.__queue.put(id_field)
60
""" Worker core code. """
61
while not self.__end_thread:
62
id_field = self.__queue.get()
63
# Call provider method to get fields to pass in the request
64
url = self.__provider_method(id_field)
66
response = urllib2.urlopen(url)
67
# Fire the callback method
68
if not self.__callback_method is None:
69
self.__callback_method(response, id_field)
70
self.__queue.task_done()