ENCOA-311

ielts_be/utils/pick_difficulty.py

@@ -0,0 +1,14 @@
+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)