在Django Admin中实现按多级父类Category搜索Product的方案

你需要在Django Admin中实现按产品关联分类的父类、祖父类等多级祖先分类进行搜索，替代硬编码多级Q查询的非扩展方案。以下是两种可行的实现方式：

方案1：预计算分类路径（最简单实现） #

通过给Category模型添加path字段存储完整层级路径，实现一次搜索覆盖所有祖先分类。

步骤1：更新Category模型 #

添加path字段并修改save方法维护路径：

class Category(models.Model):
    name = models.CharField(max_length=255)
    parent = models.ForeignKey('self', on_delete=models.SET_NULL, related_name='subcategories', null=True, blank=True)
    # 添加带索引的路径字段，加速搜索
    path = models.CharField(max_length=1000, blank=True, editable=False, db_index=True)

    class Meta:
        verbose_name_plural = 'Categories'

    def __str__(self):
        return self.name
    
    def save(self, *args, **kwargs):
        # 替换名称中的斜杠，避免路径结构断裂
        safe_name = self.name.replace('/', '-')
        
        # 根据父类路径构建当前分类路径
        if self.parent:
            self.path = f"{self.parent.path}/{safe_name}"
        else:
            self.path = safe_name
        
        super().save(*args, **kwargs)
        
        # 若名称或父类变更，递归更新所有子分类的路径
        if self.pk:
            old_category = Category.objects.get(pk=self.pk)
            if old_category.name != self.name or old_category.parent != self.parent:
                for subcategory in self.subcategories.all():
                    subcategory.save()

步骤2：更新Product Admin的搜索字段 #

修改search_fields，用category__path替代硬编码的父类层级：

class ProductAdmin(admin.ModelAdmin):
    # ... 其他已有字段 ...
    search_fields = ['name', 'price', 'brand__name', 'category__path']
    # ... 剩余代码 ...

现在搜索任意分类名称（如"电子产品"）时，会自动匹配该分类及其所有子分类下的产品。

方案2：递归CTE查询（无需修改模型，适合大数据集） #

使用递归公共表表达式（CTE）动态获取搜索分类的所有子分类，无需修改Category模型（要求Django 3.2+）。

更新Product Admin的get_queryset方法 #

修改get_queryset方法，在搜索时使用递归CTE：

from django.db.models.expressions import RawSQL

class ProductAdmin(admin.ModelAdmin):
    # ... 其他已有字段 ...
    search_fields = ['name', 'price', 'brand__name', 'category__name']

    def get_queryset(self, request):
        queryset = super().get_queryset(request).select_related('brand', 'category')
        search_query = request.GET.get('q')
        
        if search_query:
            # 获取匹配搜索关键词的分类ID（如需精确匹配，将name__icontains改为name=）
            matching_category_ids = Category.objects.filter(name__icontains=search_query).values_list('id', flat=True)
            
            if matching_category_ids:
                # 递归CTE查询所有子分类ID，替换core_category为你的实际表名（appname_modelname）
                cte_sql = """
                WITH RECURSIVE category_tree AS (
                    SELECT id, parent_id FROM core_category WHERE id IN %s
                    UNION ALL
                    SELECT c.id, c.parent_id FROM core_category c
                    JOIN category_tree ct ON c.parent_id = ct.id
                )
                SELECT id FROM category_tree
                """
                # 执行CTE获取所有子分类ID
                descendant_ids = Category.objects.raw(cte_sql, [tuple(matching_category_ids)]).values_list('id', flat=True)
                
                # 过滤出属于这些分类的产品
                queryset = queryset.filter(category_id__in=descendant_ids)
        
        return queryset

该方案动态获取所有层级的子分类，无需硬编码任何层级，且通过单次数据库查询实现，性能高效。

方案对比 #

• 路径字段方案：适合中小数据集，实现简单，直接利用Django内置搜索功能。
• CTE查询方案：适合大数据集或无法修改模型的场景，需熟悉SQL CTE语法，但无需维护额外字段。

内容的提问来源于stack exchange，提问作者Anee Mes