def updateDataset(self, request, context): properties = None if request.properties == {} else request.properties name_list = request.name_list type_list = request.type_list if not name_list: name_list = None if not type_list: data_type_list = None else: data_type_list = [] for data_type in type_list: data_type_list.append(DataType(DataTypeProto.Name(data_type))) dataset_meta = self.store.update_dataset(dataset_name=request.name, data_format=request.data_format.value if request.HasField( 'data_format') else None, description=request.description.value if request.HasField( 'description') else None, uri=request.uri.value if request.HasField('uri') else None, properties=properties, name_list=name_list, type_list=data_type_list, catalog_name=request.catalog_name.value if request.HasField( 'catalog_name') else None, catalog_type=request.catalog_type.value if request.HasField( 'catalog_type') else None, catalog_database=request.catalog_database.value if request.HasField( 'catalog_database') else None, catalog_connection_uri=request.catalog_connection_uri.value \ if request.HasField('catalog_connection_uri') else None, catalog_table=request.catalog_table.value if request.HasField( 'catalog_table') else None) return _wrap_meta_response( MetaToProto.dataset_meta_to_proto(dataset_meta))
def registerDatasetWithCatalog(self, request, context): dataset = transform_dataset_meta(request.dataset) dataset_meta = self.store.register_dataset_with_catalog( name=dataset.name, catalog_name=dataset.catalog_name, catalog_type=dataset.catalog_type, catalog_database=dataset.catalog_database, catalog_connection_uri=dataset.catalog_connection_uri, catalog_table=dataset.catalog_table) return _wrap_meta_response( MetaToProto.dataset_meta_to_proto(dataset_meta))
def registerDataset(self, request, context): dataset = transform_dataset_meta(request.dataset) dataset_meta = self.store.register_dataset( name=dataset.name, data_format=dataset.data_format, description=dataset.description, uri=dataset.uri, properties=dataset.properties, name_list=dataset.schema.name_list, type_list=dataset.schema.type_list) return _wrap_meta_response( MetaToProto.dataset_meta_to_proto(dataset_meta))
def getDatasetByName(self, request, context): dataset = self.store.get_dataset_by_name(request.name) return _wrap_meta_response(MetaToProto.dataset_meta_to_proto(dataset))