from rest_framework.serializers import ModelSerializer

from .models import Material, Process, ProductProcess, Step
from apps.system.serializers import FileSimpleSerializer


class MaterialSerializer(ModelSerializer):
    class Meta:
        model = Material
        fields = '__all__'

class MaterialSimpleSerializer(ModelSerializer):
    class Meta:
        model = Material
        fields = ['id', 'name', 'number']

class ProcessSerializer(ModelSerializer):
    instruction_ = FileSimpleSerializer(source='instruction', read_only=True)
    class Meta:
        model = Process
        fields = '__all__'

class ProcessSimpleSerializer(ModelSerializer):
    class Meta:
        model = Process
        fields = ['id', 'name', 'number']

class StepSerializer(ModelSerializer):
    class Meta:
        model = Step
        fields = '__all__'

class ProductProcessListSerializer(ModelSerializer):
    process_ = ProcessSimpleSerializer(source='process', read_only=True)
    product_ = MaterialSimpleSerializer(source='product', read_only=True)
    class Meta:
        model = ProductProcess
        fields = '__all__'

class ProductProcessUpdateSerializer(ModelSerializer):
    class Meta:
        model = ProductProcess
        fields = ['sort']