Как в ModelSerializer сделать ListField на основании CharField модели?

Мне нужно чтоб сериалайзер мне давал поле scope с массивом значений из поля user_type моей модели.

models.py
class CustomUser(AbstractBaseUser, PermissionsMixin):
    USER_TYPE_CHOICES = (
      (1, 'customer'),
      (2, 'manufacturer'),
      (3, 'seller'),
    )
    name = models.CharField(max_length=50, null=True, blank=True)
    phone = models.CharField(max_length=12, verbose_name='Телефон')
    email = models.EmailField('Адрес электронной почты', unique=True)
    is_staff = models.BooleanField(default=False)
    is_active = models.BooleanField(default=True)
    date_joined = models.DateTimeField(default=timezone.now)
    user_type = models.PositiveSmallIntegerField(choices=USER_TYPE_CHOICES)

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = []

    objects = CustomUserManager()
    # Функция возвращает строковое описание поля ролм, для фронта
    def get_user_scope(self):
        if self.user_type == 1:
            scope = 'customer'
        elif self.user_type == 2:
            scope = 'manufacturer'
        elif self.user_type == 3:
            scope = 'seller'
        else:
            scope = ''
        return scope                        

    def __str__(self):
        return self.email

serializers.py
class UserSerializer(serializers.ModelSerializer):
    scope = serializers.ListField(child=serializers.CharField(source='get_user_scope'))
    
    def create(self, validated_data):
        return get_user_model().objects.create_user(**validated_data)
    
    class Meta:
        model = get_user_model()
        fields = ('email', 'password', 'phone', 'name', 'scope')
        extra_kwargs = {
            'password': {'write_only': True, 'min_length': 8},
        }


Такой код не сработал
class UserSerializer(serializers.ModelSerializer):
    scope = serializers.ListField(child=serializers.CharField(source='get_user_scope'))


получаю ошибку
AssertionError: The `source` argument is not meaningful when applied to a `child=` field. Remove `source=` from the field declaration.
  • Вопрос задан
  • 38 просмотров
Решения вопроса 1
mgis
@mgis Автор вопроса
В общем расклад такой.
В модели добавляем метод get_scope()
class CustomUser(AbstractBaseUser, PermissionsMixin):
    USER_TYPE_CHOICES = (
      (1, 'customer'),
      (2, 'manufacturer'),
      (3, 'seller'),
    )
    name = models.CharField(max_length=50, null=True, blank=True)
    phone = models.CharField(max_length=12, verbose_name='Телефон')
    email = models.EmailField('Адрес электронной почты', unique=True)
    is_staff = models.BooleanField(default=False)
    is_active = models.BooleanField(default=True)
    date_joined = models.DateTimeField(default=timezone.now)
    user_type = models.PositiveSmallIntegerField(choices=USER_TYPE_CHOICES)

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = []
    objects = CustomUserManager()
                      
    def get_scope(self):
      arr = []
      arr.append(self.get_user_type_display()) 
      return arr 
         
    def __str__(self):
        return self.email

Ну и правим сам serializer

class UserSerializer(serializers.ModelSerializer):
    user_type = serializers.ListField(source='get_scope')
    # scope = serializers.CharField(source='get_user_type_display')
    
    def create(self, validated_data):
        return get_user_model().objects.create_user(**validated_data)
    
    class Meta:
        model = get_user_model()
        fields = ('email', 'password', 'phone', 'name', 'user_type')
        extra_kwargs = {
            'password': {'write_only': True, 'min_length': 8},
        }
Ответ написан
Пригласить эксперта
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы