added AdminCompanyViewSet for users with role SITE_ADMIN

companies/views.py

@@ -18,13 +18,11 @@ from stats.models import StatsLog
 from companies.models import Company
 from companies.serializers import CompanySerializer
 from utils.tag_filters import CompanyTagFilter
-from back_latienda.permissions import IsCreator
+from back_latienda.permissions import IsCreator, IsSiteAdmin
 
 from utils import woocommerce
 
 
-
-
 class CompanyViewSet(viewsets.ModelViewSet):
     queryset = Company.objects.filter(is_validated=True).order_by('-created')
     serializer_class = CompanySerializer
@@ -169,6 +167,17 @@ class MyCompanyViewSet(viewsets.ModelViewSet):
         serializer.save(creator=self.request.user)
 
 
+class AdminCompanyViewSet(viewsets.ModelViewSet):
+    """ Allows user with role 'SITE_ADMIN' to access all company instances
+    """
+    queryset = Company.objects.all()
+    serializer_class = CompanySerializer
+    permission_classes = [IsSiteAdmin]
+
+    def perform_create(self, serializer):
+        serializer.save(creator=self.request.user)
+
+
 @api_view(['GET',])
 @permission_classes([IsAuthenticatedOrReadOnly,])
 def random_company_sample(request):