35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
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
|
|
)
|