Django .values_list()替代方法为ForeignKey字段的模型返回QuerySet?

时间:2016-12-07 13:44:59

标签: python django django-queryset

我正在寻找一种基于模型ForeignKey字段将一种类型的QuerySet转换为另一种类型的干净方法,所以基本上类似于.values_list('my_fk', flat=True)但返回正确的QuerySet而不是values_list()变体。< / p>

例如:

class Parent(models.Model):
    child = models.ForeignKey(Child)
    ...

children_qs = Parent.objects.filter(...).theMagicMethod('child')  

此处children_qs现在应该是先前查询中使用的所有Child实例的查询集,而不是返回Parent实例的查询集。

您可以使用自定义查询集和__in查找来执行此操作,但感觉有点臭:

class ParentQuerySet(models.QuerySet):
    ...
    def children(self):
        return Child.objects.filter(id__in=self.values_list('child_id', flat=True))

这将从Parent的查询集中的记录中获取所有child_id FK,并直接重新查询Child。当我检查SQL它做了一个子查询,我不确定这是最优还是有一些奇怪的副作用。它确实看起来像原始父查询的排序消失了,因此重复。

有没有人比这更好?

注意:我知道我可以通过Child直接查询并使用反向查找过滤Parent的字段,但这不支持您可以在主模型上执行的所有操作。

2 个答案:

答案 0 :(得分:3)

试试这个,它将返回Child类的query_set

parent_primary_keys = Parent.objects.filter(...).values_list('pk',flat=True)

 children_qs = Child.objects.filter(id__in=parent_primary_keys)

答案 1 :(得分:0)

听起来你可以利用Django的函数prefetch_relatedCheck out this answerdjango's documentation