VarshaChukka/SQL_Debugging_REPL_Environment

1def get_task(task_id):
2
3    tasks = [
4        # EASY
5        {
6            "problem": "Get all users",
7            "schema": "users(id, name)",
8            "setup": [
9                "CREATE TABLE users(id INT, name TEXT)",
10                "INSERT INTO users VALUES (1,'A'), (2,'B')"
11            ],
12            "expected": [(1,'A'), (2,'B')]
13        },
14
15        # MEDIUM
16        {
17            "problem": "Get users with id > 1",
18            "schema": "users(id, name)",
19            "setup": [
20                "CREATE TABLE users(id INT, name TEXT)",
21                "INSERT INTO users VALUES (1,'A'), (2,'B'), (3,'C')"
22            ],
23            "expected": [(2,'B'), (3,'C')]
24        },
25
26        # HARD
27        {
28            "problem": "Join users and orders",
29            "schema": "users(id, name), orders(user_id, amount)",
30            "setup": [
31                "CREATE TABLE users(id INT, name TEXT)",
32                "CREATE TABLE orders(user_id INT, amount INT)",
33                "INSERT INTO users VALUES (1,'A'), (2,'B')",
34                "INSERT INTO orders VALUES (1,100), (2,200)"
35            ],
36            "expected": [(1,'A',100), (2,'B',200)]
37        }
38    ]
39
40    # ✅ SAFETY CHECK (IMPORTANT)
41    if task_id < 0 or task_id >= len(tasks):
42        raise ValueError("Invalid task_id")
43
44    return tasks[task_id]