仅查询具有关系的对象的PostgreSQL数据库(使用Django)

时间:2017-01-12 14:14:54

标签: python django postgresql

我目前正在通过官方教程学习Django,并决定尝试按建议添加一些其他功能和测试here

我现在在DB中有两个模型,看起来基本上是这样的:

class Question(models.Model):
    question_text = models.CharField(max_length=200)
    pub_date = models.DateTimeField('date published')

class Choice(models.Model):
    question = models.ForeignKey(Question, on_delete=models.CASCADE)
    choice_text = models.CharField(max_length=200)
    votes = models.IntegerField(default=0)

我设置了一个视图来显示问题列表。我想要做的是为视图创建一个查询,该查询仅包含具有相关选项的问题对象。任何没有相关选择对象的问题对象都不应包含在结果集中。这个查询也忽略了将来的问题并对它们进行了排序,但这部分很好,就在本教程之外。

这是我到目前为止所提出的。它似乎有效,但我不禁想到我已经以一种非常倒退和低效的方式完成它。是否可以创建一个处理数据库中所有内容的查询,而不是两个查询和一个列表解析?

choices = Choice.objects.prefetch_related(
    'question').distinct('question')
question_ids = [x.question.id for x in choices]
return Question.objects.filter(
    id__in=question_ids).filter(
    pub_date__lte=timezone.now()).order_by('-pub_date')[:5]

1 个答案:

答案 0 :(得分:1)

您可以通过与其关联的Question个对象的计数对Choice个查询集进行注释,然后排除零计数的对象:

from django.db.models import Count
questions = Question.objects \
        .annotate(choice_cnt=Count('choice')) \
        .exclude(choice_cnt=0) \
        .filter(pub_date__lte=timezone.now()) \
        .order_by('-pub_date')[:5]