def create_idle_task(self, task_id: str, task_struct: dict):
return self.backend.write_task_info(task_id, TaskInfo.create({
'task_id': task_id,
'structure': task_struct
}))
def write_task_info(self, task_id: str, task_info: TaskInfo):
return self.tasks_db.put(task_id, task_info.to_json())
def list_tasks(self, with_info: bool = False) -> 'Iterator[Tuple[str, Optional[TaskInfo]]]':
for task_id, task_info in self.tasks_db.iterate_all(include_value=with_info):
yield task_id, TaskInfo.create(task_info) if task_info else None
def read_task_info(self, task_id: str) -> TaskInfo:
return TaskInfo.create(self.tasks_db.get(task_id))
