from rest_framework import serializers

from .models import Requirement, Submission


class RequirementSerializer(serializers.ModelSerializer):
    submission_count = serializers.IntegerField(read_only=True)

    class Meta:
        model = Requirement
        fields = ["id", "title", "content", "spec", "llm_config", "created_at", "submission_count"]
        read_only_fields = ["id", "spec", "llm_config", "created_at"]


class SubmissionSerializer(serializers.ModelSerializer):
    class Meta:
        model = Submission
        fields = ["id", "requirement", "name", "file", "status", "report",
                  "submitted_by", "created_at"]
        read_only_fields = ["id", "status", "report", "submitted_by", "created_at"]