def import_product_sets(project_id, location, gcs_uri):
"""Import images of different products in the product set.
Args:
project_id: Id of the project.
location: A compute region name.
gcs_uri: Google Cloud Storage URI.
Target files must be in Product Search CSV format.
"""
client = vision.ProductSearchClient()
# A resource that represents Google Cloud Platform location.
location_path = f"projects/{project_id}/locations/{location}"
# Set the input configuration along with Google Cloud Storage URI
gcs_source = vision.ImportProductSetsGcsSource(csv_file_uri=gcs_uri)
input_config = vision.ImportProductSetsInputConfig(gcs_source=gcs_source)
# Import the product sets from the input URI.
response = client.import_product_sets(parent=location_path,
input_config=input_config)
print('Processing operation name: {}'.format(response.operation.name))
# synchronous check of operation status
result = response.result()
print('Processing done.')
for i, status in enumerate(result.statuses):
print('Status of processing line {} of the csv: {}'.format(i, status))
# Check the status of reference image
# `0` is the code for OK in google.rpc.Code.
if status.code == 0:
reference_image = result.reference_images[i]
print(reference_image)
else:
print('Status code not OK: {}'.format(status.message))
def test_import_product_sets_blocked(self):
# The csv file is outside the secure perimeter.
gcs_source = vision.ImportProductSetsGcsSource(
csv_file_uri=self.gcs_uri_blocked_file)
input_config = vision.ImportProductSetsInputConfig(
gcs_source=gcs_source)
with pytest.raises(exceptions.Forbidden) as exc:
self.ps_client.import_product_sets(parent=self.location_path,
input_config=input_config)
assert self.gcs_read_error_message in exc.value.message
def test_import_product_sets(self):
# Generate the ids that will be used in the import.
product_set_id = "set" + unique_resource_id()
product_set_path = self.ps_client.product_set_path(
project=PROJECT_ID,
location=self.location,
product_set=product_set_id)
self.product_sets_to_delete.append(product_set_path)
product_id = "product" + unique_resource_id()
product_path = self.ps_client.product_path(project=PROJECT_ID,
location=self.location,
product=product_id)
self.products_to_delete.append(product_path)
reference_image_id_1 = "reference_image_1" + unique_resource_id()
reference_image_path = self.ps_client.reference_image_path(
project=PROJECT_ID,
location=self.location,
product=product_id,
reference_image=reference_image_id_1,
)
self.reference_images_to_delete.append(reference_image_path)
reference_image_id_2 = "reference_image_2" + unique_resource_id()
reference_image_path = self.ps_client.reference_image_path(
project=PROJECT_ID,
location=self.location,
product=product_id,
reference_image=reference_image_id_2,
)
self.reference_images_to_delete.append(reference_image_path)
# Upload images to gcs.
gcs_uri_image_1 = self._upload_image("import_sets_image_1.jpg")
gcs_uri_image_2 = self._upload_image("import_sets_image_2.jpg")
# Build the string that will be uploaded to gcs as a csv file.
csv_data = "\n".join([
self._build_csv_line(gcs_uri_image_1, reference_image_id_1,
product_set_id, product_id),
self._build_csv_line(gcs_uri_image_2, reference_image_id_2,
product_set_id, product_id),
])
# Upload a csv file to gcs.
csv_filename = "import_sets.csv"
blob = self.test_bucket.blob(csv_filename)
self.to_delete_by_case.append(blob)
blob.upload_from_string(csv_data)
# Make the import_product_sets request.
gcs_source = vision.ImportProductSetsGcsSource(
csv_file_uri="gs://{bucket}/{blob}".format(
bucket=self.test_bucket.name, blob=csv_filename))
input_config = vision.ImportProductSetsInputConfig(
gcs_source=gcs_source)
response = self.ps_client.import_product_sets(
parent=self.location_path, input_config=input_config)
# Verify the result.
image_prefix = "import_sets_image_"
for ref_image in response.result().reference_images:
self.assertTrue(
image_prefix in ref_image.uri,
"'{image_prefix}' not in '{uri}'".format(
image_prefix=image_prefix, uri=ref_image.uri),
)
for status in response.result().statuses:
self.assertEqual(status.code, grpc.StatusCode.OK.value[0])