added drf-extra-fields to handle PointField serialization

This commit is contained in:
Sam
2021-02-09 11:41:26 +00:00
parent 4460f1121a
commit d0deec9417
3 changed files with 6 additions and 3 deletions

View File

@@ -1,11 +1,14 @@
from rest_framework import serializers
from companies.models import Company
from drf_extra_fields.geo_fields import PointField
from utils.tag_serializers import TagListSerializerField, TaggitSerializer
class CompanySerializer(TaggitSerializer, serializers.ModelSerializer):
tags = TagListSerializerField(required=False)
geo = PointField(required=False)
class Meta:
model = Company

View File

@@ -61,6 +61,5 @@ class CompanyViewSet(viewsets.ModelViewSet):
@permission_classes([IsAuthenticated,])
def my_company(request):
qs = Company.objects.filter(creator=request.user)
data = serializers.serialize('json', qs)
return Response(data=data)
company_serializer = CompanySerializer(qs)
return Response(data=company_serializer.data)