models/task.py

"""Модель задачи"""

from typing import List, Dict, Any
from dataclasses import dataclass, asdict


@dataclass
class Task:
    """Модель задачи по теории вероятности"""

    type: str
    question: str
    answer: float
    answer_fraction: str
    solution: str
    steps: List[str]
    complexity: str = 'medium'

    def to_dict(self) -> Dict[str, Any]:
        """
        Конвертация в словарь

        Возвращает:
            dict: словарь с данными задачи
        """
        return asdict(self)