父表上的django prefetch_related和select_related

时间:2017-05-16 13:28:24

标签: python django postgresql

我有以下数据库结构:

catogory
   id
   name
   parent_id

class Category(models.Model)
  name = models.CharField(max_length=400, blank=True, null=True)
  parent = models.ForeignKey("self", blank=True, null=True)

我需要获取所有类别及其父级。

如果我这样做:

Category.objects.select_related("parent").filter(pk__in=[ids])

它只会返回第一级的父级。

如何使用最少的db调用来检索所有级别的父级?

我的方法是构建一个新的非db模型对象,例如: CategorySerializer将这些Category模型转移到非db模型,以便逻辑层可以使用它

1 个答案:

答案 0 :(得分:1)

如果你创建了一个可以递归调用的函数,直到没有更多父母离开

,该怎么办?
def get_categories(ids):
    ids = list(Category.objects.select_related("parent").filter(pk__in=[ids]).values_list('parent_id', flat=True))
    if len(ids) == 0:
        return []
    else:
        return ids + get_categories(ids)

所以从这个电话开始

all_parent_ids = []
for category in list(Category.objects.all()):
    all_parent_ids.append(get_categories(category.parent_id))

我还没有对此进行测试,因此可能需要进行一些改进,但我相信这个想法是正确的。