Имя, возвращающее число после сериализации данных

У меня сервировка стола :

from django.db import models
from users.models import CustomUser

SERVICE_CHOICES = (
    ('Carpenter', 'Carpenter'),
    ('Driver', 'Driver'),
    ('Ambulanve', 'Ambulanve'),
    ('Spa', 'Spa'),
    ('Barber', 'Barber'),
    ('Cleaning', 'Cleaning'),
    ('Cook', 'Cook'),
)



class Service(models.Model):
    name  = models.ForeignKey(CustomUser, on_delete=models.CASCADE, limit_choices_to={'is_worker': True},)

    service = models.CharField(choices=SERVICE_CHOICES,max_length=100)

    def __str__(self):
        return f'{self.service} - {self.name}'

и таблицу CustomUser :

from django.db import models
from django.contrib.auth.models import AbstractUser

class CustomUser(AbstractUser):
    is_worker = models.BooleanField(default=False)
    is_customer = models.BooleanField(default=True)

Я сериализую таблицу Service ниже :

from rest_framework import serializers

from django.contrib.auth.models import User
from .models import *

class ServiceSerializer(serializers.ModelSerializer):
    class Meta:
        model = Service
        fields = '__all__'

Но когда я добавляю услугу из панели администратора, имя показывает в браузере число, а не строку. Как изменить, чтобы отображалось имя, а не число? enter image description here

В своем ServiceSerializer вы можете попробовать:

class ServiceSerializer(serializers.ModelSerializer):
    class Meta:
        model = Service
        fields = '__all__'

    def to_representation(self, instance):
        rep = super(ServiceSerializer, self).to_representation(instance)
        rep['name'] = instance.name.name    # instance.name.what_you_use_in_Custom_User_model_to_represent_name
        return rep

Однако подумайте, этого ли вы хотите, потому что это изменение может привести к большему количеству проблем, чем решает.

Это происходит потому, что name является forignkey для CustomUser, и по умолчанию возвращается PK, связанный с экземпляром CustomUser.

Вместо этого вы можете использовать SerializerMethodField.

class UserSerializer(serializers.ModelSerializer):
    name = serializers.SerializerMethodField()

    class Meta:
        model = Service
        fields = '__all__'

    def get_name(self, obj):
        return obj.name.first_name
Вернуться на верх