Azure_CosmosDB_Python/cosmosdb_crud.py - main - osbornray/PythonHacks

Azure_CosmosDB_Python/cosmosdb_crud.py
98 строк3 KB
RekhuGopal
Cosmos DB Python
21 ноя 2023, 06:17
21 ноя 2023, 06:17b644198
import os
#from azure.cosmos import CosmosClient, exceptions, PartitionKey
from azure.cosmos.aio import CosmosClient 
import json
import asyncio

URL = "<Cosmos DB URL>"
KEY = "<Cosmos DB  Key>"
client = CosmosClient(URL, credential=KEY)
DATABASE_NAME = 'ToDoList'
database = client.get_database_client(DATABASE_NAME)
CONTAINER_NAME = 'Items1'
container = database.get_container_client(CONTAINER_NAME)

'''
## Create a database
try:
    database = client.create_database(DATABASE_NAME)
except exceptions.CosmosResourceExistsError:
    database = client.get_database_client(DATABASE_NAME)


## Create a container
try:
    container = database.create_container(id=CONTAINER_NAME, partition_key=PartitionKey(path='/id', kind='Hash'))
except exceptions.CosmosResourceExistsError:
    container = database.get_container_client(CONTAINER_NAME)
except exceptions.CosmosHttpResponseError:
    raise

## Create an analytical store enabled container
try:
    container = database.create_container(id=CONTAINER_NAME, partition_key=PartitionKey(path='/id', kind='Hash'),default_ttl=-1)
except exceptions.CosmosResourceExistsError:
    container = database.get_container_client(CONTAINER_NAME)
except exceptions.CosmosHttpResponseError:
    raise


## Create records
for i in range(1, 10):
    container.upsert_item({
            'id': 'item{0}'.format(i),
            'TYkey' : '1',
            'productName': 'Widget',
            'productModel': 'Model {0}'.format(i)
        }
    )



## Delete data
for item in container.query_items(query='SELECT * FROM products p WHERE p.productModel = "Model 2"', enable_cross_partition_query=True):
    print(item)
    container.delete_item(item, partition_key=item['id'])

## Query the database
for item in container.query_items(query='SELECT * FROM mycontainer r WHERE r.id="item3"',enable_cross_partition_query=True):
    print(json.dumps(item, indent=True))



## Using the asynchronous client
async def create_products():
    async with CosmosClient(URL, credential=KEY) as client: # the with statement will automatically initialize and close the async client
        database = client.get_database_client(DATABASE_NAME)
        container = database.get_container_client(CONTAINER_NAME)
        for i in range(20):
            await container.upsert_item({
                    'id': 'item{0}'.format(i),
                    'productName': 'Widget',
                    'productModel': 'Model {0}'.format(i)
                }
            )
async def main():
    await create_products()

# Run the event loop to execute the asynchronous function
asyncio.run(main())

'''
## Queries with the asynchronous client
async def create_lists():
    results = container.query_items(query='SELECT * FROM products p WHERE p.productModel = "Model 2"')
    item_list = []
    async for item in results:
        item_list.append(item)
    await client.close()
    return item_list

async def main():
    items = await create_lists()
    for item in items:
        print(item)  
    await client.close()

# Run the event loop to execute the asynchronous function 
asyncio.run(main())