from apps.utils.serializers import CustomModelSerializer
from .models import Work
from rest_framework import serializers
from apps.system.serializers import FileSerializer

class WorkSerializer(CustomModelSerializer):
    create_by_name = serializers.CharField(source='create_by.name', read_only=True)
    belong_dept_name = serializers.CharField(source='belong_dept.name', read_only=True)
    dq_file1_ = FileSerializer(source='dq_file1', read_only=True)
    dq_file2_ = FileSerializer(source='dq_file2', read_only=True)
    dq_file3_ = FileSerializer(source='dq_file3', read_only=True)
    dq_file4_ = FileSerializer(source='dq_file4', read_only=True)
    dq_file5_ = FileSerializer(source='dq_file5', read_only=True)
    dq_file6_ = FileSerializer(source='dq_file6', read_only=True)
    dh_file1_ = FileSerializer(source='dh_file1', read_only=True)
    dh_file2_ = FileSerializer(source='dh_file2', read_only=True)
    dh_file3_ = FileSerializer(source='dh_file3', read_only=True)
    dh_file4_ = FileSerializer(source='dh_file4', read_only=True)
    dh_file5_ = FileSerializer(source='dh_file5', read_only=True)
    dh_file6_ = FileSerializer(source='dh_file6', read_only=True)
    class Meta:
        model = Work
        exclude = ['fingerprint_dq', 'fingerprint_dh']

class WorkCreateSerializer(CustomModelSerializer):
    class Meta:
        model = Work
        fields = ["name", "type", "description"]


class WorkDqCalSerializer(CustomModelSerializer):
    class Meta:
        model = Work
        fields = ["dq_file1", "dq_file2", "dq_file3", "dq_file4", "dq_file5", "dq_file6"]


class WorkDhCalSerializer(CustomModelSerializer):
    class Meta:
        model = Work
        fields = ["dh_file1", "dh_file2", "dh_file3", "dh_file4", "dh_file5", "dh_file6"]