azure_cosmos_db_nosql_crud_operations_python_sdk_quickstart.py

1import os
2import json
3from azure.cosmos import CosmosClient, PartitionKey
4
5# Replace with your own Azure Cosmos DB configuration
6ENDPOINT = "https://<your-account-name>.documents.azure.com:443/"
7KEY = "<your-account-key>"
8
9# Initialize the Cosmos client
10client = CosmosClient(ENDPOINT, KEY)
11
12# Create a database
13database_name = 'adventureworks'
14database = client.create_database_if_not_exists(id=database_name)
15
16# Create a container
17container_name = 'products'
18container = database.create_container_if_not_exists(
19    id=container_name, 
20    partition_key=PartitionKey(path="/categoryId"),
21    offer_throughput=400
22)
23
24# Create a new item
25new_item = {
26    'id': '001',
27    'categoryId': '61dba35b-4aa0-4b0d-92c2-39314d0abc4d',
28    'categoryName': 'gear-surf-surfboards',
29    'name': 'Yama Surfboard',
30    'quantity': 12,
31    'sale': False
32}
33
34container.create_item(body=new_item)
35
36# Read an item
37item_response = container.read_item(item='001', partition_key='61dba35b-4aa0-4b0d-92c2-39314d0abc4d')
38print(f"Read item: {item_response['name']}")
39
40# Query items
41query = "SELECT * FROM c WHERE c.categoryId IN ('61dba35b-4aa0-4b0d-92c2-39314d0abc4d')"
42items = list(container.query_items(
43    query=query,
44    enable_cross_partition_query=True
45))
46
47print(f"Found {len(items)} items")
48for item in items:
49    print(json.dumps(item, indent=True))