from pydantic import BaseModel
from typing import List, Any, Annotated
from solverforge_legacy.solver.score import HardMediumSoftScore
from .json_serialization import ScoreSerializer


class MatchAnalysisDTO(BaseModel):
    name: str
    score: Annotated[HardMediumSoftScore, ScoreSerializer]
    justification: Any


class ConstraintAnalysisDTO(BaseModel):
    name: str
    weight: Annotated[HardMediumSoftScore, ScoreSerializer]
    score: Annotated[HardMediumSoftScore, ScoreSerializer]
    matches: List[MatchAnalysisDTO]