update utils module

scripts/quiz/utils.py

@@ -1,13 +1,42 @@
+import pandas
+import random
+from scipy.interpolate import interp1d
 from typing import Literal
 
-def get_next_level(cur_level, levep_up: Literal[True, False] = True):
-    if levep_up:
-        if cur_level == "easy":
-            return "medium"
+def get_next_difficulty(difficulty, do_increase: Literal[True, False] = True):
+    if do_increase:
+        if difficulty >= 0.95:
+            next_difficulty = round(random.uniform(0.95, 0.99), 2)
         else:
-            return "hard"
+            next_difficulty = round(random.uniform(difficulty + 0.01, difficulty + 0.05), 2)
     else:
-        if cur_level == "medium":
-            return "easy"
+        if difficulty <= 0.05:
+            next_difficulty = round(random.uniform(0.01, 0.05), 2)
         else:
-            return "medium"
+            next_difficulty = round(random.uniform(difficulty - 0.05, difficulty - 0.01), 2)
+
+    return next_difficulty
+
+
+def generate_start_step(difficulty: float, path_to_csv_file: str = "data.csv"):
+    """generate start and step values interpolating results to function built from data from file"""
+    df = pandas.read_csv(path_to_csv_file, delimiter=',', header=0, names=['difficulty', 'start'])
+    all_rows = df.loc[:]
+
+    difficulties = [row_data['difficulty'] for _, row_data in all_rows.iterrows()]
+    starts = [row_data['start'] for _, row_data in all_rows.iterrows()]
+
+    interp_start_func = interp1d(difficulties, starts)
+    generated_start = round(float(interp_start_func(difficulty)))
+    if difficulty <= 0.3:
+        step = 1
+    elif difficulty > 0.6:
+        step = 10
+    else:
+        step = 5
+    return (generated_start, step)
+
+
+def convert_sequence_to_string(start, step, sep=", "):
+    stop = start + 3 * step
+    return sep.join([str(num) for num in range(start, stop, step)])