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


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


class AreaCreateUpdateSerializer(CustomModelSerializer):
    class Meta:
        model = Area
        fields = ['name', 'level', 'sort_srt', 'visitor_yes', 'remployee_yes', 'employee_yes', 'belong_dept']


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