Django使用django-treebeard时获取分类后代关联全部商品的问题

错误原因 #

原有代码存在两处逻辑错误：

1. 父查询直接调用Category.objects没有过滤条件，会返回全部分类，不是需要的当前分类后代集合
2. Prefetch的queryset参数需要传入关联模型（此处为Product）的查询集，你传入的是Category类型的后代查询集，模型不匹配触发字段匹配报错。同时查询目标是关联商品，但原有写法返回的结果是分类集合，和预期不符。

正确实现 #

根据使用场景可选择以下两种方案：

方案1：直接获取所有关联商品列表（无需按分类分组） #

def get_context_data(self, **kwargs):
    context = super().get_context_data(**kwargs)
    # 如需包含当前分类自身的商品，可添加参数include_self=True
    target_categories = self.object.get_descendants()
    # 反向查询所有关联商品
    context["products_in_category"] = Product.objects.filter(category__in=target_categories)
    return context

模板中可直接遍历products_in_category展示所有商品。

• 若分类量级较大，可利用django-treebeard的path字段特征优化查询，避免in查询的性能损耗。

方案2：按分类分组展示，每个分类携带对应商品 #

def get_context_data(self, **kwargs):
    context = super().get_context_data(**kwargs)
    # 预加载每个分类下的商品，避免N+1查询，如需包含当前分类添加include_self=True
    context["target_categories"] = self.object.get_descendants().prefetch_related("products")
    return context

模板中嵌套遍历即可：

{% for category in target_categories %}
    <h3>{{ category.title }}</h3>
    <ul>
    {% for product in category.products.all %}
        <li>{{ product.name }}</li>
    {% endfor %}
    </ul>
{% endfor %}

内容的提问来源于stack exchange，提问作者Jonas Fagerlund