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']


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']


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__'