# -*- coding: utf-8 -*-

"""
byceps.blueprints.user_avatar.service
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

:Copyright: 2006-2016 Jochen Kupperschmidt
:License: Modified BSD, see LICENSE for details.
"""

from datetime import datetime

from ...database import db
from ...util.image import create_thumbnail, Dimensions, guess_type, ImageType, \
    read_dimensions
from ...util import upload


MAXIMUM_DIMENSIONS = Dimensions(110, 110)


def get_image_type_names():
    """Return the names of the known/accepted image types."""
    return frozenset(type.name.upper() for type in ImageType)


def update_avatar_image(user, stream):
    """Set a new avatar image for the user."""
    image_type = _determine_image_type(stream)
    user.set_avatar_image(datetime.now(), image_type)

    if _is_image_too_large(stream):
        stream = create_thumbnail(stream, image_type.name, MAXIMUM_DIMENSIONS)

    # Might raise `FileExistsError`.
    upload.store(stream, user.avatar.path)

    db.session.commit()


def _determine_image_type(stream):
    image_type = guess_type(stream)
    if image_type is None:
        abort(400, 'Only GIF, JPEG and PNG images are allowed.')

    stream.seek(0)
    return image_type


def _is_image_too_large(stream):
    actual_dimensions = read_dimensions(stream)
    stream.seek(0)
    return actual_dimensions > MAXIMUM_DIMENSIONS


def remove_avatar_image(user):
    """Remove the user's avatar image.

    The image file itself isn't removed, though.
    """
    user.remove_avatar_image()
    db.session.commit()

@@ -383,16 +383,44 @@
 upload%0A%0A
+from .models import Avatar%0A%0A
 %0AMAXIMUM
@@ -914,16 +914,82 @@
 SIONS)%0A%0A
+    avatar = Avatar(user, image_type)%0A    db.session.add(avatar)%0A%0A
     # Mi

Create avatar database record on avatar update.