Question

我有一个循环的Category模型（每个类别可以有一个父Category），我也有Product模型，其中ForeignKey指向“核心类别”，如下所示。是否有可能查询顶级类别（而非核心）中的所有产品？

例如，我的类别树是：

---Sales (main) (parent = None)
    ---Crm (core) (parent = Sales)
    ---Marketing Automation (core) (parent = Sales)
    ---AI (parent = Sales)
       ---BOTS (core) (parent = AI)
       ---NLP (core) (parent = AI)

我要查询“销售”或“ AI”中的所有产品

型号：

class Category(models.Model):
    name = models.CharField(max_length=255)
    link = models.URLField(max_length=255, null=True, blank=True)
    core = models.BooleanField()
    main = models.BooleanField()
    parent = models.ForeignKey('self', on_delete=models.SET_NULL,
                               null=True, default=None)

    def __str__(self):
        return self.name


class Product(models.Model):
    name = models.CharField(max_length=255)
    link = models.URLField(max_length=255, null=True, blank=True)
    category = models.ForeignKey(Category, on_delete=models.SET_NULL,
                                 null=True, default=None)

    def __str__(self):
       return self.name

感谢您的帮助！

编辑：我认为在这种情况下，最好的等同方法是文件夹中的文件。单个文件（产品）仅知道其父目录（核心类别），但是此父目录可以是其父目录的子文件夹。我想查询顶层文件夹（类别）中子目录中的所有文件（产品）。

Answer 1

首先，通过在category_set实例上使用反向外键查找Category获取所有子类别。

解决方案：

def recursive_sub_category(top_level_category_name, sub_core_categories=None):
    if sub_core_categories is None:
        sub_core_categories = list()

    top_level_category = Category.objects.get(name=top_level_category_name)

    _sub_core_categories = top_level_category.category_set.filter(core=True, main=False)

    if _sub_core_categories.count() > 0:
        sub_core_categories.append(*_sub_core_categories)

    sub_main_categories = top_level_category.category_set.filter(main=True, core=False)
    for category in sub_main_categories:
        _sub_core_categories = recursive_sub_category(
            category.name,
            sub_core_categories=sub_core_categories
        )
        if len(_sub_core_categories) > 0:
            sub_core_categories.append(*_sub_core_categories)

    return sub_core_categories

sub_core_categories = recursive_sub_category("Sales")
core_products_under_sales = Product.objects.filter(
    category__in=sub_core_categories
)

django递归模型查找

1 个答案: