def add_classifications_by_type(self, type_name, uniq_attributes,
classifications):
query_param = attributes_to_params(uniq_attributes)
self.client.call_api(
EntityClient.ADD_CLASSIFICATION_BY_TYPE_AND_ATTRIBUTE.format_path(
{'type_name': type_name}), None, query_param, classifications)
def get_entity_by_attribute(self, type_name, uniq_attributes, min_ext_info=False, ignore_relationships=False):
query_params = attributes_to_params(uniq_attributes)
query_params["minExtInfo"] = min_ext_info
query_params["ignoreRelationships"] = ignore_relationships
return self.client.call_api(EntityClient.GET_ENTITY_BY_UNIQUE_ATTRIBUTE.format_path_with_params(type_name),
AtlasEntityWithExtInfo, query_params)
def remove_classification_by_name(self, type_name, uniq_attributes,
classification_name):
query_params = attributes_to_params(uniq_attributes)
query = {
'type_name': type_name,
'classification_name': classification_name
}
self.client.call_api(
EntityClient.DELETE_CLASSIFICATION_BY_TYPE_AND_ATTRIBUTE.
format_path({query}), None, query_params)
def get_lineage_info_attr(self, type_name, attributes, direction, depth):
query_params = attributes_to_params(attributes, {"direction": direction, "depth": depth})
return self.client.call_api(LineageClient.GET_LINEAGE_BY_ATTRIBUTES.format_path_with_params(type_name), AtlasLineageInfo, query_params)
def set_labels_by_name(self, type_name, uniq_attributes, labels):
query_param = attributes_to_params(uniq_attributes)
self.client.call_api(
EntityClient.ADD_LABELS_BY_UNIQUE_ATTRIBUTE.format(
{'type_name': type_name}), None, query_param, labels)
def delete_entity_by_attribute(self, type_name, uniq_attributes):
query_param = attributes_to_params(uniq_attributes)
return self.client.call_api(
EntityClient.DELETE_ENTITY_BY_ATTRIBUTE.format_path_with_params(
type_name), EntityMutationResponse, query_param)
def get_entity_header_by_attribute(self, type_name, uniq_attributes):
query_params = attributes_to_params(uniq_attributes)
return self.client.call_api(
EntityClient.GET_ENTITY_HEADER_BY_UNIQUE_ATTRIBUTE.format_path(
{'type_name': type_name}), AtlasEntityHeader, query_params)
