added custom permission IsCreator

This commit is contained in:
Sam
2021-01-22 11:47:09 +00:00
parent 1df9aac998
commit 6c3ff8f1fa
2 changed files with 26 additions and 0 deletions

View File

@@ -2,10 +2,15 @@ from django.shortcuts import render
# Create your views here.
from rest_framework import viewsets
from rest_framework.permissions import IsAuthenticatedOrReadOnly
from companies.models import Company
from companies.serializers import CompanySerializer
from back_latienda.permissions import IsCreator
class CompanyViewSet(viewsets.ModelViewSet):
queryset = Company.objects.all()
serializer_class = CompanySerializer
permission_classes = [IsAuthenticatedOrReadOnly, IsCreator]