cola/gravatar.py

   1 from __future__ import division, absolute_import, unicode_literals
   2 import time
   3 import hashlib
   4
   5 from qtpy import QtCore
   6 from qtpy import QtGui
   7 from qtpy import QtWidgets
   8 from qtpy import QtNetwork
   9
  10 from . import core
  11 from . import icons
  12 from .compat import bstr
  13 from .compat import ustr
  14 from .compat import parse
  15 from .widgets import defs
  16
  17
  18 class Gravatar(object):
  19
  20     @staticmethod
  21     def url_for_email(email, imgsize):
  22         email_hash = core.decode(hashlib.md5(core.encode(email)).hexdigest())
  23         # Python2.6 requires byte strings for urllib2.quote() so we have
  24         # to force
  25         default_url = 'https://git-cola.github.io/images/git-64x64.jpg'
  26         encoded_url = parse.quote(core.encode(default_url), core.encode(''))
  27         query = '?s=%d&d=%s' % (imgsize, core.decode(encoded_url))
  28         url = 'https://gravatar.com/avatar/' + email_hash + query
  29         return url
  30
  31
  32 class GravatarLabel(QtWidgets.QLabel):
  33
  34     def __init__(self, parent=None):
  35         QtWidgets.QLabel.__init__(self, parent)
  36
  37         self.email = None
  38         self.response = None
  39         self.timeout = 0
  40         self.imgsize = defs.medium_icon
  41         self.pixmaps = {}
  42
  43         self.network = QtNetwork.QNetworkAccessManager()
  44         self.network.finished.connect(self.network_finished)
  45
  46     def set_email(self, email):
  47         if email in self.pixmaps:
  48             self.setPixmap(self.pixmaps[email])
  49             return
  50         if (self.timeout > 0 and
  51                 (int(time.time()) - self.timeout) < (5 * 60)):
  52             self.set_pixmap_from_response()
  53             return
  54         if email == self.email and self.response is not None:
  55             self.set_pixmap_from_response()
  56             return
  57         self.email = email
  58         self.request(email)
  59
  60     def request(self, email):
  61         url = Gravatar.url_for_email(email, self.imgsize)
  62         self.network.get(QtNetwork.QNetworkRequest(QtCore.QUrl(url)))
  63
  64     def default_pixmap_as_bytes(self):
  65         xres = self.imgsize
  66         pixmap = icons.cola().pixmap(xres)
  67         byte_array = QtCore.QByteArray()
  68         buf = QtCore.QBuffer(byte_array)
  69         buf.open(QtCore.QIODevice.WriteOnly)
  70         pixmap.save(buf, 'PNG')
  71         buf.close()
  72         return byte_array
  73
  74     def network_finished(self, reply):
  75         email = self.email
  76
  77         header = QtCore.QByteArray(bstr('Location'))
  78         location = ustr(reply.rawHeader(header)).strip()
  79         if location:
  80             request_location = Gravatar.url_for_email(self.email, self.imgsize)
  81             relocated = location != request_location
  82         else:
  83             relocated = False
  84
  85         if reply.error() == QtNetwork.QNetworkReply.NoError:
  86             if relocated:
  87                 # We could do get_url(parse.unquote(location)) to
  88                 # download the default image.
  89                 # Save bandwidth by using a pixmap.
  90                 self.response = self.default_pixmap_as_bytes()
  91             else:
  92                 self.response = reply.readAll()
  93             self.timeout = 0
  94         else:
  95             self.response = self.default_pixmap_as_bytes()
  96             self.timeout = int(time.time())
  97
  98         pixmap = self.set_pixmap_from_response()
  99
 100         # If the email has not changed (e.g. no other requests)
 101         # then we know that this pixmap corresponds to this specific
 102         # email address.  We can't blindly trust self.email else
 103         # we may add cache entries for thee wrong email address.
 104         url = Gravatar.url_for_email(email, self.imgsize)
 105         if url == reply.url().toString():
 106             self.pixmaps[email] = pixmap
 107
 108     def set_pixmap_from_response(self):
 109         pixmap = QtGui.QPixmap()
 110         pixmap.loadFromData(self.response)
 111         self.setPixmap(pixmap)
 112         return pixmap