"""Notification app common views."""
from django.shortcuts import get_object_or_404
from rest_framework import generics, permissions
from rest_framework.response import Response

from notification import models
from notification.serializers import common as serializers


class CreateSubscribeView(generics.CreateAPIView):
    """Create subscribe View."""

    queryset = models.Subscriber.objects.all()
    permission_classes = (permissions.AllowAny,)
    serializer_class = serializers.CreateSubscribeSerializer


class SubscribeInfoView(generics.RetrieveAPIView):
    """Subscribe info view."""

    lookup_field = 'update_code'
    lookup_url_kwarg = 'code'
    permission_classes = (permissions.AllowAny,)
    queryset = models.Subscriber.objects.all()
    serializer_class = serializers.SubscribeSerializer


class SubscribeInfoAuthUserView(generics.RetrieveAPIView):
    """Subscribe info auth user view."""

    permission_classes = (permissions.IsAuthenticated,)
    serializer_class = serializers.SubscribeSerializer
    lookup_field = None

    def get_object(self):
        return get_object_or_404(models.Subscriber, user=self.request.user)


class UnsubscribeView(generics.UpdateAPIView):
    """Unsubscribe view."""

    lookup_field = 'update_code'
    lookup_url_kwarg = 'code'
    permission_classes = (permissions.AllowAny,)
    queryset = models.Subscriber.objects.all()
    serializer_class = serializers.SubscribeSerializer

    def patch(self, request, *args, **kw):
        obj = self.get_object()
        obj.unsubscribe(request.query_params)
        serializer = self.get_serializer(instance=obj)
        return Response(data=serializer.data)


class UnsubscribeAuthUserView(generics.GenericAPIView):
    """Unsubscribe auth user view."""

    permission_classes = (permissions.IsAuthenticated,)
    queryset = models.Subscriber.objects.all()
    serializer_class = serializers.SubscribeSerializer

    def patch(self, request, *args, **kw):
        user = request.user
        obj = get_object_or_404(models.Subscriber, user=user)
        obj.unsubscribe(request.query_params)
        serializer = self.get_serializer(instance=obj)
        return Response(data=serializer.data)


class SubscriptionTypesView(generics.ListAPIView):
    pagination_class = None
    permission_classes = (permissions.AllowAny,)
    queryset = models.SubscriptionType.objects.all()
    serializer_class = serializers.SubscriptionTypeSerializer