如何在 django 中过滤/注释预取的对象（更有效）？

Question

我想用 prefetch_related 对象过滤查询集。此代码有效，但我想更有效地编写它。你有什么想法吗？

queryset = Song.objects.prefetch_related(
                Prefetch('harmonies', queryset=Harmony.objects.filter(someQuery)))

for s in queryset:
    if s.harmonies.count() > 0:
         songs.append(s.id)
queryset = queryset.filter(id__in=songs)

我尝试了类似的方法，但没有成功。

queryset = queryset.annotate(h_count=Count('harmonies')).exclude(h_count=0)

谢谢你的帮助。

Answer 1

您可以使用Exists subquery [Django-doc]，所以：

from django.db.models import Exists, OuterRef

queryset = Song.objects.filter(
    Exists(Harmony.objects.filter(someQuery, song_id=OuterRef('pk')))
)

这里的song_id=… 是从Harmony 到song 的ForeignKey，所以可能会略有不同。