23 lines
722 B
Python
23 lines
722 B
Python
from rest_framework import serializers
|
|
from .models import Organization
|
|
|
|
|
|
class OrganizationSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = Organization
|
|
fields = ['id', 'name', 'parent', 'logo', 'description', 'email', 'is_active']
|
|
|
|
|
|
class OrganizationTreeSerializer(serializers.ModelSerializer):
|
|
"""带子公司列表，用于门户展示"""
|
|
children = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = Organization
|
|
fields = ['id', 'name', 'logo', 'description', 'email', 'children']
|
|
|
|
def get_children(self, obj):
|
|
return OrganizationSerializer(
|
|
obj.children.filter(is_active=True), many=True, context=self.context
|
|
).data
|