update samples

Author: shemogumbe

samples/batch_requests/batch_get_response_body_as_stream.py

@@ -0,0 +1,26 @@
+# ------------------------------------
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+# ------------------------------------
+#pylint: disable=undefined-variable
+""" Demonstrate getting response body as stream in Batch Responses"""
+batch_request_content = {
+    batch_request_item1.id: batch_request_item1,
+    batch_request_item2.id: batch_request_item2
+}
+
+batch_content = BatchRequestContent(batch_request_content)
+
+
+async def main():
+    batch_response_content = await batch_request_builder.post(batch_request_content=batch_content)
+
+    try:
+        stream_response = batch_response_content.get_response_stream_by_id(batch_request_item1.id)
+        print(f"Stream Response: {stream_response}")
+        print(f"Stream Response Content: {stream_response.read()}")
+    except AttributeError as e:
+        print(f"Error getting response by ID: {e}")
+
+
+asyncio.run(main())

samples/batch_requests/batch_request_with_parsable_as_response_type.py

@@ -53,14 +53,18 @@
 batch_content = BatchRequestContent(batch_request_content)
 
 
+# Function to demonstrate the usage of BatchRequestBuilder
 async def main():
-    # Send the batch request and get the response content
-    batch_response_content = await batch_request_builder.post(
-        batch_request_content=batch_content, response_type=User
-    )
-
-    # Print the batch response content - User model type
-    print(f"Batch Response Content: {batch_response_content}")
+    batch_response_content = await batch_request_builder.post(batch_request_content=batch_content)
+    # response_type=User
+
+    try:
+        individual_response = batch_response_content.get_response_by_id(
+            batch_request_item1.id, User
+        )
+        print(f"Individual Response: {individual_response}")
+    except AttributeError as e:
+        print(f"Error getting response by ID: {e}")
 
 
 asyncio.run(main())

samples/batch_requests/batch_response_get_status_codes.py

@@ -0,0 +1,25 @@
+# ------------------------------------
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+# ------------------------------------
+#pylint: disable=undefined-variable
+"""Demonstrates getting status codes in Batch Responses"""
+batch_request_content = {
+    batch_request_item1.id: batch_request_item1,
+    batch_request_item2.id: batch_request_item2
+}
+
+batch_content = BatchRequestContent(batch_request_content)
+
+
+async def main():
+    batch_response_content = await batch_request_builder.post(batch_request_content=batch_content)
+
+    try:
+        status_codes = batch_response_content.get_response_status_codes()
+        print(f"Status Codes: {status_codes}")
+    except AttributeError as e:
+        print(f"Error getting respons status codes: {e}")
+
+
+asyncio.run(main())