Django фильтрует пустой массив в prefetch_related

Я использовал prefetch_related и Prefetch для создания списка областей (основная категория), потребностей (подкатегория) и категорий продуктов (подкатегория) для моих продуктов.

filter_needs = Area.objects.filter(Exists(Need.objects.filter(category_need=OuterRef('pk'))))

products = Area.objects.prefetch_related(Prefetch('need_area', 
           queryset=filter_needs, to_attr='need_area__category_need__product'))
           .filter(need_area__category_need__product__isnull=False).distinct()

models.py

class Area(models.Model):
    title = models.CharField(max_length=75, blank=False)
    body = models.CharField(max_length=150, default='-', blank=False)
    publish = models.DateTimeField('publish', default=timezone.now)   

class Need(models.Model):
    title = models.CharField(max_length=75, blank=False, null=False, help_text='max 75 characters')
    body = models.CharField(max_length=150, default='-', blank=False)
    publish = models.DateTimeField(default=timezone.now)
    need_area = models.ForeignKey(Area, on_delete=models.CASCADE, related_name='need_area')

class ProductCategory(models.Model):
    title = models.CharField(max_length=400, blank=False, null=False, help_text='max 400 characters')
    body = models.TextField(default='-')
    publish = models.DateTimeField('publish', default=timezone.now)
    category_area = models.ForeignKey(Area, on_delete=models.CASCADE, related_name='category_area', null=True)
    category_need = models.ForeignKey(Need, on_delete=models.CASCADE, related_name='category_need', null=True)

class Product(models.Model):
    id = models.AutoField(primary_key=True)
    title = models.CharField(max_length=400, blank=False)
    category = models.ForeignKey(ProductCategory, on_delete = models.CASCADE, blank=True, related_name='product')

Сейчас запрос отображает все области, содержащие товар, но со всеми потребностями (даже если в потребности нет товара). Как исключить все пустые потребности?

Попробуйте это: 

from django.db.models import Prefetch


Area.objects.prefetch_related(
    Prefetch(
        "need_area"
        queryset=Need.objects.filter(
            category_need__products__isnull=False,
            category_need__products__status=1,
        ).distinct(),
    )
).filter(
    need_area__category_need__products__isnull=False, 
    need_area__category_need__products__status=1,
).distinct()

Вы должны указать тот же фильтр в пользовательской предварительной выборке, что и во внешнем фильтре.

Вернуться на верх