首页 > 解决方案 > 如果不存在分区键和唯一键,则使用 Python 程序创建容器

问题描述

我只想编写 python 脚本来创建容器,前提是它不存在分区键和唯一键。

Steps for  Creating Alert Container
 1. Create Container With Container ID: alerts
 2. Add Partition Key as /user_tenant
 3. Add Unique Key as /alert_id

参考链接:https ://github.com/Azure/azure-cosmos-python#create-a-container

如果不存在,请建议将创建容器的 api。

标签: pythonazure-cosmosdb

解决方案


@Gaurav Mantri 下面是您建议的工作代码。对于uniqueKeys,我们需要将其添加到uniqueKeyPolicy中,如下面的代码所示。

import azure.cosmos.documents as documents
from azure.cosmos import cosmos_client, http_constants, errors
import os

url = os.environ['COSMOS_DB_END_POINT']
key = os.environ['COSMOS_DB_MASTER_KEY']
database_name = os.environ["COSMOS_DB_DATABASE_ID"]
client = cosmos_client.CosmosClient(url, {'masterKey': key})

container_definition = {'id': 'alerts_test',
                        'partitionKey':
                            {
                                'paths': ['/user_tenant'],
                                'kind': documents.PartitionKind.Hash
                            },
                        'uniqueKeyPolicy': {
                            'uniqueKeys':
                                [
                                    {'paths': ['/alert_id']}
                                ]
                        }
                        }

try:
    container = client.CreateContainer("dbs/" + database_name, container_definition, {'offerThroughput': 400})
    print("New Container Created:")
    print(container)
except errors.HTTPFailure as e:
    if e.status_code == http_constants.StatusCodes.CONFLICT:
        container = client.ReadContainer("dbs/" + database_name + "/colls/" + container_definition['id'])
        print(container)
    else:
        raise e

推荐阅读