from apps.opm.models import Operation, OptCate
from apps.utils.serializers import CustomModelSerializer
from apps.wf.serializers import WorkflowSimpleSerializer
from apps.utils.constants import EXCLUDE_FIELDS


class OptCateCreateUpdateSerializer(CustomModelSerializer):
    class Meta:
        model = OptCate
        fields = ['code', 'name', 'template_export', 'workflow']


class OptCateSerializer(CustomModelSerializer):
    workflow_ = WorkflowSimpleSerializer(source='workflow', read_only=True)

    class Meta:
        model = OptCate
        fields = '__all__'


class OperationCreateUpdateSerializer(CustomModelSerializer):
    class Meta:
        model = Operation
        fields = EXCLUDE_FIELDS + ['number']


class OperationSerializer(CustomModelSerializer):
    class Meta:
        model = Operation
        fields = "__all__"