auto-add creator from request use in viewsets
This commit is contained in:
@@ -18,6 +18,9 @@ class CompanyViewSet(viewsets.ModelViewSet):
|
||||
serializer_class = CompanySerializer
|
||||
permission_classes = [IsAuthenticatedOrReadOnly, IsCreator]
|
||||
|
||||
def perform_create(self, serializer):
|
||||
serializer.save(creator=self.request.user)
|
||||
|
||||
|
||||
@api_view(['GET',])
|
||||
@permission_classes([IsAuthenticated,])
|
||||
|
||||
@@ -1,4 +1,5 @@
|
||||
from django.shortcuts import render
|
||||
from django.core import serializers
|
||||
|
||||
# Create your views here.
|
||||
from rest_framework import viewsets
|
||||
@@ -12,3 +13,6 @@ class HistorySyncViewSet(viewsets.ModelViewSet):
|
||||
queryset = HistorySync.objects.all()
|
||||
serializer_class = HistorySyncLogSerializer
|
||||
permission_classes = [IsStaff,]
|
||||
|
||||
def perform_create(self, serializer):
|
||||
serializer.save(creator=self.request.user)
|
||||
|
||||
@@ -35,6 +35,9 @@ class ProductViewSet(viewsets.ModelViewSet):
|
||||
serializer_class = ProductSerializer
|
||||
permission_classes = [IsAuthenticatedOrReadOnly, IsCreator]
|
||||
|
||||
def perform_create(self, serializer):
|
||||
serializer.save(creator=self.request.user)
|
||||
|
||||
|
||||
@api_view(['GET',])
|
||||
@permission_classes([IsAuthenticated,])
|
||||
|
||||
@@ -1,4 +1,5 @@
|
||||
from django.shortcuts import render
|
||||
from django.core import serializers
|
||||
|
||||
# Create your views here.
|
||||
from rest_framework import viewsets
|
||||
@@ -12,3 +13,8 @@ class StatsLogViewSet(viewsets.ModelViewSet):
|
||||
queryset = StatsLog.objects.all()
|
||||
serializer_class = StatsLogSerializer
|
||||
permission_classes = [IsStaff,]
|
||||
|
||||
def perform_create(self, serializer):
|
||||
serializer.save(creator=self.request.user)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user