在DRF ModelSerializer中生成自定义响应

时间:2015-08-19 06:39:57

标签: python python-2.7 django-rest-framework django-serializer

我需要一些帮助,我在ModelSerializer中使用了DRF,问题出在这里:

我想动态生成响应,我的意思是我决定用户的输入,所以我需要将请求参数传递给ModelSerializer,然后决定我的条件是否已建立,生成字段,否则生成一些其他领域。

这样的事情:

class AlbumSerializer(serializers.ModelSerializer):

    class Meta:
        model = Album
        if request.SOME_CONDITION: 
            fields = ('id', 'name_fa', 'name_en', 'price')
        else: 
            fields = ('description', 'image_path_absolute', 'publisher')

我已经使用Google搜索并确定我可以使用context将参数传递给序列化程序,但在序列化程序中我需要通过self对象获取参数。

我该怎么做?

1 个答案:

答案 0 :(得分:0)

我找到了。

我们需要覆盖序列化器类的__init__()方法,然后使用上下文获取参数,排除不需要的字段,最后调用超类。

def __init__(self, *args, **kwargs):
    if 'context' in kwargs:
        context = kwargs.get('context')
        if 'user' in context:
            if SOME_CONDITION:
                super(AlbumSerializer, self).__init__(*args, **kwargs)
                self.fields.pop(key)
                pass
    super(AlbumSerializer, self).__init__(*args, **kwargs)