Django Rest DRF: obtention du nombre d'objets pour la relation inverse

Question

Disons que j'ai deux modèles.

Modèle

class Item(models.Model): 
    name = models.CharField(max_length=32) 
    # other fields 

class ItemRelation(models.Model): 
    item = models.ForeignKey(Item, related_name='relations_item') 
    user = models.ForeignKey(User, related_name='relations_user') 
    has_viewed = models.BooleanField(default=False) 
    has_loved = models.BooleanFields(default=False) 

Maintenant, ce que je veux faire est que je veux obtenir le view_count et love_count pour tous les éléments à l'aide django reste api.

views.py

class ItemView(ListAPIView): 
    queryset = Items.objects.all().prefetch_related(Prefetch('relations_item', 
      queryset=ItemRelation.objects.filter(Q(has_viewed=True) | Q(has_loved=True)) 
      ) 
    serializer_class = ItemSerializer 

Eh bien, ce fut le plan, mais je n'ai absolument aucune idée de la façon d'obtenir le view_count et love_count pour chaque élément de la liste-api-vue. J'ai essayé pas mal de choses sur mon sérialiseur mais je ne pense pas que ça va marcher. Je peux cependant utiliser SerializerMethod() pour faire le travail, mais cela passerait par le DB N + 1 nombre de fois. J'ai lu les docs avec quelques autres blogs pour prefetch_related et j'ai été capable de faire les choses facilement jusqu'à ce que ce problème de compte apparaisse.

Juste un échantillon de mon sérialiseur.

class ItemSerializer(serializers.ModelSerializer): 

    class Meta: 
     model = Item 
     fields = ['name', 'relations_item'] 

Answer 1

Je suppose que vous voulez compter le nombre de relations dans lequel a) et b has_viewed==True) has_loved==True.

Cela devrait être possible en utilisant annotate() avec Django Conditional Expressions:

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


class ItemView(ListAPIView): 
    queryset = Items.objects.annotate(
        view_count=Sum(
         When(relations_item__has_viewed=True, then=1), 
         output_field=IntegerField(), 
        ), 
        love_count=Sum(
         When(relations_item__has_loved=True, then=1), 
         output_field=IntegerField(), 
        ), 
       ) 
    # ...