from django_filters.rest_framework import DjangoFilterBackend
from rest_framework import generics

from partner.models import Partner
from partner.serializers import back as serializers
from utils.methods import get_permission_classes
from utils.permissions import IsEstablishmentManager, IsEstablishmentAdministrator


class PartnerLstView(generics.ListCreateAPIView):
    """Partner list create view."""
    queryset = Partner.objects.all()
    serializer_class = serializers.BackPartnerSerializer
    pagination_class = None
    filter_backends = (DjangoFilterBackend,)
    permission_classes = get_permission_classes(
        IsEstablishmentManager,
        IsEstablishmentAdministrator
    )
    filterset_fields = (
        'establishment',
        'type',
    )


class PartnerRUDView(generics.RetrieveUpdateDestroyAPIView):
    """Partner RUD view."""
    queryset = Partner.objects.all()
    serializer_class = serializers.BackPartnerSerializer
    lookup_field = 'id'
    permission_classes = get_permission_classes(
        IsEstablishmentManager,
        IsEstablishmentAdministrator
    )