Django Rest Framework - return different serializer in same viewset

Question

I have two models: where

class Hunter(Model):
    name = CharField()

searches for

class Target(Model):
    name = CharField()
    user = ForeignKey(User) 
    approved_hunters = ManyToManyField(Hunter)

my serializers are

class AnonymousTargetSerializer(ModelSerializer):
    class Meta:
        name = 'target'
        model = Target
        fields = ['id',]

and also

class TargetSerializer(ModelSerializer):
    class Meta:
        name = 'target'
        model = Target
        fields = ['id',]

Im my viewsets I would like to:

hunter = Hunter.objects.get(user=self.request.user)
if hunter in Target.approved_hunters.all()
    return TagetSerializer
else:
    return AnonymousTargetSerializer

However I only seem to get this done for the viewset, but cannot filter on an individual object level.

neverwalkaloner · Accepted Answer · 2018-09-16 09:11:20Z

3

If you need to modify representation of object based on some attribute you can override serializer's to_representation method:

class TargetSerializer(ModelSerializer):
    class Meta:
        name = 'target'
        model = Target
        fields = ['id',]

    def to_representation(self, obj):
        hunter = Hunter.objects.get(user=self.context['request'].user)
        if hunter not in Target.approved_hunters.all():
            return AnonymousTargetSerializer(obj).data
        return super().to_representation(obj)

In viewset you dont need any customization, just use TargetSerializer:

serializer_class = TargetSerializer

answered Sep 16, 2018 at 9:11

neverwalkaloner

47.6k7 gold badges101 silver badges103 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Django Rest Framework - return different serializer in same viewset

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related