[FIXED] DRF - Filter queryset in PrimaryKeyRelatedField

Issue

I've the following models:

class ModelX(models.Model):

    STATUS_CHOICES = (
        (0, 'ABC'),
        (1, 'DEF'),
    )
    status = models.IntegerField(choices=STATUS_CHOICES)
    user = models.ForeignKey(Users)


class Users(models.Model):
    phone_number = models.Charfield()

and the serializer for ModelX is :

class ModelXSerializer(serializers.ModelSerializer):
    phone_number = serializers.PrimaryKeyRelatedField(
        source='user', queryset=Users.objects.get(phone_number=phone_number))

    class Meta:
        model = ModelX
        fields = ('phone_number',)

In the request for creating the ModelX record, I get phone_number instead of the user_id. Now, I've to fire a filter query in order get the user instance. How do I do that, ie Users.objects.get(phone_number=phone_number)

Also, when creating a record, the status field will always be 0. The client wont post the status parameter in the body. Its the internal logic. Is there a way in serializers that can set the status field to 0 automatically. Please dont recommend to set this field as default=0 in models. There's some logic behind this. This is just a shorter version of the problem statement.

Solution

You can add it in validate function:

class ModelXSerializer(serializers.ModelSerializer):

    def validate(self, attrs):
        attrs = super(ModelXSerializer, self).validate(attrs)
        attrs['status'] = 0
        return attrs

Answered By - Huu-Danh Pham