# Este ejemplo usa la biblioteca Azure SDK para Python
from azure.cosmos import CosmosClient, exceptions, PartitionKey

# Configuración de la cuenta y la clave
endpoint = "https://<your-account-name>.documents.azure.com:443/"
key = '<your-account-key>'

# Crear un cliente de Cosmos DB
client = CosmosClient(endpoint, key)

# Crear una base de datos
database_name = 'AzureSampleFamilyDatabase'
database = client.create_database_if_not_exists(id=database_name)

# Crear un contenedor
container_name = 'FamilyContainer'
container = database.create_container_if_not_exists(
    id=container_name,
    partition_key=PartitionKey(path="/lastName"),
    offer_throughput=400
)

# Crear un documento JSON
family_item = {
    'id': 'AndersenFamily',
    'lastName': 'Andersen',
    'parents': [
        {'firstName': 'Thomas'},
        {'firstName': 'Mary Kay'}
    ],
    'children': [
        {'firstName': 'Henriette Thaulow', 'gender': 'female', 'grade': 5}
    ],
    'address': {'state': 'WA', 'county': 'King', 'city': 'Seattle'}
}

# Insertar el documento en el contenedor
container.create_item(body=family_item)

# Leer un documento por su ID y clave de partición
response = container.read_item(item='AndersenFamily', partition_key='Andersen')
print(response)

# Actualizar un documento existente
family_item['address']['city'] = 'Redmond'
container.upsert_item(body=family_item)

# Eliminar un documento por su ID y clave de partición
container.delete_item(item='AndersenFamily', partition_key='Andersen')

# Crear un nuevo documento
new_family_item = {
    'id': 'WakefieldFamily',
    'lastName': 'Wakefield',
    'parents': [
        {'firstName': 'Robin'},
        {'firstName': 'Ben'}
    ],
    'children': [
        {'firstName': 'Jesse', 'gender': 'male', 'grade': 8},
        {'firstName': 'Lisa', 'gender': 'female', 'grade': 5}
    ],
    'address': {'state': 'NY', 'county': 'Westchester', 'city': 'Mount Kisco'}
}

# Insertar el nuevo documento
container.create_item(body=new_family_item)

# Leer el documento
response = container.read_item(item='WakefieldFamily', partition_key='Wakefield')
print(response)

# Actualizar el documento
new_family_item['address']['city'] = 'New York'
container.upsert_item(body=new_family_item)

# Eliminar el documento
container.delete_item(item='WakefieldFamily', partition_key='Wakefield')