Django ORM:选择相关集

Question

说我有2个型号:

class Poll(models.Model):
    category = models.CharField(u"Category", max_length = 64)
    [...]

class Choice(models.Model):
    poll = models.ForeignKey(Poll)
    [...]

给定一个Poll对象,我可以用以下方式查询其选择:

poll.choice_set.all()

但是,是否有一个实用程序函数来查询一组Poll中的所有选项？

实际上,我正在寻找类似下面的东西(不支持,我不寻求它是如何):

polls = Poll.objects.filter(category = 'foo').select_related('choice_set')
for poll in polls:
    print poll.choice_set.all() # this shouldn't perform a SQL query at each iteration

我做了一个(丑陋的)功能来帮助我实现这个目标:

def qbind(objects, target_name, model, field_name):
    objects = list(objects)
    objects_dict = dict([(object.id, object) for object in objects])
    for foreign in model.objects.filter(**{field_name + '__in': objects_dict.keys()}):
        id = getattr(foreign, field_name + '_id')
        if id in objects_dict:
            object = objects_dict[id]
            if hasattr(object, target_name):
                getattr(object, target_name).append(foreign)
            else:
                setattr(object, target_name, [foreign])
    return objects

用法如下:

polls = Poll.objects.filter(category = 'foo')
polls = qbind(polls, 'choices', Choice, 'poll')
# Now, each object in polls have a 'choices' member with the list of choices.
# This was achieved with 2 SQL queries only.

Django已经提供了更简单的东西吗？或者至少,一个片段以更好的方式做同样的事情.

你经常如何处理这个问题？

Answer 1

时间已过,现在Django 1.4中提供了此功能,并引入了prefetch_related() QuerySet函数.该功能有效地完成了建议qbind功能所执行的操作.即.执行两个查询,并且连接发生在Python域中,但现在由ORM处理.

原始查询请求现在将变为:

polls = Poll.objects.filter(category = 'foo').prefetch_related('choice_set')

如以下代码示例所示,pollsQuerySet可用于获取所有Choice对象,Poll而无需任何进一步的数据库命中:

for poll in polls:
    for choice in poll.choice_set:
        print choice

Answer 2

我想你所说的是,"我希望所有选择都能进行一系列民意调查." 如果是这样,试试这个:

polls = Poll.objects.filter(category='foo')
choices = Choice.objects.filter(poll__in=polls)

Answer 3

更新:自Django 1.4以来,内置了此功能:请参阅prefetch_related.

第一个答案:在你编写一个正在运行的应用程序,对其进行分析并证明N个查询实际上是数据库和负载场景的性能问题之前,不要浪费时间编写类似qbind的东西.

但也许你已经做到了.所以第二个答案:qbind()做了你需要做的事情,但是如果打包在一个自定义的QuerySet子类中,并且伴随着一个返回自定义QuerySet实例的Manager子类,它会更加惯用.理想情况下,您甚至可以使它们具有通用性,并可以重复使用任何反向关系.然后你可以这样做:

Poll.objects.filter(category='foo').fetch_reverse_relations('choices_set')

有关Manager/QuerySet技术的示例,请参阅此代码段,它解决了类似的问题,但对于通用外键的情况,而不是反向关系.将qbind()函数的内容与其中显示的结构相结合,为您的问题提供一个非常好的解决方案并不是太难.