Tastypie,为多对多关系添加元素

时间:2012-04-11 18:24:01

标签: django rest tastypie

我正在构建一个django tastypie api,我在ManyToMany关系中添加元素时遇到问题

实施例, models.py

class Picture(models.db):
    """ A picture of people"""
    people = models.ManyToManyField(Person, related_name='pictures',
        help_text="The people in this picture",
    )

class Person(models.db):
    """ A model to represet a person """
    name = models.CharField(max_length=200,
        help_text="The name of this person",
    )

资源:

class PictureResource(ModelResource):
    """ API Resource for the Picture model """
    people = fields.ToManyField(PersonResource, 'people', null=True,
        related_name="pictures", help_text="The people in this picture",
    )
class PersonResource(ModelResource):
    """ API Resource for the Person model """
    pictures = fields.ToManyField(PictureResource, 'pictures', null=True,
        related_name="people", help_text="The pictures were this person appears",
    )

我的问题是我希望在我的图片资源中有add_person个终点。 如果我使用PUT,那么我需要指定图片中的所有数据 如果我使用PATCH,我仍然需要指定图片中的所有人。 当然,我可以简单地生成/api/picture/:id/add_people URL,在那里我可以处理我的问题。问题在于它感觉不干净。

另一种解决方案是生成/api/picture/:id/people终点,我可以GETPOSTPUT,这是一个新资源,但我不知道我不知道如何实现这一点,在这个资源下创建新人似乎很奇怪。

有什么想法吗?

1 个答案:

答案 0 :(得分:4)

我通过覆盖API资源的save_m2m函数实现了这一点。以下是使用模型的示例。

def save_m2m(self, bundle):
    for field_name, field_object in self.fields.items():
        if not getattr(field_object, 'is_m2m', False):
            continue

        if not field_object.attribute:
            continue

        if field_object.readonly:
            continue

        # Get the manager.
        related_mngr = getattr(bundle.obj, field_object.attribute)
            # This is code commented out from the original function
            # that would clear out the existing related "Person" objects
            #if hasattr(related_mngr, 'clear'):
            # Clear it out, just to be safe.
            #related_mngr.clear()

        related_objs = []

        for related_bundle in bundle.data[field_name]:
            # See if this person already exists in the database
            try:
                person = Person.objects.get(name=related_bundle.obj.name)
            # If it doesn't exist, then save and use the object TastyPie
            # has already prepared for creation
            except Person.DoesNotExist:
                person = related_bundle.obj
                person.save()

            related_objs.append(person)

        related_mngr.add(*related_objs)