ship task setting from request body

1 files changed, 6 insertions, 6 deletions

diff --git a/apirouters/tasks.py b/apirouters/tasks.py
index 293ab52..6cc9a25 100644
--- a/apirouters/tasks.py
+++ b/apirouters/tasks.py
@@ -27,15 +27,15 @@ async def get_tasks(ship_symbol: str, agent: Annotated[Agent, Depends(auth_agent
     return {"error": "Unknown ship symbol."}
 
 
-@router.post("/task/{ship_symbol}/set/{task_type}")
-async def get_tasks(ship_symbol: str, task: str, agent: Annotated[Agent, Depends(auth_agent)]):
-    if task not in task_type.task_types:
-        return {"error": "Invalid task."}
+class SetTaskBody(BaseModel):
+    task: task_types.task_type
 
+
+@router.post("/task/{ship_symbol}/set")
+async def get_tasks(ship_symbol: str, set_task: SetTaskBody, agent: Annotated[Agent, Depends(auth_agent)]):
     for current_ship in agent.ships.values():
         if current_ship.symbol == ship_symbol:
-            current_ship.set_task(task)
+            current_ship.set_task(set_task.task)
             return current_ship.get_data()
 
     return {"error": "Unknown ship symbol."}
-