在单个查询中获取外键对象 - Django

时间:2010-05-14 09:16:37

标签: python django django-queryset

我的django代码中有2个模型:

class ModelA(models.Model):
    name = models.CharField(max_length=255)
    description = models.CharField(max_length=255)
    created_by = models.ForeignKey(User)

class ModelB(models.Model):
    category = models.CharField(max_length=255)
    modela_link = models.ForeignKey(ModelA, 'modelb_link')
    functions = models.CharField(max_length=255)
    created_by = models.ForeignKey(User)

Say ModelA有100条记录,所有记录都可能有也可能没有指向ModelB的链接

现在说我想获得每个ModelA记录的列表以及ModelB的数据

我愿意:

list_a = ModelA.objects.all()

然后要获取ModelB的数据,我必须这样做

for i in list_a:
    i.additional_data = i.modelb_link.all()

然而,这会在i的每个实例上运行查询。从而使101个查询运行。

有没有办法在一个查询中运行这一切。或者至少少于101个查询。

我已尝试输入ModelA.objects.select_related().all(),但这似乎没有任何效果。

由于

3 个答案:

答案 0 :(得分:7)

正如Ofri所说,select_related只适用于前锋关系,而不是反向关系。

在Django中没有自动跟踪反向关系的内置方法,但请参阅my blog post以获得合理有效地执行此操作的技术。基本思路是一次获取每个项目的所有相关对象,然后手动将它们与相关项目相关联 - 这样您就可以在2个查询中而不是n + 1中进行。

答案 1 :(得分:2)

Django ORM是一件好事,但有些事情比手动更好。 您可以在单个查询中导入连接游标并执行原始sql。

from django.db import connection
cur=connection.cursor()
cur.execute(query)
rows = cur.fetchall()

您的查询应该是( for MySQL

SELECT * FROM appname_modela INNER JOIN appname_modelb ON appname_modela.id=appname_modelb.modela_link_id

答案 2 :(得分:1)

原因.select_related()不起作用,是.select_related()用于跟踪外键。您的ModelA没有ModelB的外键。它的ModelB具有ModelA的外键。 (因此ModelA实例可以有多个与之相关的ModelB实例)。

您可以使用它在2个查询和一些python代码中执行此操作:

list_b = ModelB.objects.all()
list_a = ModelA.objects.all()
for a in list_a:
    a.additional_data = [b for b in list_b if b.modela_link_id==a.id]