auto-add creator from request use in viewsets

This commit is contained in:
Sam
2021-02-04 13:24:14 +00:00
parent 65c571a887
commit 9fd8c87517
4 changed files with 16 additions and 0 deletions

View File

@@ -18,6 +18,9 @@ class CompanyViewSet(viewsets.ModelViewSet):
serializer_class = CompanySerializer serializer_class = CompanySerializer
permission_classes = [IsAuthenticatedOrReadOnly, IsCreator] permission_classes = [IsAuthenticatedOrReadOnly, IsCreator]
def perform_create(self, serializer):
serializer.save(creator=self.request.user)
@api_view(['GET',]) @api_view(['GET',])
@permission_classes([IsAuthenticated,]) @permission_classes([IsAuthenticated,])

View File

@@ -1,4 +1,5 @@
from django.shortcuts import render from django.shortcuts import render
from django.core import serializers
# Create your views here. # Create your views here.
from rest_framework import viewsets from rest_framework import viewsets
@@ -12,3 +13,6 @@ class HistorySyncViewSet(viewsets.ModelViewSet):
queryset = HistorySync.objects.all() queryset = HistorySync.objects.all()
serializer_class = HistorySyncLogSerializer serializer_class = HistorySyncLogSerializer
permission_classes = [IsStaff,] permission_classes = [IsStaff,]
def perform_create(self, serializer):
serializer.save(creator=self.request.user)

View File

@@ -35,6 +35,9 @@ class ProductViewSet(viewsets.ModelViewSet):
serializer_class = ProductSerializer serializer_class = ProductSerializer
permission_classes = [IsAuthenticatedOrReadOnly, IsCreator] permission_classes = [IsAuthenticatedOrReadOnly, IsCreator]
def perform_create(self, serializer):
serializer.save(creator=self.request.user)
@api_view(['GET',]) @api_view(['GET',])
@permission_classes([IsAuthenticated,]) @permission_classes([IsAuthenticated,])

View File

@@ -1,4 +1,5 @@
from django.shortcuts import render from django.shortcuts import render
from django.core import serializers
# Create your views here. # Create your views here.
from rest_framework import viewsets from rest_framework import viewsets
@@ -12,3 +13,8 @@ class StatsLogViewSet(viewsets.ModelViewSet):
queryset = StatsLog.objects.all() queryset = StatsLog.objects.all()
serializer_class = StatsLogSerializer serializer_class = StatsLogSerializer
permission_classes = [IsStaff,] permission_classes = [IsStaff,]
def perform_create(self, serializer):
serializer.save(creator=self.request.user)