changes for my endpoints

2021-03-12 10:22:23 +00:00
parent 6862d62bf5
commit 37f222e6c9
2 changed files with 13 additions and 1 deletions
--- a/companies/views.py
+++ b/companies/views.py
@@ -171,7 +171,7 @@ class MyCompanyViewSet(viewsets.ModelViewSet):
    permission_classes = [IsAuthenticated]

    def get_queryset(self):
-        return self.model.objects.filter(creator=self.request.user)
+        return self.model.objects.filter(company=self.request.user.company)

    def perform_create(self, serializer):
        serializer.save(creator=self.request.user)