Джанго-страны и TastyPie: получить название страны

Question

Я получил такую ​​модель:

from django.db import models
from django_countries.fields import CountryField

class Location(models.Model):
    company = models.CharField(max_length=64)
    country = CountryField()

    def __unicode__(self):
        return self.company

Сейчас я использую TastyPie для API. Я получил очень простую модель, подобную этой (хотя я редактировал ее ранее с помощью фильтров и полей, но безуспешно):

class LocationResource(ModelResource):
    class Meta:
        queryset = Location.objects.all()
        resource_name = 'location'

Что это возвращает:

{"company": "testcompany", "country":"DE" "resource_uri": "/api/location/1/"}

Что мне понадобится, так это название страны или, что еще лучше, что-нибудь из поля страны.

Answer 1

Вы можете добавить метод обезвоживания для country на ваш LocationResource

def dehydrate_country(self, bundle):
    return bundle.obj.country.name 

OR

если вы используете экземпляр DRF Country field лайк

 from django_countries.serializer_fields import CountryField

 country = CountryField(country_dict=True)

в вашем serializer,

Answer 2

Вы можете получить имя из словаря данных django_countries

from django_countries.data import COUNTRIES

country_name = COUNTRIES[country_code]

Answer 3

В случае DRF небольшие настройки сериализатора могут работать

from django_countries.serializer_fields import CountryField
from django_countries.fields import CountryField as ModelCountryField


class CustomCountryMixin(CountryFieldMixin):
    """Custom mixin to serialize country with name instead of country code"""

    def to_representation(self, instance):
        data = super().to_representation(instance)
        for field in instance._meta.fields:
            if field.__class__ == ModelCountryField:
                if getattr(instance, field.name).name:
                    data[field.name] = getattr(instance, field.name).name
        return data



class StudentSerializer(CustomCountryMixin, serializers.ModelSerializer):

    class Meta:
        model = Student
        fields = ('mobile_number', 'study_destination', 'country_of_residence',
                  'university', 'study_level', 'course_start_date')