auto-add creator from request use in viewsets

2021-02-04 13:24:14 +00:00
parent 65c571a887
commit 9fd8c87517
4 changed files with 16 additions and 0 deletions
--- 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,])