Как сделать QuerySet для Post в месяц

Question

У меня есть модель как таковая

class Post(models.Model):
   submission_time = models.DateTimeField()
   user = models.ForeignKey(User)

, и я хотел бы иметь набор запросов, в котором будет возвращаться количество сообщений, созданных пользователем за месяц.

Я пытался использоватьExtractMonth как:

user.post_set.annotate(month_sub=ExtractMonth('submission_time')).values('month_sub').annotate(count=Count('month_sub'))

Но он дает мне QuerySet, например, такой:

<QuerySet [{'month_sub': 5, 'count': 1}, {'month_sub': 6, 'count': 1}, {'month_sub': 6, 'count': 1}, {'month_sub': 6, 'count': 1}, {'month_sub': 6, 'count': 1}, {'month_sub': 6, 'count': 1}, {'month_sub': 6, 'count': 1}]>

Вместо одного такого (который я хотел бы иметь):

<QuerySet [{'month_sub': 5, 'count': 1}, {'month_sub': 6, 'count': 7}]>

Любые идеи о том, как получить набор запросов, как это?

Answer 1

Вы совсем близко, единственное, чего вам не хватает, это .order_by('month_sub') (да, я знаю, это звучит немного странно ).Поэтому вы должны написать:

user.post_set.annotate(
    month_sub=ExtractMonth('submission_time')
).values('month_sub').annotate(
    count=Count('month_sub')
)<b>.order_by('month_sub')</b>

В результате запрос будет выглядеть следующим образом:

SELECT EXTRACT(MONTH FROM `post`.`submission_time`) AS `month_sub`,
       COUNT(EXTRACT(MONTH FROM `post`.`submission_time`)) AS `count`
FROM `post`
WHERE `post`.`user_id` = 123
GROUP BY EXTRACT(MONTH FROM `post`.`submission_time`)
ORDER BY `month_sub` ASC

(где 123 следует заменить на id из user).

Возможно, вы можете повысить производительность на крошечный бит, используя Count('id'), хотя возможно, что это не оказывает (заметного) воздействия:

user.post_set.annotate(
    month_sub=ExtractMonth('submission_time')
).values('month_sub').annotate(
    count=Count(<b>'id'</b>)
).order_by('month_sub')

Если я запускаю это на примере базы данных, я получаю:

<QuerySet [{'count': 273, 'month_sub': 1},
           {'count': 172, 'month_sub': 2},
           {'count': 565, 'month_sub': 3},
           {'count': 59, 'month_sub': 4},
           {'count': 452, 'month_sub': 5},
           {'count': 550, 'month_sub': 6},
           {'count': 622 'month_sub': 7},
           {'count': 43, 'month_sub': 8},
           {'count': 357, 'month_sub': 9},
           {'count': 378, 'month_sub': 10},
           {'count': 868, 'month_sub': 11},
           {'count': 293, 'month_sub': 12}]>

(сделал некоторое форматирование, чтобы упростить проверку результата)