perform_create in MyProductsViewSet for creating with creator and company

This commit is contained in:
Diego Calvo
2021-03-15 11:45:44 +01:00
parent 4a737d4287
commit 048f6dbb70

View File

@@ -74,6 +74,9 @@ class MyProductsViewSet(viewsets.ModelViewSet):
def get_queryset(self): def get_queryset(self):
return self.model.objects.filter(company=self.request.user.company).order_by('-created') 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): class AdminProductsViewSet(viewsets.ModelViewSet):
""" Allows user with role 'SITE_ADMIN' to access all product instances """ Allows user with role 'SITE_ADMIN' to access all product instances