web-dev-qa-db-ja.com

Django Rest Frameworkでシリアライザークラスに追加の引数を渡します

私はViewsetからDRF Serializerクラスにいくつかの引数を渡したいので、これを試しました:

class OneZeroSerializer(rest_serializer.ModelSerializer):

    def __init__(self, *args, **kwargs):
        print args # show values that passed

    location = rest_serializer.SerializerMethodField('get_alternate_name')

    def get_alternate_name(self, obj):
        return ''


    class Meta:
        model = OneZero

        fields = ('id', 'location')

ビュー

class OneZeroViewSet(viewsets.ModelViewSet):

   serializer_class = OneZeroSerializer(realpart=1)
   #serializer_class = OneZeroSerializer

   queryset = OneZero.objects.all()

基本的に、クエリ文字列に基づいた値をビューからシリアライザークラスに渡したい場合、これらはフィールドに割り当てられます。

これらのフィールドは、実際には動的に作成されたフィールドにはモデルに含まれていません。

この質問で同じケース stackoverflow が、答えを理解できません。

この場合、誰でも私を助けたり、より良いオプションを提案したりできますか?.

55
Shoaib Ijaz

「ModelSerializer」コンストラクターの「context」引数を使用すると、非常に簡単です。

例えば:

ビューで:

my_objects = MyModelSerializer(
    input_collection, 
    many=True, 
    context={'user_id': request.user.id}
).data

シリアライザーで:

class MyModelSerializer(serializers.ModelSerializer):
...

    is_my_object = serializers.SerializerMethodField('_is_my_find')
...

    def _is_my_find(self, obj):
        user_id = self.context.get("user_id")
        if user_id:
            return user_id in obj.my_objects.values_list("user_id", flat=True)
        return False
...

そのため、追加のパラメータを取得するために「self.context」を使用できます。

参照

77
redcyb

Redcybの答えを実現するには、ビューでget_serializer_context次のようなGenericAPIViewのメソッド:

def get_serializer_context(self):
    return {'user': self.request.user.email}
20
andilabs

View override get_serializer_contextこのようなメソッド:

def get_serializer_context(self):
    return {"customer_id": self.kwargs['customer_id']}

serializerのどこでも取得できます:

customer_id = self.context["customer_id"]
16
M.Void

私が書いた古いコードは、ネストされたシリアライザーをフィルターするのに役立つかもしれません:

class MySerializer(serializers.ModelSerializer):

    field3  = serializers.SerializerMethodField('get_filtered_data')

    def get_filtered_data(self, obj):
        param_value = self.context['request'].QUERY_PARAMS.get('Param_name', None)
        if param_value is not None:
            try:
                data = Other_model.objects.get(pk_field=obj, filter_field=param_value)
            except:
                return None
            serializer = OtherSerializer(data)
            return serializer.data
        else:
            print "Error stuff"

    class Meta:
        model = Model_name
        fields = ('filed1', 'field2', 'field3')

Get_serializer_classをオーバーライドする方法:

class ViewName(generics.ListAPIView):

    def get_serializer_class(self):
        param_value = self.context['request'].QUERY_PARAMS.get('Param_name', None)
        if param_value is not None:
            return Serializer1
        else:
            return Serializer2

    def get_queryset(self):
       .....

これがこれを探している人々に役立つことを願っています。

7
yeaske