Azure Cosmos DB Interaction through Python

Since I have already installed the library, it is showing as already satisfied. If it is not installed previously, it will install the library.

• Open the Python IDLE Shell.
• In the IDLE shell, click on File -> New and open a new file
• The architecture of Cosmos DB is such that you have the Azure Cosmos DB account, under which you will have the Database. The Container will be created under the Database and individual Items will be inserted in the Container. This will be the same sequence which we will be following.

First step is to import the CosmosClient and PartitionKey class from the azure.cosmos module. CosmosClient will have the method to instantiate the Cosmos DB account which we will then use to create the Database. PartitionKey class is used to define the Partition Key for the Container items. Paste the below command in the new IDLE Shell file.

from azure.cosmos import CosmosClient,PartitionKey

3. Connect to account, create database and container

The next step is to provide the Endpoint URI and the Key to access the Cosmos DB account.

ENDPOINT = "<YOURENDPOINTURI>"
KEY = "<YOURENDPOINTKEY>"

DATABASE_NAME = "Toyota"
CONTAINER_NAME = "products"

Note. I have provided the endpoint and key here in the open for illustration purpose. However, the best practice is to store the secrets in Keyvault and access it from there.

Next step is to instantiate the Cosmos DB with the below command:

client = CosmosClient(url=ENDPOINT, credential=KEY)

With the help of the “client” object we will now create the Database object with the below command. DATABASE_NAME is the constant value provided at the beginning.

database = client.create_database_if_not_exists(id=DATABASE_NAME)

Once the “database” object is created, we will use it to create the container object. We will also be defining the Partition Key using the PartitionKey class imported in the first line. Use the below codes to perform the same.

key_path = PartitionKey(path="/categoryId")

container = database.create_container_if_not_exists(
    id=CONTAINER_NAME, 
    partition_key=key_path, 
    offer_throughput=400
)

print("Container\t", container.id)

/CategoryId is one of the properties in the JSON input that we will be providing. 400 Throughput is the maximum throughput offered for a Free Tier. Throughput is nothing but the number of request unit per second. CONTAINER_NAME is provided as a constant at the beginning of the code.

4. Create a sample input data

Below are the inputs that we are going to store in the DB.

new_item1 = {
    "id": "80b63682-b93a-4c77-aad2-65501347265f",
    "categoryId": "61dba35b-4f02-45c5-b173-c6badc0cbd79",
    "categoryName": "SUV",
    "name": "Brezza",
    "Drive": "2WD",
}

new_item2 = {
    "id": "80b63683-b93b-4c77-agd2-65501384965f",
    "categoryId": "61dsc35b-4e50-45c5-b728-c6banm8cbd79",
    "categoryName": "Hatchback",
    "name": "Baleno",
    "Drive": "2WD",
}

Through below steps, we are inserting the inputs into our containers.

container.create_item(new_item1)
container.create_item(new_item2)

5. Write our CRUD operations

Through below code we will read the container and populate the Item with ID “80b63683-b93b-4c77-agd2-65501384965f” and CategoryID as “61dsc35b-4e50-45c5-b728-c6banm8cbd79”

item = container.read_item(
item = "80b63683-b93b-4c77-agd2-65501384965f",

partition_key = "61dsc35b-4e50-45c5-b728-c6banm8cbd79")

print("output",item["name"])

On executing the code, you should be viewing the output as below.

As you can see, the Database “Toyota” and Container “products” has been created and the inputs were inserted, which were then populated through read_item method.

6. Login to Cosmos DB and verify our data

Go to the Azure Portal and go to the Cosmos DB account you had created. On the left hand side, click on “Data Explorer” You will see the output as shown below:

This is how you programmatically converse with Cosmos DB.