Django дерево категорий как ограничить вложенность?

Question

[Animkim]

class Category(models.Model):
    parent = models.ForeignKey('self', null=True, blank=True, related_name='children')
    title = models.CharField(max_length=50)

Есть такая модель, заранее известно что максимальная вложенность не больше трех.

Answer 1

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

Во-первых, использовать MPTT. Во-вторых, переопределить метод save() и проверять в нём атрибут level родителя.

class Category(mptt.models.MPTTModel):
    parent = mptt.models.TreeForeignKey('self', null=True, blank=True, related_name='children')
    title = models.CharField(max_length=50)

    def save(self, *args, **kwargs):
        if parent.level == 3:
            raise ValueError(u'Достигнута максимальная вложенность!')
        super(Category, self).save(*args, **kwargs)

Comments

[Animkim]

Не использую МРТТ, хоть и знаю о нем, гугл на любой вопрос про деревья в джанго выдает первые десять ссылок на МРТТ. Я пока учусь и обхожусь без модулей, так эффективнее идет учеба.
Но спасибо за наводку, буду смотреть в метод save.

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

Animkim: тогда эффективным решением будет хранить атрибут level типа PositiveIntegerField, так же его проверять в методе save и, если он меньше 3, то вписывать в него parent.level + 1.

[Animkim]

Сергей Горностаев: Спасибо!