Django计算多对一关系的特定项目

Question

我有一个 Django 应用程序，用户可以在其中发布消息，其他用户可以对答案进行投票，这与 SO 非常相似。我在尝试从模板中获取“竖起大拇指”和“竖起大拇指”计数时遇到问题，我希望有人可以帮助我。 PostVote 与 Post 类是多对一的关系。这是我的模型的样子：

class Post(models.Model):
    account = models.ForeignKey(Account)
    message = models.CharField(max_length=1024)
    timestamp = models.DateTimeField('post timestamp')

class PostVote(models.Model):
    post = models.ForeignKey(Post)
    account = models.ForeignKey(Account)
    vote = models.CharField(max_length=16, choices=VOTE_CHOICES)
    timestamp = models.DateTimeField('vote timestamp')

这是我获取帖子的方式：

posts = Post.objects.all().order_by('-timestamp')[:10]

我的模板大致如下：

{% for post in posts %}
<div>Thumbs up count: {{ WHAT_HERE }}</div>
<div>Thumbs down count: {{ WHAT_HERE }}</div>
{% endfor %}

我怎样才能得到那里的计数？我确信它以某种方式涉及“注释”，但我很难想出这个。任何帮助将不胜感激！

Answer 1

您不应该真的在模板中做逻辑。为您的 Post 模型添加几个计数方法：

class Post(models.Model):
    account = models.ForeignKey(Account)
    message = models.CharField(max_length=1024)
    timestamp = models.DateTimeField('post timestamp')

    def upvote_count(self):
        return self.postvote_set.filter(vote=VOTE_CHOICES[0][0]).count()

    def downvote_count(self):
        return self.postvote_set.filter(vote=VOTE_CHOICES[1][0]).count()

然后在您的模板中使用它们：

{% for post in posts %}
<div>Thumbs up count: {{ post.upvote_count }}</div>
<div>Thumbs down count: {{ post.downvote_count }}</div>
{% endfor %}