From 048f6dbb708d941939b512161ba1b47e8f6676be Mon Sep 17 00:00:00 2001 From: Diego Calvo Date: Mon, 15 Mar 2021 11:45:44 +0100 Subject: [PATCH] perform_create in MyProductsViewSet for creating with creator and company --- products/views.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/products/views.py b/products/views.py index 4e132ae..44f8d9e 100644 --- a/products/views.py +++ b/products/views.py @@ -74,6 +74,9 @@ class MyProductsViewSet(viewsets.ModelViewSet): def get_queryset(self): return self.model.objects.filter(company=self.request.user.company).order_by('-created') + def perform_create(self, serializer): + serializer.save(creator=self.request.user, company=self.request.user.company) + class AdminProductsViewSet(viewsets.ModelViewSet): """ Allows user with role 'SITE_ADMIN' to access all product instances