from django.conf import settings
from django.db.transaction import on_commit
from rest_framework import generics, status
from rest_framework.permissions import AllowAny
from rest_framework.response import Response

from utils.methods import get_permission_classes
from utils.permissions import IsContentPageManager, IsCountryAdmin, IsEstablishmentManager, \
    IsProducerFoodInspector, IsEstablishmentAdministrator
from . import tasks, models, serializers

class ImageBaseView(generics.GenericAPIView):
    """Base Image view."""
    model = models.Image
    queryset = models.Image.objects.all()
    serializer_class = serializers.ImageSerializer
    permission_classes = get_permission_classes(
        IsContentPageManager
    )


class ImageListCreateView(ImageBaseView, generics.ListCreateAPIView):
    """List/Create Image view."""


class MediaForEstablishmentView(ImageBaseView, generics.ListCreateAPIView):
    """View for creating and retrieving certain establishment media."""
    pagination_class = None
    # permission_classes = (IsCountryAdmin, IsEstablishmentAdministrator, IsEstablishmentManager, IsProducerFoodInspector)
    permission_classes = (AllowAny, )
    serializer_class = serializers.EstablishmentGallerySerializer

    def get_queryset(self):
        return super().get_queryset().filter(establishment__pk=self.kwargs['establishment_id'])\
            .order_by('-order').prefetch_related('created_by', 'establishment_gallery')


class MediaUpdateView(ImageBaseView, generics.UpdateAPIView, generics.DestroyAPIView):
    """View for updating media data"""
    serializer_class = serializers.EstablishmentGallerySerializer
    # permission_classes = (IsCountryAdmin, IsEstablishmentAdministrator, IsEstablishmentManager, IsProducerFoodInspector)
    permission_classes = (AllowAny, )

    def delete(self, request, *args, **kwargs):
        """Override destroy view"""
        instance = self.get_object()
        if settings.USE_CELERY:
            on_commit(lambda: tasks.delete_image.delay(image_id=instance.id))
        else:
            on_commit(lambda: tasks.delete_image(image_id=instance.id))
        return Response(status=status.HTTP_204_NO_CONTENT)


class ImageRetrieveDestroyView(ImageBaseView, generics.RetrieveDestroyAPIView):
    """Destroy view for model Image"""

    def delete(self, request, *args, **kwargs):
        """Override destroy view"""
        instance = self.get_object()
        if settings.USE_CELERY:
            on_commit(lambda: tasks.delete_image.delay(image_id=instance.id))
        else:
            on_commit(lambda: tasks.delete_image(image_id=instance.id))
        return Response(status=status.HTTP_204_NO_CONTENT)


class CropImageCreateView(ImageBaseView, generics.CreateAPIView):
    """Create crop image."""
    serializer_class = serializers.CropImageSerializer