forked from 00dani/lemoncurry
56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
from django.db.models import Q
|
|
from django.http import HttpResponse, HttpResponseRedirect
|
|
from django.views.decorators.cache import cache_page
|
|
from PIL import Image
|
|
|
|
from lemoncurry import utils
|
|
from .models import User
|
|
|
|
|
|
def try_libravatar_org(hash, get):
|
|
url = "https://seccdn.libravatar.org/avatar/" + hash
|
|
if get:
|
|
url += "?" + get.urlencode()
|
|
return HttpResponseRedirect(url)
|
|
|
|
|
|
@cache_page(60 * 15)
|
|
def libravatar(request, hash):
|
|
g = request.GET
|
|
size = g.get("s", g.get("size", 80))
|
|
try:
|
|
size = int(size)
|
|
except ValueError:
|
|
size = 80
|
|
if not 1 <= size <= 512:
|
|
size = 512
|
|
|
|
if len(hash) == 32:
|
|
where = Q(email_md5=hash)
|
|
elif len(hash) == 64:
|
|
where = Q(email_sha256=hash) | Q(openid_sha256=hash)
|
|
else:
|
|
return utils.bad_req("hash must be either md5 or sha256")
|
|
|
|
# If the user doesn't exist or lacks an avatar, see if libravatar.org has
|
|
# one for them - libravatar.org falls back to Gravatar when possible (only
|
|
# for MD5 hashes, since Gravatar doesn't support SHA-256), so this ensures
|
|
# all the most likely places are checked.
|
|
try:
|
|
user = User.objects.get(where)
|
|
except User.DoesNotExist:
|
|
return try_libravatar_org(hash, g)
|
|
|
|
if not user.avatar:
|
|
return try_libravatar_org(hash, g)
|
|
|
|
im = Image.open(user.avatar)
|
|
image_type = im.format
|
|
natural_size = min(im.size)
|
|
|
|
im = im.crop((0, 0, natural_size, natural_size))
|
|
im = im.resize((size, size), resample=Image.HAMMING)
|
|
|
|
response = HttpResponse(content_type="image/" + image_type.lower())
|
|
im.save(response, image_type)
|
|
return response
|