在Tastypie中序列化django-mptt树

时间:2013-03-08 01:52:15

标签: django tastypie django-mptt

如何序列化django-mptt中的Tastypie树?

我想使用django-mptt的{​​{1}}。我尝试过应用不同的Tastypie钩子,但它会引发错误。

2 个答案:

答案 0 :(得分:6)

如果没有cache_tree_children方法,您可以通过简单地挂钩ToManyFieldfull=True指向children属性来序列化您的孩子:

class MenuResource(ModelResource):

    children = fields.ToManyField('self', 'children', null=True, full=True)
    parent = fields.ToOneField('self', 'parent', null=True)

    class Meta:
        queryset = Menu.objects.all()

要实现cache_tree_children函数,您可以编写自己的ToManyField子类来覆盖标准dehydrate函数。请注意,我只是非常表面地测试了这个解决方案:

def dehydrate(self, bundle):
    if not bundle.obj or not bundle.obj.pk:
    if not self.null:
        raise ApiFieldError("The model '%r' does not have a primary key and can not be used in a ToMany context." % bundle.obj)

        return []

    the_m2ms = None
    previous_obj = bundle.obj
    attr = self.attribute

    if isinstance(self.attribute, basestring):
        attrs = self.attribute.split('__')
        the_m2ms = bundle.obj

        for attr in attrs:
            previous_obj = the_m2ms
            try:
                the_m2ms = getattr(the_m2ms, attr, None)
            except ObjectDoesNotExist:
                the_m2ms = None

            if not the_m2ms:
                break

    elif callable(self.attribute):
        the_m2ms = self.attribute(bundle)

    if not the_m2ms:
        if not self.null:
            raise ApiFieldError("The model '%r' has an empty attribute '%s' and doesn't allow a null value." % (previous_obj, attr))

        return []

    self.m2m_resources = []
    m2m_dehydrated = []

    # There goes your ``cache_tree_children``
    for m2m in cache_tree_children(the_m2ms.all()):
        m2m_resource = self.get_related_resource(m2m)
        m2m_bundle = Bundle(obj=m2m, request=bundle.request)
        self.m2m_resources.append(m2m_resource)
        m2m_dehydrated.append(self.dehydrate_related(m2m_bundle, m2m_resource))

    return m2m_dehydrated

此方法的一个主要优点是您不必再关心细节/列表视图约束/差异。您甚至可以进一步参数化资源的这一方面,直到您获得某种符合您需求的默认行为。基于现场,即。我认为这很酷。

答案 1 :(得分:2)

这就是我解决它的方法:

class MenuResource(ModelResource):
    parent = fields.ForeignKey('self', 'parent', null=True)

    class Meta:
        serializer = PrettyJSONSerializer()
        queryset = Menu.objects.all().select_related('parent')
        include_resource_uri = False
        fields = ['name']

    def get_child_data(self, obj):
        data =  {
            'id': obj.id,
            'name': obj.name,
        }
        if not obj.is_leaf_node():
            data['children'] = [self.get_child_data(child) \
                                for child in obj.get_children()]
        return data

    def get_list(self, request, **kwargs):

        base_bundle = self.build_bundle(request=request)
        objects = self.obj_get_list(bundle=base_bundle, 
                                    **self.remove_api_resource_names(kwargs))
        sorted_objects = self.apply_sorting(objects, options=request.GET)

        paginator = self._meta.paginator_class(
            request.GET, sorted_objects, 
            resource_uri=self.get_resource_uri(), limit=self._meta.limit, 
            max_limit=self._meta.max_limit, 
            collection_name=self._meta.collection_name
        )
        to_be_serialized = paginator.page()

        from mptt.templatetags.mptt_tags import cache_tree_children
        objects = cache_tree_children(objects)

        bundles = []

        for obj in objects:
            data = self.get_child_data(obj)
            bundle = self.build_bundle(data=data, obj=obj, request=request)
            bundles.append(self.full_dehydrate(bundle))

        to_be_serialized[self._meta.collection_name] = bundles
        to_be_serialized = self.alter_list_data_to_serialize(request, 
                                                            to_be_serialized)
        return self.create_response(request, to_be_serialized)

如果你没有使用分页,你可以把它拿走。这就是我所做的。