Tastypie obj_create - how to use newly created object?

后端 未结 2 609
闹比i
闹比i 2020-12-18 04:58

When a new item is created using Tastypie, I want to be able to add it to a user\'s attribute which is a many-to-many field. RIght now my obj_create looks like this:

2条回答
  •  我在风中等你
    2020-12-18 05:28

    You didn't show us your resource definition, but assuming you are using tastypie.resources.ModelResource as your base class, this should work:

    def obj_create(self, bundle, request=None, **kwargs):
        bundle = super(GoalResource, self).obj_create(
            bundle, request, user=request.user)
    
        user = request.user
        user.goals.add( bundle.obj )
        user.save()
        return bundle
    

    This is because the obj_create method of ModelResource class returns a bundle which contains the saved object (bundle.obj) and you can manipulate this object in your obj_create method as shown and only then return it.

    I have also assumed that request.user contains a valid User object (i.e. authenticated), you need to make sure it does for above to work or you should add some error handling code for the case when it does not.

    Hope this helps :)

提交回复
热议问题