from typing import List, Optional

from pydantic import BaseModel

from ielts_be.configs.constants import LevelExerciseType


class LevelExercises(BaseModel):
    type: LevelExerciseType
    quantity: int
    text_size: Optional[int] = None
    sa_qty: Optional[int] = None
    mc_qty: Optional[int] = None
    topic: Optional[str] = None

class LevelExercisesDTO(BaseModel):
    exercises: List[LevelExercises]
    difficulty: Optional[str] = None