from rest_framework import serializers
from .models import GroupBusinessTarget, EmployeePerformanceTarget, TargetAudit


class GroupBusinessTargetSerializer(serializers.ModelSerializer):
    class Meta:
        model = GroupBusinessTarget
        fields = '__all__'


class EmployeePerformanceTargetSerializer(serializers.ModelSerializer):
    class Meta:
        model = EmployeePerformanceTarget
        fields = '__all__'


class TargetAuditSerializer(serializers.ModelSerializer):
    modification_date = serializers.DateTimeField(format='%Y-%m-%d %H:%M')
    modified_obj = serializers.SerializerMethodField()

    class Meta:
        model = TargetAudit
        fields = ('modified_obj', 'modified_field', 'old_value', 'new_value', 'modification_date', 'modified_by')

    def get_modified_obj(self, obj):
        target = obj.target_id
        return f"{target.primary_department}{target.year}年度目标"


class EmployeeTargetAuditSerializer(serializers.ModelSerializer):
    modification_date = serializers.DateTimeField(format='%Y-%m-%d %H:%M')
    modified_obj = serializers.SerializerMethodField()

    class Meta:
        model = TargetAudit
        fields = ('modified_obj', 'modified_field', 'old_value', 'new_value', 'modification_date', 'modified_by')

    def get_modified_obj(self, obj):
        target = obj.target_id
        return f"{target.department}{target.year}年度目标"