def ExecutePipeline(self, request, context):
sessioncontext = request.context
if not sessioncontext.session_id in self.sessions:
yield core_pb2.PipelineExecuteResult(\
response_info=core_pb2.Response(\
status=core_pb2.Status(\
code=core_pb2.FAILED_PRECONDITION,
details="Unknown session id: %s" % sessioncontext.session_id)))
return
pipeline_id = request.pipeline_id
logger.info("Got ExecutePipeline request, session=%s",
sessioncontext.session_id)
time.sleep(1)
yield core_pb2.PipelineExecuteResult(
response_info=core_pb2.Response(
status=core_pb2.Status(code=core_pb2.OK), ),
progress_info=core_pb2.RUNNING,
pipeline_id=pipeline_id,
)
time.sleep(1)
# try to create a legit file uri
file_uri_dict = get_predict_file_info_dict('CLASSIFICATION')
file_uri_dict2 = get_predict_file_info_dict('REGRESSION')
yield core_pb2.PipelineExecuteResult(response_info=core_pb2.Response(
status=core_pb2.Status(code=core_pb2.OK), ),
progress_info=core_pb2.COMPLETED,
pipeline_id=pipeline_id,
result_uri=file_uri_dict.get(
TEST_KEY_FILE_URI,
'no file uri'))
def GetExecutePipelineResults(self, request, context):
sessioncontext = request.context
if not sessioncontext.session_id in self.sessions:
yield core_pb2.PipelineExecuteResult(\
response_info=core_pb2.Response(\
status=core_pb2.Status(\
code=core_pb2.FAILED_PRECONDITION,
details="Unknown session id: %s" % sessioncontext.session_id)))
return
pipeline_ids = request.pipeline_ids
logger.info("Got GetExecutePipelineResults request, session=%s",
sessioncontext.session_id)
file_uri_dict = get_predict_file_info_dict()
for pipeline_id in pipeline_ids:
time.sleep(1)
progress_info = random.choice(\
[core_pb2.UPDATED,
core_pb2.RUNNING,
core_pb2.COMPLETED,])
yield core_pb2.PipelineExecuteResult(
response_info=core_pb2.Response(
status=core_pb2.Status(code=core_pb2.OK), ),
progress_info=progress_info,
pipeline_id=pipeline_id,
result_uri=file_uri_dict.get(TEST_KEY_FILE_URI, 'no file uri'))
def pipeline_exec_result():
req = core_pb2.PipelineExecuteResult()
req.response_info.status.code = core_pb2.OK
req.response_info.status.details = "(static test response)"
req.progress_info = core_pb2.COMPLETED
req.pipeline_id = 'pipline_1'
req.result_uris.append('file://results/output/file_001.txt')
req.result_uris.append('file://results/output/file_002.txt')
print('-' * 40)
content = MessageToJson(req, including_default_value_fields=True)
print(content)
print('-' * 40)
content = content.replace('pipelineIds',
'pipeline_ids').replace('sessionId',
'session_id')
print(content)
print('-' * 40)
json_parse(content, core_pb2.PipelineExecuteResult)
print('-' * 40)
"""
def ExecutePipeline(self, request, context):
"Placeholder."
logging.info("Message received: ExecutePipelines")
yield core_pb2.PipelineExecuteResult(
response_info=core_pb2.Response(
status=core_pb2.Status(code=core_pb2.OK), ),
progress_info=core_pb2.COMPLETED,
pipeline_id=request.pipeline_id,
result_uri="file://TODO, return valid pipeline URI")
def StartExecutingPipeline(self, pipeline):
response = self._create_response("Pipeline Started Running", "OK")
progress = self._create_progress("RUNNING")
result = core.PipelineExecuteResult(
response_info = response,
progress_info = progress,
pipeline_id = pipeline.id
)
self.session.cache_test_result(pipeline, result)
return result
def ExecutedPipeline(self, pipeline):
result_uri = None
if pipeline.test_result is not None:
resultfile = self.session.save_prediction_file(pipeline.test_result)
result_uri = self._create_uri_from_path(resultfile)
response = self._create_response("Pipeline Completed", "OK")
else:
response = self._create_response("Pipeline Failed to run", "INTERNAL")
progress = self._create_progress("COMPLETED")
result = core.PipelineExecuteResult(
response_info = response,
progress_info = progress,
pipeline_id = pipeline.id,
result_uri = result_uri
)
self.session.cache_test_result(pipeline, result)
return result