如何使用Django模型以更有效的方式获取此查询集?

时间:2016-01-24 16:25:42

标签: django django-models django-queryset

我的 models.py 如下所示:

class Post(models.Model):
    user = models.ForeignKey(User, related_name = 'user_posts')
    title = models.CharField(max_length = 140)
    votes = models.IntegerField(default = 0)

class Vote(models.Model):
    user = models.ForeignKey(User, related_name = 'user_votes')
    post = models.ForeignKey(Post, related_name = 'post_votes')
    is_voted = models.BooleanField(default = True)
    class Meta:
        unique_together = [('user', 'post')]

让我解释一下我的投票系统是如何设置的。当用户第一次投票时,会创建一个新的投票对象。如果用户首先投票,则用户可以对该帖子进行投票。在这种情况下,Vote对象中的is_voted属性设置为False。

现在在视图中,我需要一个用户已经投票的帖子列表。这意味着存在用于帖子和用户组合的投票对象,AND,该对象的is_voted属性为True。

以下是我目前正在尝试的方法:

views.py

def user_profile(request, pk):
    # Get user using pk
    u = User.objects.get(pk = pk)

    # Get a list of Votes using the user instance
    votes = Vote.objects.filter(user = u, is_voted = True)

    # Getting the list of posts using the votes list
    post_list = Post.objects.none() # Generating empty list
    for vote in votes:
        # Adding vote.post to post_list using chaining.
        .....

这实际上有效,但感觉非常黑客。有没有办法在没有for循环的情况下生成查询集?我猜我可以使用related_name,但我不知道该怎么做。

2 个答案:

答案 0 :(得分:2)

我认为此查询集应该为用户提供所有帖子(我还没试过):

votes = Post.objects.filter(post_votes__user=u, post_votes__is_voted=True)

这是从模型的反向查找,将外键指向包含它的模型。

答案 1 :(得分:0)

你可以试试这个:

post_list = [p.post for p in Vote.objects.filter(user = u, is_voted = True).select_related('post')]