Просмотр исходного кода

恢复任务增加查询状态

xueyiming 2 дней назад
Родитель
Сommit
6e5ca915ae
1 измененных файлов с 4 добавлено и 1 удалено
  1. 4 1
      pqai_agent_server/task_server.py

+ 4 - 1
pqai_agent_server/task_server.py

@@ -190,7 +190,10 @@ class TaskManager:
     def get_in_progress_task(self):
         """获取执行中任务"""
         with self.session_maker() as session:
-            return session.query(AgentTestTask).filter(AgentTestTask.status == TestTaskStatus.IN_PROGRESS.value).all()
+            return session.query(AgentTestTask).filter(AgentTestTask.status.in_([
+                TestTaskStatus.NOT_STARTED.value,
+                TestTaskStatus.IN_PROGRESS.value
+            ])).all()
 
     def get_creating_task(self):
         """获取执行中任务"""