forked from 00dani/lemoncurry
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
|
from django.http import HttpResponse, HttpResponseRedirect
|
||
|
|
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)
|
||
|
|
|
||
|
|
|
||
|
|
def libravatar(request, hash):
|
||
|
|
g = request.GET
|
||
|
|
size = g.get('s', g.get('size', 80))
|
||
|
|
try:
|
||
|
|
size = int(size)
|
||
|
|
except ValueError:
|
||
|
|
return utils.bad_req('size parameter must be an integer')
|
||
|
|
if not 1 <= size <= 128:
|
||
|
|
return utils.bad_req('size parameter must be between 1 and 128')
|
||
|
|
|
||
|
|
if len(hash) == 32:
|
||
|
|
where = {'email_md5': hash}
|
||
|
|
elif len(hash) == 64:
|
||
|
|
where = {'email_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)
|
||
|
|
im_resized = im.resize((size, size))
|
||
|
|
|
||
|
|
response = HttpResponse(content_type='image/'+im.format.lower())
|
||
|
|
im_resized.save(response, im.format)
|
||
|
|
return response
|