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) + +