def _update_run():
request_message = _get_request_message(UpdateRun())
run_id = request_message.run_id or request_message.run_uuid
updated_info = _get_store().update_run_info(run_id, request_message.status,
request_message.end_time)
response_message = UpdateRun.Response(run_info=updated_info.to_proto())
response = Response(mimetype='application/json')
response.set_data(message_to_json(response_message))
return response
def _update_run():
request_message = _get_request_message(UpdateRun())
updated_info = _get_store().update_run_info(request_message.run_uuid,
request_message.status,
request_message.end_time)
response_message = UpdateRun.Response(run_info=updated_info.to_proto())
response = Response(mimetype='application/json')
response.set_data(
MessageToJson(response_message, preserving_proto_field_name=True))
return response
def _update_run():
request_message = _get_request_message(UpdateRun())
run_id = request_message.run_id or request_message.run_uuid
run = _get_tracking_store().get_run(run_id)
run_auth_check(run)
updated_info = _get_tracking_store().update_run_info(
run_id, request_message.status, request_message.end_time)
response_message = UpdateRun.Response(run_info=updated_info.to_proto())
response = Response(mimetype="application/json")
response.set_data(message_to_json(response_message))
return response
def update_run_info(self, run_id, run_status, end_time):
""" Updates the metadata of the specified run. """
req_body = message_to_json(
UpdateRun(run_uuid=run_id, run_id=run_id, status=run_status, end_time=end_time)
)
response_proto = self._call_endpoint(UpdateRun, req_body)
return RunInfo.from_proto(response_proto.run_info)
