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