How to override default creation method in django-rest-framework

Question

I have a model that uses a different create method in the manager. How to override this method so that the post method in ListCreateAPIView uses the method that I wrote instead of the default method. Here is the method.

class WeddingInviteManager(models.Manager): def create(self, to_user, from_user, wedding): wedding_invitation = self.create(from_user=from_user,to_user=to_user, wedding=wedding) notification.send([self.to_user], 'wedding_invite',{'invitation':wedding_invitation}) return wedding_invitation 

Answer 1

I believe the reason for this is the notification system.

I would recommend doing something like this:

 class MyModel(models.Model): ... def save(self, silent=False, *args, **kwargs): # Send notification if this is a new instance that has not been saved # before: if not silent and not self.pk: notification.send([self.to_user], 'wedding_invite', {'invitation': self}) return super(MyModel, self).save(*args, **kwargs) 

But if you need, this is (theoretically) how you do it (code not verified):

 from rest_framework import serializers, viewsets class MyModelSerializer(serializers.ModelSerializer): def save_object(self, obj, **kwargs): """ Save the deserialized object. """ if getattr(obj, '_nested_forward_relations', None): # Nested relationships need to be saved before we can save the # parent instance. for field_name, sub_object in obj._nested_forward_relations.items(): if sub_object: self.save_object(sub_object) setattr(obj, field_name, sub_object) ##### EDITED CODE ##### if obj.pk: obj.save(**kwargs) else: # Creating a new object. This is silly. obj = MyModel.objects.create(obj.to_user, obj.from_user, obj.wedding) ##### /EDITED CODE ##### if getattr(obj, '_m2m_data', None): for accessor_name, object_list in obj._m2m_data.items(): setattr(obj, accessor_name, object_list) del(obj._m2m_data) if getattr(obj, '_related_data', None): related_fields = dict([ (field.get_accessor_name(), field) for field, model in obj._meta.get_all_related_objects_with_model() ]) for accessor_name, related in obj._related_data.items(): if isinstance(related, RelationsList): # Nested reverse fk relationship for related_item in related: fk_field = related_fields[accessor_name].field.name setattr(related_item, fk_field, obj) self.save_object(related_item) # Delete any removed objects if related._deleted: [self.delete_object(item) for item in related._deleted] elif isinstance(related, models.Model): # Nested reverse one-one relationship fk_field = obj._meta.get_field_by_name(accessor_name)[0].field.name setattr(related, fk_field, obj) self.save_object(related) else: # Reverse FK or reverse one-one setattr(obj, accessor_name, related) del(obj._related_data) class MyModelViewSet(viewsets.ModelViewSet): serializer_class = MyModelSerializer queryset = MyModel.objects.all()