Il prefetch_jango di Django è valido solo per il conteggio

Question

Ho una situazione simile a questa (il codice effettivo è associato a un modello e omesso per brevità).

threads = Thread.objects.all() 
for thread in threads: 
    print(thread.comments.count()) 
    print(thread.upvotes.count()) 

sono riuscito a ridurre notevolmente il numero totale di query utilizzando il metodo impressionante prefetch_related di Django.

threads = Thread.objects.prefetch_related('comments').prefetch_related('upvotes') 

Tuttavia mi chiedo se questa situazione potrebbe essere ulteriormente ottimizzata. Da quanto ho capito, prefetch_related recupera tutti i dati associati ai modelli correlati. Visto che mi interessa solo la quantità di modelli correlati e non i modelli stessi, sembra che questa query possa essere ulteriormente ottimizzata in modo che non recuperi un mucchio di dati non necessari. C'è un modo per farlo in Django senza scendere a SQL raw?

Answer 1

Hai ragione, è uno spreco recuperare tutti i dati dal database se tutto ciò che vuoi è ottenere il conteggio. Suggerisco di annotazione:

threads = (Thread.objects.annotate(Count('comments', distinct=True)) 
         .annotate(Count('upvotes', distinct=True))) 
for thread in threads: 
    print(thread.comments__count) 
    print(thread.upvotes__count) 

Vedere la annotation documentation per ulteriori informazioni.