Làm cách nào để lọc các đối tượng để đếm số chú thích trong Django?

Hãy xem xét các mô hình Django đơn giản Eventvà 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)

Thật dễ dàng để chú thích truy vấn sự kiện với tổng số người tham gia:

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

Làm cách nào để chú thích với số lượng người tham gia được lọc theo is_paid=True?

Tôi cần truy vấn tất cả các sự kiện bất kể số lượng người tham gia, ví dụ: tôi không cần lọc theo kết quả có chú thích. Nếu có 0người tham gia, đó là ok, tôi chỉ cần 0trong giá trị chú thích.

Các ví dụ từ tài liệu không làm việc ở đây, bởi vì nó không bao gồm các đối tượng từ truy vấn thay vì chú thích chúng với 0.

Cập nhật. Django 1.8 có tính năng biểu thức điều kiện mới , vì vậy bây giờ chúng ta có thể làm như sau:

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()
    )))

Cập nhật 2. Django 2.0 có tính năng tổng hợp có Điều kiện mới , hãy xem câu trả lời được chấp nhận bên dưới.

Tính năng tổng hợp có điều kiện trong Django 2.0 cho phép bạn giảm thêm lượng lỗi mà điều này đã có trong quá khứ. Điều này cũng sẽ sử dụng Postgres 'filter logic , nhanh hơn một chút so với trường hợp tổng (tôi đã thấy những con số như 20-30% được chia thành xung quanh).

Dù sao, trong trường hợp của bạn, chúng tôi đang xem xét một cái gì đó đơn giản như:

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

Có một phần riêng biệt trong tài liệu về lọc các chú thích . Nó giống như tập hợp có điều kiện nhưng giống như ví dụ của tôi ở trên. Dù theo cách nào thì điều này cũng lành mạnh hơn rất nhiều so với các truy vấn phụ gnarly mà tôi đã làm trước đây.

Vừa phát hiện ra rằng Django 1.8 có tính năng biểu thức điều kiện mới , vì vậy bây giờ chúng ta có thể làm như sau:

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()
    )))

CẬP NHẬT

Phương pháp truy vấn phụ mà tôi đề cập hiện được hỗ trợ trong Django 1.11 thông qua biểu thức truy vấn con .

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()
    )
)

Tôi thích điều này hơn là tổng hợp (tổng + trường hợp) , vì nó sẽ nhanh hơn và dễ dàng hơn được tối ưu hóa (với lập chỉ mục thích hợp) .

Đối với phiên bản cũ hơn, bạn có thể đạt được điều tương tự bằng cách sử dụng .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`"
})

Tôi khuyên bạn nên sử dụng .valuesphương pháp của bộ truy vấn của bạn Participantđể thay thế.

Tóm lại, những gì bạn muốn làm được đưa ra bởi:

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

Một ví dụ đầy đủ như sau:

1. Tạo 2 Events:

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

2. Thêm Participants vào chúng:

   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. Nhóm tất cả các Participants theo eventtrường của chúng :

   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)...']>

   Ở đây cần có sự khác biệt:

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

   Điều đã .valuesvà .distinctđang làm ở đây là họ đang tạo hai nhóm Participantđược nhóm theo phần tử của họ event. Lưu ý rằng những thùng chứa Participant.

4. Sau đó, bạn có thể chú thích các nhóm đó vì chúng chứa nhóm gốc Participant. Ở đây chúng tôi muốn đếm số lượng Participant, điều này được thực hiện đơn giản bằng cách đếm ids của các phần tử trong các nhóm đó (vì chúng là Participant):

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

5. Cuối cùng, bạn chỉ muốn Participantvới một thực is_paidthể True, bạn có thể chỉ cần thêm một bộ lọc vào trước biểu thức trước đó và điều này mang lại biểu thức được hiển thị ở trên:

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

Hạn chế duy nhất là bạn phải truy xuất Eventsau đó vì bạn chỉ có idtừ phương pháp trên.

Kết quả tôi đang tìm kiếm:

• Những người (người được giao) đã thêm nhiệm vụ vào báo cáo. - Tổng số người duy nhất
• Những người có nhiệm vụ được thêm vào báo cáo nhưng đối với nhiệm vụ có khả năng thanh toán chỉ hơn 0.

Nói chung, tôi sẽ phải sử dụng hai truy vấn khác nhau:

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

Nhưng tôi muốn cả hai trong một truy vấn. Vì thế:

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

Kết quả:

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