Misago/misago/users/avatars/store.py

import os
from hashlib import md5
from io import BytesIO

from PIL import Image

from django.core.files.base import ContentFile
from django.utils.crypto import get_random_string

from misago.conf import settings


def normalize_image(image):
    """strip image of animation, convert to RGBA"""
    image.seek(0)
    return image.copy().convert('RGBA')


def delete_avatar(user):
    if user.avatar_tmp:
        user.avatar_tmp.delete(save=False)

    if user.avatar_src:
        user.avatar_src.delete(save=False)

    for avatar in user.avatar_set.all():
        avatar.image.delete(save=False)
    user.avatar_set.all().delete()


def store_avatar(user, image):
    from misago.users.models import Avatar
    image = normalize_image(image)

    avatars = []
    for size in sorted(settings.MISAGO_AVATARS_SIZES, reverse=True):
        image_stream = BytesIO()

        image = image.resize((size, size), Image.ANTIALIAS)
        image.save(image_stream, "PNG")

        avatars.append(
            Avatar.objects.create(
                user=user,
                size=size,
                image=ContentFile(image_stream.getvalue(), 'avatar'),
            )
        )

    user.avatars = [{'size': a.size, 'url': a.url} for a in avatars]
    user.save(update_fields=['avatars'])

    # delete old user avatar images
    new_avatars_ids = [i.id for i in avatars]
    old_avatars_queryset = user.avatar_set.exclude(id__in=new_avatars_ids)
    for avatar in old_avatars_queryset:
        avatar.image.delete(save=False)
    old_avatars_queryset.delete()


def store_new_avatar(user, image):
    delete_avatar(user)
    store_avatar(user, image)


def store_temporary_avatar(user, image):
    image_stream = BytesIO()

    normalize_image(image)
    image.save(image_stream, "PNG")

    if user.avatar_tmp:
        user.avatar_tmp.delete(save=False)

    user.avatar_tmp = ContentFile(image_stream.getvalue(), 'avatar')
    user.save(update_fields=['avatar_tmp'])


def store_original_avatar(user):
    if user.avatar_src:
        user.avatar_src.delete(save=False)
    user.avatar_src = user.avatar_tmp
    user.avatar_tmp = None
    user.save(update_fields=['avatar_tmp', 'avatar_src'])


def upload_to(instance, filename):
    spread_path = md5(get_random_string(64).encode()).hexdigest()
    secret = get_random_string(32)
    filename_clean = '%s.png' % get_random_string(32)

    return os.path.join('avatars', spread_path[:2], spread_path[2:4], secret, filename_clean)