django annoter modèles avec une valeur globale basée sur requête

Question

Disons que j'ai la structure du modèle suivant:

Parent(): 

Child(): 
parent = ForeignKey(Parent) 

GrandChild(): 
child = ForeignKey(Child) 
state = BooleanField() 
num = FloatField() 

Je suis en train de le parent ViewSet, récupérer les éléments suivants:

1. Le Nombre d'enfants.
2. Somme des champs 'num' lorsque 'state' a la valeur True.

je peux faire ce qui suit:

queryset = Parent.objects\ 
    .annotate(child_count=Count('child'))\ 
    .annotate(sum_total=Sum('child__grandchild__num')) 

Cela me donne (1) mais au lieu de (2) il me donne la somme pour TOUS les petits-enfants. Comment puis-je filtrer correctement les petits-enfants tout en m'assurant que tous les objets Parent sont toujours dans le QuerySet?

Answer 1

Essayez d'utiliser le filtre avant la annoter

queryset = Parent.objects.filter(child__grandchild__state=True')\ 
    .annotate(child_count=Count('child'))\ 
    .annotate(sum_total=Sum('child__grandchild__num')) 

Answer 2

Quelle version de django utilisez-vous? Vous pouvez également utiliser la sous-requête si la version est prise en charge.

Parent.objects 
.annotate(child_count=Count('child')) 
.annotate(
    grandchild_count_for_state_true=Subquery(
     GrandChild.objects.filter(
      state=True, 
      child=OuterRef('pk') 
     ).values('parent') 
     .annotate(cnt=Sum('child__grandchild__num')) 
     .values('cnt'), 
     num=models.IntegerField() 
    ) 
) 

Vous pouvez optimiser cela via une requête d'agrégation.