Django mit Anmerkungen versehen queryset auf einem bestimmten Wert des relationalen Modells Attribut

Question

Angenommen, es eine Struktur ist wie folgt:

PARTICIPATION_STATUSES = (
    (0, 'No, thanks'), 
    (1, 'I may attend'), 
    (2, 'I\'ll be there'), 
) 

class Model1(models.Model): 
    # ... 

class Model2(models.Model): 
    status = models.PositiveIntegerField(
     _('participation status'), choices=PARTICIPATION_STATUSES)  
    field = models.ForeignKey(Model1, related_name='model1_participation') 

Was ich tun möchte, ist jedes Objekt von Model1 mit der Zählung der Model2 Objekte mit Anmerkungen versehen, wo Status a gleich spezifischer Wert (Statusnummer ist dieses spezielle Beispiel).

In meinem Pseudo-Code würde es wie folgt aussehen:

queryset = Model1.objects.all() 
queryset.annotate(declined=Count('model1_participation__status=0')) 
queryset.annotate(not_sure=Count('model1_participation__status=1')) 
queryset.annotate(accepted=Count('model1_participation__status=2')) 

Aber ich kann den queryset auf diese Weise nicht mit Anmerkungen versehen, wie Django nicht status=<n> lösen.

Was ist der richtige Weg, um zu erreichen, was ich will?

Answer 1

Wenn Sie Django 1.8 oder höher verwenden, können Sie Conditional Aggregations verwenden, diese sollten für annotate Querysets funktionieren.

from django.db.models import IntegerField, Case, When, Count 


queryset = Model1.objects.all() 

queryset = queryset.annotate(
    declined=Count(
     Case(When(model1_participation__status=0, then=1), 
      output_field=IntegerField()) 
    ), 
    not_sure=Count(
     Case(When(model1_participation__status=1, then=1), 
      output_field=IntegerField()) 
    ), 
    accepted=Count(
     Case(When(model1_participation__status=2, then=1), 
      output_field=IntegerField()) 
    ) 
)