CODEWITHSUNDEEP
djangoserialization
wasp256
wasp256

Reputation: 6242

django serializer dynamically set attributes

Is it possible to dynamically set the attributes for a Serializer dynamically?

CONTEXT_CLASSES = dict()
contexts = ['EmailContext', 'SmsMessageContext',...]

for ctx in contexts:
    Ctx_class = getattr(sys.modules[__name__], f'{ctx}Serializer')
    ctx_class_instance = Ctx_class(many=False, required=False, read_only=True)
    attr_name = ctx.split('Context')[0].lower() + '_ctx'
    CONTEXT_CLASSES[attr_name] = ctx_class_instance


class ContextContainerSerializer(serializers.ModelSerializer):

    for attr_name, instance in CONTEXT_CLASSES.items():
        # set attributes somehow

    class Meta:
         ...
         fields = tuple([value[0] for value in CONTEXT_CLASSES.values()])

Putting it in the __init__ will not work since the class Meta is executed prior to it and setting the fields variable which contains the attribute names already and will raise a django.core.exceptions.ImproperlyConfigured.

Upvotes: 0

Views: 1041

Answers (1)

Tobey
Tobey

Reputation: 1440

You can set the attributes on initialisation

class ContextContainerSerializer(serializers.ModelSerializer):

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        # should be able to access "self.Meta"
        for attr_name, instance in CONTEXT_CLASSES.items():
            setattr(self.Meta, attr_name, instance)

Or for more dynamic control/patterns, use metaclasses

from rest_framework.serializers import SerializerMetaclass


class ExtraContextMetaclass(SerializerMetaclass):
     def __new__(cls, name, bases, dct):
         """

         :param name:  name of class to be created
         :param bases: sub classes
         :param dct: other attributes
         :return:
         """

         new_class = super().__new__(cls, name, bases, dct)
         new_class.Meta = cls.dynamic_meta(new_class.Meta)
         return new_class

     @classmethod
     def dynamic_meta(cls, old_meta):
         # create a class called "Meta"
         new_meta = type('Meta', (), dict(fields=None))

         # copy over required attributes from original Meta
         new_meta.model = old_meta.model
         new_meta.fields = list(old_meta.fields[:])

         # set new attributes dynamically 
         for attr_name, instance in CONTEXT_CLASSES.items():
             setattr(new_meta, attr_name, instance)

         return new_meta


class ContextContainerSerializer(serializers.ModelSerializer, 
                                 metaclass=ExtraContextMetaclass):

    class Meta:
        model = Model
        fields = ['some_field']

Upvotes: 1

Related Questions