diff --git a/companies/views.py b/companies/views.py index 9c49dfe..3667b47 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,])