从方法订购

Question

大家好，我有一个像这样的简单模型

def Article(models.Model):
    upvotes = models.ManyToManyField(User, related_name='article_upvotes')
    downvotes = models.ManyToManyField(User, related_name='article_downvotes')

    def votes(self):
        return self.upvotes - self.downvotes

，鉴于我可以做类似的事情，

article_votes = article.votes

我可以通过投票功能订购吗？ 之类的东西

article = Article.objects.order_by('votes')

像编辑

我目前不在我的开发系统附近，所以语法可能有点不对。

Answer 1

您可以在查询返回结果后对列表进行排序：

article_votes = sorted(article.votes, key=lambda a: a.votes())

sorted 获取一个列表并对其进行排序。您可以提供一个自定义函数，该函数接受一个元素并返回比较元素时要使用的值。 lambda a: a.votes() 是一个匿名函数，它接受一篇文章并返回该文章的投票数。

如果您无论如何都要检索所有文章，那么此解决方案没有任何缺点。另一方面，如果您只想要投票数最高的 10 篇文章，那么您将从数据库中提取所有文章，而不是让数据库进行排序，并且只返回前 10 篇文章。与纯 SQL 解决方案相比，这从数据库中检索更多的数据。

Answer 2

这是 Ned Batchelder 建议的更快版本 - 因为它在数据库中进行计数：

articles = list(Article.objects.annotate(upvote_num=models.Count('upvotes'), downvote_num=models.Count('downvotes')))
articles.sort(key=lambda a: a.upvotes - a.downvotes)

您也可以完全在数据库内执行此操作：

articles = Article.objects.raw("""
    SELECT DISTINCT id from articles_article,
    COUNT(DISTINCT upv) AS num_upvotes,
    COUNT(DISTINCT downv) AS num_downvotes,
    (num_upvotes - num_downvotes) AS score

    INNER JOIN [article_user_upvotes_m2m_table_name] AS upv
    ON upv.article_id = id

    INNER JOIN [article_user_downvotes_m2m_table_name] AS downv
    ON downv.article_id = id

    GROUP BY id
    ORDER BY score
""")

- 但我不确定双连接在您的情况下是否是一个好主意。另外，我不确定是否需要所有这些 DISCTINCT。这个查询很可能可以用更好的方式重写，但我目前没有想法。