husanjon
ab73d05ecc
Some checks failed
Build and Push to Docker Hub / build-test-push (push) Failing after 1m55s
22 lines
632 B
Python
22 lines
632 B
Python
from rest_framework import viewsets
|
|
from rest_framework.permissions import IsAuthenticated
|
|
|
|
from core.apps.eggs.models import User
|
|
from core.apps.eggs.serializers import (
|
|
AddSkladSerializer,
|
|
SkladListSerializer,
|
|
)
|
|
from core.http.permissions import IsRole
|
|
|
|
|
|
class AddSkladViewSet(viewsets.ModelViewSet):
|
|
permission_classes = [IsAuthenticated, IsRole(["admin"])]
|
|
queryset = User.objects.filter(role="sklad")
|
|
http_method_names = ["get", "post"]
|
|
|
|
def get_serializer_class(self):
|
|
if self.request.method == "GET":
|
|
return SkladListSerializer
|
|
else:
|
|
return AddSkladSerializer
|