my_company correction

This commit is contained in:
Diego Calvo
2021-02-10 15:02:57 +01:00
parent 3b801768e4
commit 862b65a5b2

View File

@@ -59,6 +59,6 @@ class CompanyViewSet(viewsets.ModelViewSet):
@api_view(['GET',]) @api_view(['GET',])
@permission_classes([IsAuthenticated,]) @permission_classes([IsAuthenticated,])
def my_company(request): def my_company(request):
qs = Company.objects.filter(creator=request.user) qs = Company.objects.filter(creator=request.user).first()
company_serializer = CompanySerializer(qs) company_serializer = CompanySerializer(qs)
return Response(data=company_serializer.data) return Response(data=company_serializer.data)