import random
from typing import Optional, List

from ielts_be.configs.constants import EducationalContent


def pick_difficulty(difficulty: Optional[str], difficulties: Optional[List[str]]) -> str:
    if difficulty:
        return difficulty

    if difficulties:
        return random.choice(difficulties)

    return random.choice(EducationalContent.DIFFICULTIES)