feat(storage): update compose sample to support deleteSourceObjects option

storage/samples/snippets/snippets_test.py

@@ -15,10 +15,10 @@
 import asyncio
 import io
 import os
+import sys
 import tempfile
 import time
 import uuid
-import sys
 
 from google.cloud import storage
 import google.cloud.exceptions
@@ -668,7 +668,10 @@ def test_object_get_kms_key(test_bucket):
 
 
 def test_storage_compose_file(test_bucket):
-    source_files = ["test_upload_blob_1", "test_upload_blob_2"]
+    source_files = [
+        f"test_upload_blob_1_{uuid.uuid4().hex}",
+        f"test_upload_blob_2_{uuid.uuid4().hex}",
+    ]
     for source in source_files:
         blob = test_bucket.blob(source)
         blob.upload_from_string(source)
@@ -684,6 +687,35 @@ def test_storage_compose_file(test_bucket):
 
         assert composed.decode("utf-8") == source_files[0] + source_files[1]
 
+        # Verify sources are NOT deleted
+        assert test_bucket.blob(source_files[0]).exists()
+        assert test_bucket.blob(source_files[1]).exists()
+
+
+def test_storage_compose_file_delete_source(test_bucket):
+    source_files = [
+        f"test_upload_blob_1_{uuid.uuid4().hex}",
+        f"test_upload_blob_2_{uuid.uuid4().hex}",
+    ]
+    for source in source_files:
+        blob = test_bucket.blob(source)
+        blob.upload_from_string(source)
+
+    with tempfile.NamedTemporaryFile() as dest_file:
+        destination = storage_compose_file.compose_file(
+            test_bucket.name,
+            source_files[0],
+            source_files[1],
+            dest_file.name,
+            delete_source_objects=True,
+        )
+        composed = destination.download_as_bytes()
+        assert composed.decode("utf-8") == source_files[0] + source_files[1]
+
+        # Verify sources are deleted
+        assert not test_bucket.blob(source_files[0]).exists()
+        assert not test_bucket.blob(source_files[1]).exists()
+
 
 def test_cors_configuration(test_bucket, capsys):
     bucket = storage_cors_configuration.cors_configuration(test_bucket)

storage/samples/snippets/storage_compose_file.py

@@ -20,7 +20,13 @@
 from google.cloud import storage
 
 
-def compose_file(bucket_name, first_blob_name, second_blob_name, destination_blob_name):
+def compose_file(
+    bucket_name,
+    first_blob_name,
+    second_blob_name,
+    destination_blob_name,
+    delete_source_objects=False,
+):
     """Concatenate source blobs into destination blob."""
     # bucket_name = "your-bucket-name"
     # first_blob_name = "first-object-name"
@@ -44,11 +50,22 @@ def compose_file(bucket_name, first_blob_name, second_blob_name, destination_blo
     # There is also an `if_source_generation_match` parameter, which is not used in this example.
     destination_generation_match_precondition = 0
 
-    destination.compose(sources, if_generation_match=destination_generation_match_precondition)
+    destination.compose(
+        sources,
+        if_generation_match=destination_generation_match_precondition,
+        delete_source_objects=delete_source_objects,
+    )
 
+    deletion_message = (
+        " and the source objects were deleted." if delete_source_objects else "."
+    )
     print(
-        "New composite object {} in the bucket {} was created by combining {} and {}".format(
-            destination_blob_name, bucket_name, first_blob_name, second_blob_name
+        "New composite object {} in the bucket {} was created by combining {} and {}{}".format(
+            destination_blob_name,
+            bucket_name,
+            first_blob_name,
+            second_blob_name,
+            deletion_message,
         )
     )
     return destination