refactor: simplify assigned agent retrieval logic in tasks.py
This commit is contained in:
@@ -878,8 +878,9 @@ async def update_task(
|
|||||||
elif updates["status"] == "in_progress":
|
elif updates["status"] == "in_progress":
|
||||||
task.in_progress_at = utcnow()
|
task.in_progress_at = utcnow()
|
||||||
|
|
||||||
if "assigned_agent_id" in updates and updates["assigned_agent_id"]:
|
assigned_agent_id = updates.get("assigned_agent_id")
|
||||||
agent = await Agent.objects.by_id(updates["assigned_agent_id"]).first(session)
|
if assigned_agent_id:
|
||||||
|
agent = await Agent.objects.by_id(assigned_agent_id).first(session)
|
||||||
if agent is None:
|
if agent is None:
|
||||||
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND)
|
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND)
|
||||||
if agent.board_id and task.board_id and agent.board_id != task.board_id:
|
if agent.board_id and task.board_id and agent.board_id != task.board_id:
|
||||||
|
|||||||
Reference in New Issue
Block a user