CODEWITHSUNDEEP
pythondjangodjango-rest-framework
Iga Quintana
Iga Quintana

Reputation: 139

Determine if Django REST Framework serializer is used in context of many=True

I have a Django REST Framework serializer that is used in several places. One of the fields is a SerializerMethodField that I only wanted to include if the serializer is used to serialize only a single object. Basically, I want to not include one of the SerializerMethodField (or change it's behavior) when I have that MySerializer(objects, many=True). Any ideas how to do this?

Upvotes: 1

Views: 1276

Answers (2)

eInyzant
eInyzant

Reputation: 309

Hi here is my solution to determine if we are in context of many=True: I override the new class method and add a "has_many" key to the context kwargs object:

    class MySerializer(serializer.ModelSerializer):

        def __new__(cls, *args, **kwargs):
            if kwargs.get('many', False) is True:
                context = kwargs.get('context', {})
                context.update({'has_many': True})
                kwargs.update({'context': context})

            return super().__new__(cls, *args, **kwargs)

        def __init__(self, *args, **kwargs):
            super().__init__(*args, **kwargs)
            if self.context.get('has_many', False):
                # Do something

Upvotes: 2

Roham
Roham

Reputation: 2110

One easy way to dynamically remove fields from your serializer is to add following codes to your serializer:

class MySerializer(serializer.ModelSerializer):
    def __init__(self, *args, **kwargs):
        remove_fields = kwargs.pop('remove_fields', None)
        super(MySerializer, self).__init__(*args, **kwargs)

        if remove_fields:
            # for multiple fields in a list
            for field_name in remove_fields:
                self.fields.pop(field_name, None)

And then when you need to remove some fields from your serializer is to use MySerializer(objects, many=True, remove_fields=['list_of_your_fields',]) which will remove list_of_your_fields' fields from your output data

Upvotes: 2

Related Questions