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 _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 deleteMetricMeta(self, request, context):
metric_name = request.metric_name
try:
self.store.delete_metric_meta(metric_name)
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 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 _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 _wrap_workflow_list_response(workflow_list):
if workflow_list is not None:
workflow_proto_list = MetaToProto.workflow_meta_list_to_proto(
workflow_list)
return Response(return_code=str(SUCCESS),
return_msg=ReturnCode.Name(SUCCESS).lower(),
data=MessageToJson(
WorkflowListProto(workflows=workflow_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 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)