from apps.am.models import Access, Area
from apps.utils.serializers import CustomModelSerializer
from apps.system.serializers import PostSimpleSerializer


class AreaSimpleSerializer(CustomModelSerializer):
    class Meta:
        model = Area
        fields = ['id', 'name', 'level', 'cate']


class AreaSerializer(CustomModelSerializer):
    class Meta:
        model = Area
        fields = '__all__'


class AreaCreateUpdateSerializer(CustomModelSerializer):
    class Meta:
        model = Area
        fields = ['name', 'level', 'number', 'visitor_yes', 'remployee_yes', 'employee_yes',
                  'belong_dept', 'count_people_min', 'count_people_max', 'count_people', 'cate']


class AccessCreateSerializer(CustomModelSerializer):
    class Meta:
        model = Access
        fields = ['type', 'area', 'post']


class AccessSerializer(CustomModelSerializer):
    area_ = AreaSimpleSerializer(source='area', read_only=True)
    post_ = PostSimpleSerializer(source='post', read_only=True)

    class Meta:
        model = Access
        fields = '__all__'