Tastypie obj_create - how to use newly created object?

后端 未结 2 524
闹比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 :)

    0 讨论(0)
  • 2020-12-18 05:43

    I don't have enough reputation to comment yet so I figured I would put a second answer. The answer above is correct I just wanted to add that request no longer exists in the obj_create call. You can access the current request via bundle.request:

    http://django-tastypie.readthedocs.org/en/latest/resources.html#accessing-the-current-request

    Thanks for the answer above, it helped me as well!

    0 讨论(0)
提交回复
热议问题