<filename>src/sentry/api/bases/avatar.py<gh_stars>1-10
from __future__ import absolute_import

from rest_framework import status
from rest_framework import serializers
from rest_framework.response import Response

from sentry.api.fields import AvatarField
from sentry.api.serializers import serialize


class AvatarSerializer(serializers.Serializer):
    avatar_photo = AvatarField(required=False)
    avatar_type = serializers.ChoiceField(
        choices=(
            ('upload', 'upload'), ('gravatar', 'gravatar'), ('letter_avatar', 'letter_avatar'),
        )
    )

    def validate(self, attrs):
        attrs = super(AvatarSerializer, self).validate(attrs)
        if attrs.get('avatar_type') == 'upload':
            model_type = self.context['type']
            has_existing_file = model_type.objects.filter(
                file__isnull=False,
                **self.context['kwargs']
            ).exists()
            if not has_existing_file and not attrs.get('avatar_photo'):
                raise serializers.ValidationError(
                    {
                        'avatar_type': 'Cannot set avatar_type to upload without avatar_photo',
                    }
                )
        return attrs


class AvatarMixin(object):
    object_type = None
    model = None

    def get(self, request, **kwargs):
        obj = kwargs[self.object_type]
        return Response(serialize(obj, request.user))

    def get_serializer_context(self, obj, **kwargs):
        return {
            'type': self.model,
            'kwargs': {self.object_type: obj},
        }

    def get_avatar_filename(self, obj):
        return u'{}.png'.format(obj.id)

    def put(self, request, **kwargs):
        obj = kwargs[self.object_type]
        serializer = AvatarSerializer(
            data=request.data,
            context=self.get_serializer_context(obj),
        )
        if not serializer.is_valid():
            return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

        result = serializer.validated_data

        self.model.save_avatar(
            relation={self.object_type: obj},
            type=result['avatar_type'],
            avatar=result.get('avatar_photo'),
            filename=self.get_avatar_filename(obj),
        )

        return Response(serialize(obj, request.user))

<filename>src/sentry/api/bases/avatar.py<gh_stars>1-10
from __future__ import absolute_import

from rest_framework import status
from rest_framework import serializers
from rest_framework.response import Response

from sentry.api.fields import AvatarField
from sentry.api.serializers import serialize


class AvatarSerializer(serializers.Serializer):
    avatar_photo = AvatarField(required=False)
    avatar_type = serializers.ChoiceField(
        choices=(
            ('upload', 'upload'), ('gravatar', 'gravatar'), ('letter_avatar', 'letter_avatar'),
        )
    )

    def validate(self, attrs):
        attrs = super(AvatarSerializer, self).validate(attrs)
        if attrs.get('avatar_type') == 'upload':
            model_type = self.context['type']
            has_existing_file = model_type.objects.filter(
                file__isnull=False,
                **self.context['kwargs']
            ).exists()
            if not has_existing_file and not attrs.get('avatar_photo'):
                raise serializers.ValidationError(
                    {
                        'avatar_type': 'Cannot set avatar_type to upload without avatar_photo',
                    }
                )
        return attrs


class AvatarMixin(object):
    object_type = None
    model = None

    def get(self, request, **kwargs):
        obj = kwargs[self.object_type]
        return Response(serialize(obj, request.user))

    def get_serializer_context(self, obj, **kwargs):
        return {
            'type': self.model,
            'kwargs': {self.object_type: obj},
        }

    def get_avatar_filename(self, obj):
        return u'{}.png'.format(obj.id)

    def put(self, request, **kwargs):
        obj = kwargs[self.object_type]
        serializer = AvatarSerializer(
            data=request.data,
            context=self.get_serializer_context(obj),
        )
        if not serializer.is_valid():
            return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

        result = serializer.validated_data

        self.model.save_avatar(
            relation={self.object_type: obj},
            type=result['avatar_type'],
            avatar=result.get('avatar_photo'),
            filename=self.get_avatar_filename(obj),
        )

        return Response(serialize(obj, request.user))
