Django模型与mptt集成

Question

我有一个django模型，如下所示：

from django.db import models
from mptt.models import MPTTModel, TreeForeignKey

class deg_course_cat(models.Model):
    degree_code = models.CharField(max_length=24)
    specialization = models.CharField(max_length=48)
    category_level1 = models.CharField(max_length=48)
    category_level2 = models.CharField(max_length=96)
    category_level3 = models.CharField(max_length=48)
    min_credit = models.IntegerField()
    max_credit = models.IntegerField()
    primarystuff = models.CharField(max_length=24)

class deg_course_cat_mptt(MPTTModel):
    name = models.CharField(max_length=100, unique=False)
    min_credit = models.IntegerField(null=True)
    max_credit = models.IntegerField(null=True)
    parent = TreeForeignKey('self', null=True, blank=True, related_name='children')

    class MPTTMeta:
        order_insertion_by = ['name']

# Create your models here.

django模型deg_course_cat表中的数据如下所示：

​

​

我想根据模型及其数据创建mptt结构，即类似于以下结构：

​

​

我该如何处理这个问题？因为我对python有点陌生，所以如果我对实现这个目标有了一个认识，那就太好了。

我尝试过以下代码，但不确定它是否正确：

from studentapp.models import deg_course_cat, deg_course_cat_mptt

#degreeroot = deg_course_cat_mptt.objects.create(name="DegreeRoot")

for degrees in deg_course_cat.objects.values_list('degree_code', flat=True):
    degreearray = set(deg_course_cat.objects.values_list('degree_code', flat=True))
    for i in range(0,len(degreearray)):
        degree(i) = deg_course_cat_mptt.objects.create(name= degreearray(i), parent=degreeroot)

        for categories_l1 in deg_course_cat.objects.values_list('category_level1', flat=True):
        category_level1 = set(deg_course_cat.objects.values_list('category_level1', flat=True))

            for categories_l2 in deg_course_cat.objects.values_list('category_level2', flat=True):
            category_level2 = set(deg_course_cat.objects.values_list('category_level2', flat=True))

            for categories_l3 in deg_course_cat.objects.values_list('category_level3', flat=True):
            category_level3 = set(deg_course_cat.objects.values_list('category_level3', flat=True))

Answer 1

你可以做一些像this这样的事情

category_l1_array = list(set(deg_course_cat.objects.filter(degree_code=degree[i]).values_list('category_level1', flat=True)))