models/settings.py

"""Модель настроек генерации"""

from typing import List
from dataclasses import dataclass


@dataclass
class GenerationSettings:
    """Настройки генерации задач"""

    count: int = 1
    complexity: str = 'medium'
    task_types: List[int] = None

    def __post_init__(self):
        if self.task_types is None:
            self.task_types = []