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)