task model

ObjectManager.py

@@ -43,15 +43,15 @@ class TaskManager(object):
         self.tasks = {}
         # TOOD: 保持一个后台线程，定时检查任务的状态，并将任务的状态更新到数据中
 
-    def create(self, task: dict):
-        task["id"] = str(uuid.uuid4())
-        self.tasks[task["id"]] = task
+    def create(self, task: Task):
+        task.id = str(uuid.uuid4())
+        self.tasks[task.id] = task
         return task
 
     def delete(self, task_id: str):
         del self.tasks[task_id]
 
-    def update(self, task_id, task: dict):
+    def update(self, task_id, task: Task):
         self.tasks[task_id] = task
 
     def query(self, task_id: str):

main.py

@@ -92,10 +92,8 @@ async def get_tasks():
 
 
 @app.post("/tasks", response_model=Task)
-async def create_task():
-    return task_manager.add({
-        "name": "test",
-    })
+async def create_task(task: Task):
+    return task_manager.add(task)
 
 # 维护一个任务队列, 任务队列中的任务会被分发给worker节点
 # 任务状态变化时通知对应的客户端