MaxKB/apps/knowledge/serializers/knowledge_folder.py

23 lines
692 B
Python

from rest_framework import serializers
from knowledge.models import KnowledgeFolder
class KnowledgeFolderTreeSerializer(serializers.ModelSerializer):
children = serializers.SerializerMethodField()
class Meta:
model = KnowledgeFolder
fields = ['id', 'name', 'user_id', 'workspace_id', 'parent_id', 'children']
def get_children(self, obj):
return KnowledgeFolderTreeSerializer(obj.get_children(), many=True).data
class KnowledgeFolderFlatSerializer(serializers.ModelSerializer):
children = serializers.SerializerMethodField()
class Meta:
model = KnowledgeFolder
fields = ['id', 'name', 'user_id', 'workspace_id', 'parent_id']