From 69af3ba908c3c3e1c838da3e95ff15f967086af0 Mon Sep 17 00:00:00 2001 From: simon987 Date: Sat, 6 Apr 2019 10:36:40 -0400 Subject: [PATCH] Bulk submit tasks --- src/tt_drone/api.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/src/tt_drone/api.py b/src/tt_drone/api.py index cb93816..895c514 100644 --- a/src/tt_drone/api.py +++ b/src/tt_drone/api.py @@ -80,6 +80,9 @@ class Worker: return self._api.submit_task(self, project, recipe, priority, max_assign_time, hash64, unique_str, verification_count, max_retries) + def bulk_submit_task(self, project, reqs): + return self._api.bulk_submit_task(self, project, reqs) + def release_task(self, task_id: int, result: int, verification): return self._api.release_task(self, task_id, result, verification) @@ -168,6 +171,20 @@ class TaskTrackerApi: "max_retries": max_retries, }, worker) + def bulk_submit_task(self, worker: Worker, project, tasks): + return self._http_post("/task/bulk_submit", { + "requests": { + "project": project, + "recipe": req.recipe, + "priority": req.priority, + "max_assign_time": req.max_assign_time, + "hash_u64": req.hash64, + "unique_str": req.unique_str, + "verification_count": req.verification_count, + "max_retries": req.max_retries, + } for req in tasks + }, worker) + def log(self, worker: Worker, level: int, message: str, timestamp: int, scope: str): if level == LOG_TRACE: return self._http_post("/log/trace",