Как получить категории, если в их потомках есть товар?

Question

[Kamo Petrosyan]

Собственно структура

Категории
ID   Название       Родитель
1    Категория1    null
2    Категория2    1

Товары
ID     Название     Категория
1       Товар 1        2

Соответственно нужно получить все категории, если в них и/или их потомках есть товар.
Подойдет подсказка и на Django ORM и на чистом SQL.
Спасибо!

Answer 1

[Сергей Горностаев]

Во-первых, для древовидных структур всегда используйте MPTT. Иначе обход дерева будет долгим, требовательным к ресурсам и зависимым от количества элементов и уровня их вложенности.

Во-вторых, ORM с этой задачей не справится. А на чистом SQL можно.
Предположим, что у нас есть приложение "catalog" и модели в нём такие:

class Category(MPTTModel):
    parent = TreeForeignKey('self', verbose_name=u'Родитель', null=True, blank=True, related_name='children')
    name = models.CharField(u'Название', max_length=100)

class Product(models.Model):
    category = TreeForeignKey(Category, verbose_name=u'Категория')
    ...

Тогда SQL-запрос будет таким:

SELECT name FROM catalog_category AS cc
INNER JOIN LATERAL (
  SELECT cc.id AS id, SUM(products) AS p_count FROM (
    SELECT cc.id AS parent_id, category_id, COUNT(id) AS products
    FROM catalog_product
    WHERE category_id IN (
      SELECT id FROM catalog_category
      WHERE lft <= cc.rght AND lft >= cc.lft AND tree_id = cc.tree_id)
      GROUP BY category_id
    ) AS sub_cс
  GROUP BY parent_id
) AS cp
USING(id) WHERE cp.p_count > 0;

Естественно, работать будет на взрослых СУБД с поддержкой современных стандартов SQL. Например, на PostgreSQL.

Comments

[Kamo Petrosyan]

Да, я использую mptt.
В моем случае это выглядит так:

class Category(MPTTModel):
parent = TreeForeignKey('self', verbose_name=u'Родитель', null=True, blank=True, related_name='children')
name = models.CharField(u'Название', max_length=100)

class Product(models.Model):
category = ManyToManyField(Category, verbose_name=u'Категория')
...

Очень сложная структура, в которой один товар может быть в нескольких категориях

[Сергей Горностаев]

Haikson: должна сработать замена в запросе "FROM catalog_product" на "FROM catalog_product_category".

[Kamo Petrosyan]

Отлично сработало. Спасибо.

Answer 2

[Alexander]

Один из вариантов, взять модель товаров, сгруппировать их по категориям = список категорий.
С родителями эту уже строить деревья. Выше ты получаешь все категории у которых есть товары, дальше смотришь есть ли родитель и если есть то лезешь по дереву вверх (если структура изначально не древовидная, а глубина категорий больше 2х).

Comments

[Kamo Petrosyan]

Структура категорий изначально древовидная. Можно конечно получить все категории с товарами, после для них просто получить root объекты. Как вариант. Но насколько это будет ресурсоемко, если категорий, нуууу так, несколько десятков или сотен?

[Alexander]

Haikson: Тебе надо по сути 1 запрос к товарам. Получить поле "Категория_ИД", сгруппировав все товары по "Категория_ИД". Всё, у тебя есть список ИД категорий которые имеют товары. Дальше думаешь сам. Нагрузка это уже отдельная тема. Но тебе в любом случае надо тянуть товары, проще заходить сразу со стороны товаров. Если тебе нужны будут потом другие поля из модели категоирй, то список ИД ты имеешь, делаешь запрос через IN к категориям и получаешь список их.