Create tasks.py

bot/core/tasks.py

@@ -0,0 +1,32 @@
+# PATH: bot/core/tasks.py
+from dataclasses import dataclass
+from typing import Optional
+import time
+
+@dataclass
+class Task:
+    id: str
+    uid: int
+    status: str = "queued"
+    detail: str = ""
+    created_at: float = time.time()
+    yt_url: Optional[str] = None
+
+_TASKS: dict[str, Task] = {}
+
+def create_task(task_id: str, uid: int) -> Task:
+    t = Task(id=task_id, uid=uid)
+    _TASKS[task_id] = t
+    return t
+
+def get_task(task_id: str) -> Optional[Task]:
+    return _TASKS.get(task_id)
+
+def set_task(task_id: str, status: str, detail: str = "", yt_url: str | None = None) -> None:
+    t = _TASKS.get(task_id)
+    if not t:
+        return
+    t.status = status
+    t.detail = detail
+    if yt_url:
+        t.yt_url = yt_url