From 9fd8c875173e4454cbc89076f4b0dc813c84071a Mon Sep 17 00:00:00 2001 From: Sam Date: Thu, 4 Feb 2021 13:24:14 +0000 Subject: [PATCH] auto-add creator from request use in viewsets --- companies/views.py | 3 +++ history/views.py | 4 ++++ products/views.py | 3 +++ stats/views.py | 6 ++++++ 4 files changed, 16 insertions(+) diff --git a/companies/views.py b/companies/views.py index 9c49dfe..3f99b55 100644 --- a/companies/views.py +++ b/companies/views.py @@ -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,]) diff --git a/history/views.py b/history/views.py index f86367c..e6a0b92 100644 --- a/history/views.py +++ b/history/views.py @@ -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) diff --git a/products/views.py b/products/views.py index b2f660a..7cfbdd2 100644 --- a/products/views.py +++ b/products/views.py @@ -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,]) diff --git a/stats/views.py b/stats/views.py index 3560e55..6b51bac 100644 --- a/stats/views.py +++ b/stats/views.py @@ -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) + +