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