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)