only validated companies showing in views

This commit is contained in:
Sam
2021-03-05 11:44:26 +00:00
parent a92b44c0b2
commit 375db09bb3
3 changed files with 37 additions and 11 deletions

View File

@@ -24,7 +24,7 @@ from utils import woocommerce
class CompanyViewSet(viewsets.ModelViewSet):
queryset = Company.objects.all().order_by('-created')
queryset = Company.objects.filter(is_validated=True).order_by('-created')
serializer_class = CompanySerializer
permission_classes = [IsAuthenticatedOrReadOnly, IsCreator]
filterset_class = CompanyTagFilter
@@ -187,6 +187,6 @@ def random_company_sample(request):
except:
return Response({"error": "Wrong value for size, must be a number"}, status=status.HTTP_406_NOT_ACCEPTABLE)
queryset = Company.objects.all().order_by('?')[:size]
queryset = Company.objects.filter(is_validated=True).order_by('?')[:size]
serializer = CompanySerializer(queryset, many=True)
return Response(serializer.data)