在 Django 中跨列聚合

Question

我试图弄清楚是否有一种方法可以使用 Django 的 ORM 进行有点复杂的聚合，或者我是否必须使用 extra() 来坚持一些原始 SQL。

这是我的对象模型（仅显示要点）：

class Submission(Models.model)
  favorite_of = models.ManyToManyField(User, related_name="favorite_submissions")

class Response(Models.model)
  submission = models.ForeignKey(Submission)
  voted_up_by = models.ManyToManyField(User, related_name="voted_up_responses")

我想要做的是对给定提交的所有投票进行求和：即对其任何响应的所有投票，然后还包括人数谁将提交标记为最喜欢的。

我使用以下代码进行第一部分的工作；这将返回每个提交的所有回复的总票数：（

submission_list = Response.objects\
  .values('submission')\
  .annotate(votes=Count('voted_up_by'))\
  .filter(votes__gt=0)\
  .order_by('-votes')[:TOP_NUM]

因此，在获得总票数后，我按降序排序并返回前 TOP_NUM 提交，以获得“最佳”列表。）

该部分有效。您是否可以建议将喜欢每个提交内容的人数纳入投票中？ （为了可移植性，我宁愿避免 extra() ，但我认为这可能是必要的，并且我愿意使用它。）

编辑：在阅读下面的建议后，我意识到我应该更清楚我的问题的描述。理想的解决方案是允许我按总票数（voted_up_by 和 favorited 的总和）排序，然后仅选择数据库中的前几名。如果这是不可能的，那么我愿意加载每个响应的一些字段并在 Python 中进行处理；但由于我将处理 100,000 多条记录，因此最好避免这种开销。 （另外，对 Adam 和 Dmitry：我很抱歉延迟回复！）

Answer 1

一种可能性是稍微重新安排当前的查询。如果您尝试如下操作会怎么样：

submission_list = Response.objects\
    .annotate(votes=Count('voted_up_by'))\
    .filter(votes__gt=0)\
    .order_by('-votes')[:TOP_NUM]
submission_list.query.group_by = ['submission_id']

这将返回 Response 对象的查询集（具有相同提交的对象将被集中在一起）。为了访问相关的提交和/或 favorite_of 列表/计数，您有两个选择：

num_votes = submission_list[0].votes
submission = submission_list[0].submission
num_favorite = submission.favorite_of.count()

或者...

submissions = []
for response in submission_list:
    submission = response.submission
    submission.votes = response.votes
    submissions.append(submission)
num_votes = submissions[0].votes
submission = submissions[0]
num_favorite = submission.favorite_of.count()

基本上第一个选项的优点是仍然是查询集，但您必须确保访问提交对象为了获取有关提交的任何信息（因为查询集中的每个对象在技术上都是一个响应）。第二个选项的优点是成为包含 favorite_of 列表和投票的提交列表，但它不再是查询集（因此请确保您之后不再需要更改查询）。

Answer 2

您可以在另一个查询中计算收藏夹数量，例如

favorite_list = Submission.objects.annotate(favorites=Count(favorite_of))

在添加两个列表中的值之后：

total_votes = {}
for item in submission_list:
    total_votes[item.submission.id] = item.voted_by
for item in favorite_list:
    has_votes = total_votes.get(item.id, 0)
    total_votes[item.id] = has_votes + item.favorites

我在字典中使用 ids，因为提交对象不会相同。如果您需要提交本身，您可以再使用一本字典或存储元组（提交、投票）而不仅仅是投票。

补充：这个解决方案比以前的更好，因为您只有两个数据库请求。