Як відфільтрувати об’єкти для анотації підрахунку в Django?

Розглянемо прості моделі Джанго Eventта Participant:

class Event(models.Model):
    title = models.CharField(max_length=100)

class Participant(models.Model):
    event = models.ForeignKey(Event, db_index=True)
    is_paid = models.BooleanField(default=False, db_index=True)

Помітити запит на події легко за допомогою загальної кількості учасників:

events = Event.objects.all().annotate(participants=models.Count('participant'))

Як коментувати кількість фільтрованих учасників is_paid=True?

Мені потрібно запитувати всі події незалежно від кількості учасників, наприклад, мені не потрібно фільтрувати за анотованим результатом. Якщо є 0учасники, це нормально, мені просто потрібно 0в анотованому значенні.

Приклад з документації тут не працює, так як він виключає об'єкти з запиту замість аннотирования їх 0.

Оновлення. Django 1.8 має нові умовні вирази , тому тепер ми можемо робити так:

events = Event.objects.all().annotate(paid_participants=models.Sum(
    models.Case(
        models.When(participant__is_paid=True, then=1),
        default=0,
        output_field=models.IntegerField()
    )))

Оновлення 2. Django 2.0 має нову функцію умовного агрегування , див . Прийняту відповідь нижче.

Умовна агрегація в Django 2.0 дозволяє ще більше зменшити кількість факсу, що було раніше. Для цього також буде використана filterлогіка Postgres , яка є дещо швидшою, ніж підсумок (я бачив цифри, як 20-30% навколо).

У будь-якому випадку, у вашому випадку ми дивимось на щось таке просто, як:

from django.db.models import Q, Count
events = Event.objects.annotate(
    paid_participants=Count('participants', filter=Q(participants__is_paid=True))
)

В документах є окремий розділ про фільтрування приміток . Це те саме, що і умовне агрегування, але більше схоже на мій приклад вище. У будь-якому випадку це набагато здоровіше, ніж те, що я робив раніше.

Щойно виявив, що Django 1.8 має нові умовні вирази , тож тепер ми можемо зробити так:

events = Event.objects.all().annotate(paid_participants=models.Sum(
    models.Case(
        models.When(participant__is_paid=True, then=1),
        default=0, output_field=models.IntegerField()
    )))

ОНОВЛЕННЯ

Підхід до запитів, про який я згадую, зараз підтримується в Django 1.11 через підзапроси-вирази .

Event.objects.annotate(
    num_paid_participants=Subquery(
        Participant.objects.filter(
            is_paid=True,
            event=OuterRef('pk')
        ).values('event')
        .annotate(cnt=Count('pk'))
        .values('cnt'),
        output_field=models.IntegerField()
    )
)

Я віддаю перевагу цьому над агрегацією (сума + регістр) , оскільки оптимізувати її слід швидше і простіше (за умови належної індексації) .

Для старшої версії те ж саме можна досягти, використовуючи .extra

Event.objects.extra(select={'num_paid_participants': "\
    SELECT COUNT(*) \
    FROM `myapp_participant` \
    WHERE `myapp_participant`.`is_paid` = 1 AND \
            `myapp_participant`.`event_id` = `myapp_event`.`id`"
})

Я б запропонував використати .valuesметод вашого Participantнабору запитів.

Коротше кажучи, те, що ви хочете зробити, дає:

Participant.objects\
    .filter(is_paid=True)\
    .values('event')\
    .distinct()\
    .annotate(models.Count('id'))

Повний приклад:

1. Створіть 2 Eventс:

   event1 = Event.objects.create(title='event1')
   event2 = Event.objects.create(title='event2')
   

2. Додайте Participantдо них s:

   part1l = [Participant.objects.create(event=event1, is_paid=((_%2) == 0))\
             for _ in range(10)]
   part2l = [Participant.objects.create(event=event2, is_paid=((_%2) == 0))\
             for _ in range(50)]
   

3. Згрупуйте всі Participants за своїм eventполем:

   Participant.objects.values('event')
   > <QuerySet [{'event': 1}, {'event': 1}, {'event': 1}, {'event': 1}, {'event': 1}, {'event': 1}, {'event': 1}, {'event': 1}, {'event': 1}, {'event': 1}, {'event': 2}, {'event': 2}, {'event': 2}, {'event': 2}, {'event': 2}, {'event': 2}, {'event': 2}, {'event': 2}, {'event': 2}, {'event': 2}, '...(remaining elements truncated)...']>
   

   Тут потрібні різні:

   Participant.objects.values('event').distinct()
   > <QuerySet [{'event': 1}, {'event': 2}]>
   

   Що .valuesі .distinctтут роблять, це те, що вони створюють два відра Participants, згруповані за своїм елементом event. Зауважте, що ці відра містять Participant.

4. Потім ви можете коментувати ці відра, оскільки вони містять набір оригіналів Participant. Тут ми хочемо порахувати кількість Participant, це робиться просто шляхом підрахунку ids елементів у цих відрах (оскільки вони є Participant):

   Participant.objects\
       .values('event')\
       .distinct()\
       .annotate(models.Count('id'))
   > <QuerySet [{'event': 1, 'id__count': 10}, {'event': 2, 'id__count': 50}]>
   

5. Нарешті, ви хочете лише Participantз is_paidістотою True, ви можете просто додати фільтр перед попереднім виразом, і це дасть вираз, показаний вище:

   Participant.objects\
       .filter(is_paid=True)\
       .values('event')\
       .distinct()\
       .annotate(models.Count('id'))
   > <QuerySet [{'event': 1, 'id__count': 5}, {'event': 2, 'id__count': 25}]>
   

Єдиний недолік полягає в тому, що ви повинні отримати Eventзгодом, як у вас є лише idметод, описаний вище.

Який результат я шукаю:

• Люди (правонаступники), які мають завдання, додані до звіту. - Загальна унікальна кількість людей
• Люди, які мають завдання, додані до звіту, але для завдань, оплачуваність яких перевищує лише 0.

Загалом, мені доведеться використовувати два різні запити:

Task.objects.filter(billable_efforts__gt=0)
Task.objects.all()

Але я хочу обидва в одному запиті. Звідси:

Task.objects.values('report__title').annotate(withMoreThanZero=Count('assignee', distinct=True, filter=Q(billable_efforts__gt=0))).annotate(totalUniqueAssignee=Count('assignee', distinct=True))

Результат:

<QuerySet [{'report__title': 'TestReport', 'withMoreThanZero': 37, 'totalUniqueAssignee': 50}, {'report__title': 'Utilization_Report_April_2019', 'withMoreThanZero': 37, 'totalUniqueAssignee': 50}]>