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

  • Committer: Curtis Hovey
  • Date: 2012-03-22 17:14:22 UTC
  • mto: This revision was merged to the branch mainline in revision 789.
  • Revision ID: sinzui.is@verizon.net-20120322171422-l7bq6muqqgr36nxr
Ensure that the per/file message widget fits at least one line.

Show diffs side-by-side

added added

removed removed

Lines of Context:
avatarproviders.py
46
46
 
47
47
    def __init__(self, provider_method):
48
48
        """Constructor
49
 
        
 
49
 
50
50
        :param provider_method: Provider method that returns fields
51
51
                 to send with the request.
52
52
        """
53
53
        threading.Thread.__init__(self)
54
54
        self.__stop = threading.Event()
55
55
        self.__queue = Queue.Queue()
56
 
 
57
56
        self.__provider_method = provider_method
58
 
        self.__end_thread = False
 
57
        self.__callback_method = None
 
58
        self.__error_method = None
59
59
 
60
60
    def stop(self):
61
61
        """ Stop this worker """
62
 
        self.__end_thread = True
63
62
        self.__stop.set()
 
63
        while self.__queue.qsize() > 0:
 
64
            self.__queue.get_nowait()
 
65
            self.__queue.task_done()
 
66
        self.__queue.join()
 
67
 
 
68
    @property
 
69
    def is_running(self):
 
70
        return not self.__stop.is_set()
64
71
 
65
72
    def set_callback_method(self, method):
66
 
        """ Fire the given callback method when treatment is finished """
 
73
        """Fire the given callback method when treatment is finished."""
67
74
        self.__callback_method = method
68
75
 
 
76
    def set_error_method(self, method):
 
77
        """Fire the given callback when retrieving a avatar fails."""
 
78
        self.__error_method = method
 
79
 
69
80
    def queue(self, id_field):
70
81
        """Put in Queue the id_field to treat in the thread.
71
82
 
72
83
        This id_field is for example with Gravatar the email address.
73
84
        """
74
 
        self.__queue.put(id_field)
 
85
        if self.is_running:
 
86
            self.__queue.put(id_field)
 
87
            if not self.is_alive():
 
88
                self.start()
75
89
 
76
90
    def run(self):
77
91
        """Worker core code. """
78
 
        while not self.__end_thread:
79
 
            id_field = self.__queue.get()
80
 
            # Call provider method to get fields to pass in the request
81
 
            url = self.__provider_method(id_field)
82
 
            # Execute the request
83
 
            response = urllib2.urlopen(url)
84
 
            # Fire the callback method
85
 
            if not self.__callback_method is None:
86
 
                self.__callback_method(response, id_field)
87
 
            self.__queue.task_done()
 
92
        while self.is_running:
 
93
            try:
 
94
                id_field = self.__queue.get_nowait()
 
95
                # Call provider method to get fields to pass in the request
 
96
                url = self.__provider_method(id_field)
 
97
                # Execute the request
 
98
                try:
 
99
                    response = urllib2.urlopen(url)
 
100
                except urllib2.URLError, e:
 
101
                    if self.__error_method is not None:
 
102
                        self.__error_method(e)
 
103
                else:
 
104
                    # Fire the callback method
 
105
                    if not self.__callback_method is None:
 
106
                        self.__callback_method(response, id_field)
 
107
                self.__queue.task_done()
 
108
            except Queue.Empty:
 
109
                # There is no more work to do.
 
110
                pass
88
111
 
89
112
 
90
113
class AvatarProviderGravatar(AvatarProvider):
97
120
        """Return a gravatar URL for an email address.."""
98
121
        return self.get_base_url() + \
99
122
                urllib.urlencode({
100
 
                    'gravatar_id':hashlib.md5(email.lower()).hexdigest(),
101
 
                    'size':str(self.size)
 
123
                    'gravatar_id': hashlib.md5(email.lower()).hexdigest(),
 
124
                    'size': str(self.size)
102
125
                })