azure_cosmos_db_database_container_crud_and_query_quickstart.py

1import os
2import json
3from azure.cosmos import CosmosClient, PartitionKey
4
5# Initialize the Cosmos client
6# Replace with your own endpoint and key from the Azure Portal
7endpoint = os.environ["COSMOS_ENDPOINT"]
8key = os.environ["COSMOS_KEY"]
9client = CosmosClient(url=endpoint, credential=key)
10
11# Create a database
12database_name = 'AdventureWorks'
13database = client.create_database_if_not_exists(id=database_name)
14
15# Create a container
16container_name = 'Products'
17container = database.create_container_if_not_exists(
18    id=container_name, 
19    partition_key=PartitionKey(path="/adventureWorks"),
20    offer_throughput=400
21)
22
23# Add items to the container
24new_item = {
25    "id": "706",
26    "adventureWorks": "Water Bottle - 30 oz.",
27    "sku": "WB-H06",
28    "category": "Accessories, Bottles and Cages",
29    "name": "Water Bottle - 30 oz.",
30}
31
32container.create_item(body=new_item)
33
34# Read item (Point read)
35item_response = container.read_item(item="706", partition_key="Water Bottle - 30 oz.")
36print(f"Read item: {item_response['name']}")
37
38# Query items
39query = "SELECT * FROM c WHERE c.adventureWorks = 'Water Bottle - 30 oz.'"
40items = list(container.query_items(
41    query=query,
42    enable_cross_partition_query=True
43))
44
45print(f"Found {len(items)} item(s)")
46for item in items:
47    print(json.dumps(item, indent=True))