[Solved]-Best way to add additional fields to django-rest-framework ModelViewSet when create

by

16👍

✅

Update: Since v3 you need to do this:

def perform_create(self, serializer):
    serializer.save(owner=self.request.user)

The principle remains the same.

You want to make the owner field of your book serializer read-only and then set the association with the user in pre_save().

Something like: 

def pre_save(self, obj):
    obj.owner = self.request.user

See the tutorial section on “Associating Snippets with Users”.

I hope that helps.

👤Carlton Gibson

8👍

For what it’s worth this has changed in Django REST Framework 3. There is now a perform_create() that replaces the older pre_save() and post_save() hooks suggested in a previous answer. For example:

from rest_framework import viewsets

class MyViewSet(viewsets.ModelViewSet):
    queryset = MyModel.objects.all()
    serializer_class = MyModelSerializer

    def perform_create(self, serializer):
        serializer.save(owner=self.request.user)

Hope this saves someone some time.

👤Paul J

1👍

If you’re using ModelSerializer it’s as easy as implementing the restore_object() method:

class MySerializer(serializers.ModelSerializer):

    def restore_object(self, attrs, instance=None):

        create = not instance
        instance = super(MySerializer, self).restore_object(attrs, instance)

        if create:
            instance.user = self.context['request'].user

        return instance

    # ...

restore_object() is used to deserialize a dictionary of attributes into an object instance. ModelSerializer implements this method and creates/updates the instance for the model you specified in the Meta class. If the given instance is None it means the object still has to be created. In this case you just set the user field with the desired value.

More information:
http://www.django-rest-framework.org/api-guide/serializers#declaring-serializers

👤gitaarik

Leave a comment