web-dev-qa-db-fra.com

Transmettez des arguments supplémentaires à la classe de sérialiseur dans Django Rest Framework

Je veux passer quelques arguments à la classe DRF Serializer de Viewset, donc pour cela j'ai essayé:

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')

Vues 

class OneZeroViewSet(viewsets.ModelViewSet):

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

   queryset = OneZero.objects.all()

Fondamentalement, je souhaite transmettre une valeur basée sur la chaîne de requête des vues à la classe Serializer, puis celles-ci seront affectées à des champs.

Ces champs ne sont pas inclus dans les champs créés de manière dynamique dans Model.

Même cas dans cette question stackoverflow , mais je ne comprends pas la réponse.

Quelqu'un peut-il m'aider dans ce cas ou me suggérer de meilleures options?.

pythondjangorestdjango-rest-framework
55
Shoaib Ijaz

C'est très facile avec "context" arg pour le constructeur "ModelSerializer".

Par exemple:

en vue:

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

en sérialiseurs:

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
...

vous pouvez donc utiliser "self.context" pour obtenir des paramètres supplémentaires.

référence

77
redcyb

Pour remplir la réponse de redcyb - pensez à utiliser dans votre vue le get_serializer_context méthode de GenericAPIView, comme ceci:

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

Vous avez besoin de la méthode View override get_serializer_context Comme ceci:

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

et n'importe où dans le serializer, vous pouvez l'obtenir:

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

Un ancien code que j'ai écrit pourrait être utile pour filtrer le sérialiseur imbriqué:

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')

Comment remplacer 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):
       .....

J'espère que cela aide les gens à la recherche de cela.

7
yeaske