Django：使用包含文件和子类别的类别创建树视图

Question

从这个答案中，我能够创建类别项目的树视图。

但我需要进行 api 调用才能以相同的方式获取文件。 我怎样才能做到这一点？

模型.py

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

    class MPTTMeta:
        order_insertion_by = ['name']

class Document(models.Model):
    name = models.CharField(max_length = 128, blank = True)
    description = models.TextField(blank = True, null = True)
    file = models.FileField(null = True, blank = True, upload_to='documents/', validators=[FileExtensionValidator(allowed_extensions=['pdf'])])
    category = models.ForeignKey(Category, null = True, blank = True, on_delete = models.SET_NULL)

    class Meta:
        verbose_name = 'Document'
        verbose_name_plural = 'Documents'

Answer 1

只需将document_set添加到CategoryTreeSerializer中的属性fields

像那样

class CategoryTreeSerializer(serializers.ModelSerializer):
    children = serializers.SerializerMethodField()

    class Meta:
        fields = ['id', 'name', 'children', 'document_set']