m

Author: Lucas McDonald

TestVectors/runtimes/python/src/aws_dbesdk_dynamodb_test_vectors/internaldafny/extern/CreateInterceptedDDBTable.py

@@ -132,16 +132,18 @@ def batch_write_item(self, **kwargs):
         # There isn't a resource shape for this;
         table_input = self._client_shape_to_resource_shape_converter.batch_write_item_request(kwargs)
         # table_output = self._table.batch_write_item(**table_input)
-        batch_writer = self._table.batch_writer()
-        for _, items in table_input["RequestItems"].items():
-            for item in items:
-                if "PutRequest" in item:
-                    batch_writer.put_item(item["PutRequest"]["Item"])
-                elif "DeleteRequest" in item:
-                    batch_writer.delete_item(item["DeleteRequest"]["Key"])
-                else:
-                    raise ValueError(f"Unknown request type: {item}")
-        table_output = batch_writer.close()
+        with self._table.batch_writer() as batch_writer:
+            for _, items in table_input["RequestItems"].items():
+                for item in items:
+                    if "PutRequest" in item:
+                        batch_writer.put_item(item["PutRequest"]["Item"])
+                    elif "DeleteRequest" in item:
+                        batch_writer.delete_item(item["DeleteRequest"]["Key"])
+                    else:
+                        raise ValueError(f"Unknown request type: {item}")
+        # There isn't a shape for the output, but luckily the output can be an empty dict:
+        # https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/dynamodb/client/batch_write_item.html
+        table_output = {}
         client_output = self._resource_shape_to_client_shape_converter.batch_write_item_response(table_output)
         return client_output