diff --git a/companies/serializers.py b/companies/serializers.py index 9ff7fa8..5e700fb 100644 --- a/companies/serializers.py +++ b/companies/serializers.py @@ -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 diff --git a/companies/views.py b/companies/views.py index 39dc40e..94556e4 100644 --- a/companies/views.py +++ b/companies/views.py @@ -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) diff --git a/requirements.txt b/requirements.txt index b13037e..f62b08f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,3 +10,4 @@ django-cors-headers==3.5.0 django-taggit-serializer==0.1.7 django-tagulous==1.1.0 Pillow==8.1.0 +drf-extra-fields==3.0.4 \ No newline at end of file