Merge pull request #271529 from wmwxwa/patch-16

Update Cosmos DB connector in functions-reference-python.md

Author: prmerger-automator[bot]

articles/azure-functions/functions-reference-python.md

@@ -219,10 +219,13 @@ When you deploy your project to a function app in Azure, the entire contents of
 To connect to Cosmos DB, first [create an account, database, and container](../cosmos-db/nosql/quickstart-portal.md). Then you may connect Functions to Cosmos DB using [trigger and bindings](functions-bindings-cosmosdb-v2.md), like this [example](functions-add-output-binding-cosmos-db-vs-code.md). You may also use the Python library for Cosmos DB, like so:
 
 ```python
-pip install azure-cosmos
+pip install azure-cosmos
+pip install aiohttp
 
-from azure.cosmos import CosmosClient, exceptions
+from azure.cosmos.aio import CosmosClient
+from azure.cosmos import exceptions
 from azure.cosmos.partition_key import PartitionKey
+import asyncio
 
 # Replace these values with your Cosmos DB connection information
 endpoint = "https://azure-cosmos-nosql.documents.azure.com:443/"
@@ -234,41 +237,42 @@ partition_key = "/partition_key"
 # Set the total throughput (RU/s) for the database and container
 database_throughput = 1000
 
-# Initialize the Cosmos client
-client = CosmosClient(endpoint, key)
+# Helper function to get or create database and container
+async def get_or_create_container(client, database_id, container_id, partition_key):
 
-# Create or get a reference to a database
-try:
-    database = client.create_database_if_not_exists(id=database_id)
+    database = await client.create_database_if_not_exists(id=database_id)
     print(f'Database "{database_id}" created or retrieved successfully.')
 
-except exceptions.CosmosResourceExistsError:
-    database = client.get_database_client(database_id)
-    print('Database with id \'{0}\' was found'.format(database_id))
-
-# Create or get a reference to a container
-try:
-    container = database.create_container(id=container_id, partition_key=PartitionKey(path='/partitionKey'))
-    print('Container with id \'{0}\' created'.format(container_id))
-
-except exceptions.CosmosResourceExistsError:
-    container = database.get_container_client(container_id)
-    print('Container with id \'{0}\' was found'.format(container_id))
-
-# Sample document data
-sample_document = {
-    "id": "1",
-    "name": "Doe Smith",
-    "city": "New York",
-    "partition_key": "NY"
-}
-
-# Insert a document
-container.create_item(body=sample_document)
-
-# Query for documents
-query = "SELECT * FROM c where c.id = 1"
-items = list(container.query_items(query, enable_cross_partition_query=True))
+    container = await database.create_container_if_not_exists(id=container_id, partition_key=PartitionKey(path=partition_key))
+    print(f'Container with id "{container_id}" created')
+ 
+    return container
+ 
+async def create_products():
+    async with CosmosClient(endpoint, credential=key) as client:
+        container = await get_or_create_container(client, database_id, container_id, partition_key)
+        for i in range(10):
+            await container.upsert_item({
+                'id': f'item{i}',
+                'productName': 'Widget',
+                'productModel': f'Model {i}'
+            })
+ 
+async def get_products():
+    items = []
+    async with CosmosClient(endpoint, credential=key) as client:
+        container = await get_or_create_container(client, database_id, container_id, partition_key)
+        async for item in container.read_all_items():
+            items.append(item)
+    return items
+ 
+async def main():
+    await create_products()
+    products = await get_products()
+    print(products)
+ 
+if __name__ == "__main__":
+    asyncio.run(main())
 ```
 
 ::: zone pivot="python-mode-decorators"