Python/Django递归模型到树结构字典

Question

我有一个 Django 模型class Category(models.Model):    parent = models.ForeignKey('Category', null=False, blank=False)    category_name = models.CharField(max_length=55,null=False, blank=False)    status = model.BooleanField(default=True)我想创建返回树结构字典的递归。例如[    {        'category_name': 'vehical',        'id': 1,        'children' : [            {                'category_name': 'cars',                'id': 11,                'children': [                    {                        'category_name': 'sport cars',                        'id': 20                    }                               ]                           }               ]    }]>>> category = Category.objects.all()>>> get_tree(category)

Answer 1

我认为您需要将您的ForeignKey关系更改为self, 以引用自身。此外，为了简化实现，让我们添加related_name='children'( doc ) 并将其设为 nullable( null=True) 以将其表示为根节点。

class Category(models.Model):

    parent = models.ForeignKey('self', null=True, blank=True, related_name='children')

    category_name = models.CharField(max_length=55,null=False, blank=False)

    status = models.BooleanField(default=True)

然后我们可以做如下实现：

from django.forms.models import model_to_dict

def get_tree(category):

    tree = model_to_dict(category, fields=['category_name', 'status', 'id'])

    if category.children.all().exists():

       children = list()

       for child in category.children.all():

          children.append(get_tree(child))

       tree['children'] = children

    return tree

final_tree = list()

for category in Category.objects.filter(parent__isnull=True):

    final_tree.append(get_tree(category))

print(final_tree)