def add_log(task_id: ObjectId, task_type: str, user: dict):
task_cls = tasks_cls_for(task_type)
task = task_cls().get(task_id)
if task is None:
raise errors.NotFound()
request_json = request.get_json()
task_cls.update_logs(
task_id,
worker_log=request_json.get("worker_log"),
installer_log=request_json.get("installer_log"),
uploader_log=request_json.get("uploader_log"),
downloader_log=request_json.get("downloader_log"),
wipe_log=request_json.get("wipe_log"),
writer_log=request_json.get("writer_log"),
)
# update ACK
Acknowlegments.busy_update(
username=user["username"],
worker_type=task_type,
slot=request.args.get("slot"),
task_id=task_id,
)
return jsonify({"_id": task_id})
def register_task(task_id: ObjectId, task_type: str, user: dict):
task_cls = tasks_cls_for(task_type)
task = task_cls().find({"_id": task_id, "status": task_cls.pending})
if task is None:
raise errors.NotFound()
task_cls.register(task_id, user)
# update ACK
Acknowlegments.busy_update(
username=user["username"],
worker_type=task_type,
slot=request.args.get("slot"),
task_id=task_id,
)
return jsonify({"_id": task_id})
