Bulk delete items from a CosmosDB container by query (cross partition key)

cosmosdb_bulk_delete_by_query.py

from azure.cosmos import CosmosClient

url = "YOUR_DB_URL"
key = "YOUR_DB_KEY"
database = "YOUR_DB_NAME"
container = "YOUR_CONTAINER"

query = input("Please input the query:")

# For example:
# SELECT * FROM c WHERE c.Data.Time>"2024-05-15"

if not query.startswith("SELECT"):
    print("Query must start with SELECT")
    exit()

client = CosmosClient(url, key)

db = client.get_database_client(database=database)

container = db.get_container_client(container=container)

items = container.query_items(
    query=query,
    enable_cross_partition_query=True)

toDelete = []
cnt = 0
for i in items:
    cnt += 1
    toDelete.append(i)

z = input(f"Delete {cnt} items (y = yes)?")

if z != 'y':
    print("Aborted")
    exit()

for i in toDelete:
    container.delete_item(item=i['id'], partition_key=i['partitionKey'])

print("Done")