def _wrap_response(response_message):
if response_message is None:
return Response(return_code=str(SUCCESS),
return_msg=ReturnCode.Name(SUCCESS).lower())
else:
return Response(return_code=str(SUCCESS),
return_msg=ReturnCode.Name(SUCCESS).lower(),
data=MessageToJson(response_message,
preserving_proto_field_name=True))
def _wrap_update_response(uuid):
if uuid != UPDATE_FAIL:
return Response(return_code=str(SUCCESS),
return_msg=ReturnCode.Name(SUCCESS).lower(),
data=str(uuid))
else:
return Response(return_code=str(INTERNAL_ERROR),
return_msg=ReturnCode.Name(INTERNAL_ERROR).lower(),
data=None)
def _wrap_delete_response(status):
if status == Status.OK:
return Response(return_code=str(SUCCESS),
return_msg=ReturnCode.Name(SUCCESS).lower(),
data=None)
else:
return Response(return_code=str(INTERNAL_ERROR),
return_msg=ReturnCode.Name(INTERNAL_ERROR).lower(),
data=None)
def deleteMetricSummary(self, request, context):
uuid = request.uuid
try:
self.store.delete_metric_summary(uuid)
return Response(return_code=str(ReturnCode.SUCCESS),
return_msg='',
data='')
except Exception as e:
return Response(return_code=str(ReturnCode.INTERNAL_ERROR),
return_msg=str(e),
data='')
def _wrap_meta_response(data):
if data is not None:
return Response(return_code=str(SUCCESS),
return_msg=ReturnCode.Name(SUCCESS).lower(),
data=MessageToJson(data,
preserving_proto_field_name=True))
else:
return Response(
return_code=str(RESOURCE_DOES_NOT_EXIST),
return_msg=ReturnCode.Name(RESOURCE_DOES_NOT_EXIST).lower(),
data=None)
def _warp_job_list_response(job_list):
if job_list is not None:
job_proto_list = MetaToProto.job_meta_list_to_proto(job_list)
return Response(return_code=str(SUCCESS),
return_msg=ReturnCode.Name(SUCCESS).lower(),
data=MessageToJson(JobListProto(jobs=job_proto_list),
preserving_proto_field_name=True))
else:
return Response(
return_code=str(RESOURCE_DOES_NOT_EXIST),
return_msg=ReturnCode.Name(RESOURCE_DOES_NOT_EXIST).lower(),
data=None)
def _warp_artifact_list_response(artifact_list):
if artifact_list is not None:
artifact_proto_list = MetaToProto.artifact_meta_list_to_proto(
artifact_list)
return Response(return_code=str(SUCCESS),
return_msg=ReturnCode.Name(SUCCESS).lower(),
data=MessageToJson(
ArtifactListProto(artifacts=artifact_proto_list),
preserving_proto_field_name=True))
else:
return Response(
return_code=str(RESOURCE_DOES_NOT_EXIST),
return_msg=ReturnCode.Name(RESOURCE_DOES_NOT_EXIST).lower(),
data=None)
def _warp_workflow_execution_list_response(workflow_execution_list):
if workflow_execution_list is not None:
workflow_execution_proto_list = MetaToProto.workflow_execution_meta_list_to_proto(
workflow_execution_list)
return Response(return_code=str(SUCCESS),
return_msg=ReturnCode.Name(SUCCESS).lower(),
data=MessageToJson(WorkFlowExecutionListProto(
workflow_executions=workflow_execution_proto_list),
preserving_proto_field_name=True))
else:
return Response(
return_code=str(RESOURCE_DOES_NOT_EXIST),
return_msg=ReturnCode.Name(RESOURCE_DOES_NOT_EXIST).lower(),
data=None)
def _warp_model_version_relation_list_response(model_version_relation_list):
if model_version_relation_list is not None:
model_version_proto_list = MetaToProto.model_version_relation_meta_list_to_proto(
model_version_relation_list)
return Response(return_code=str(SUCCESS),
return_msg=ReturnCode.Name(SUCCESS).lower(),
data=MessageToJson(ModelVersionRelationListProto(
model_versions=model_version_proto_list),
preserving_proto_field_name=True))
else:
return Response(
return_code=str(RESOURCE_DOES_NOT_EXIST),
return_msg=ReturnCode.Name(RESOURCE_DOES_NOT_EXIST).lower(),
data=None)
def _warp_example_list_response(example_meta_list):
if example_meta_list is not None:
example_proto_list = MetaToProto.example_meta_list_to_proto(
example_meta_list)
return Response(return_code=str(SUCCESS),
return_msg=ReturnCode.Name(SUCCESS).lower(),
data=MessageToJson(
ExampleListProto(examples=example_proto_list),
preserving_proto_field_name=True))
else:
return Response(
return_code=str(RESOURCE_DOES_NOT_EXIST),
return_msg=ReturnCode.Name(RESOURCE_DOES_NOT_EXIST).lower(),
data=None)
def wrapper(*args, **kwargs):
try:
return func(*args, **kwargs)
except AIFlowException as e:
return Response(return_code=str(e.error_code),
return_msg=e.error_msg)
