以下是一个使用AWS DynamoDB BatchWriteItem写入容量单位的示例代码：

import boto3

dynamodb = boto3.client('dynamodb')

table_name = 'your-table-name'

# 构建写入容量单位的参数
write_capacity_units = 5

# 构建写入请求
put_requests = []

# 添加要写入的项目
put_requests.append({
    'PutRequest': {
        'Item': {
            'id': {'N': '1'},
            'name': {'S': 'John Doe'}
        }
    }
})

# 添加更多的写入请求...

# 构建批量写入请求
request_items = {
    table_name: put_requests
}

# 发起批量写入请求
response = dynamodb.batch_write_item(
    RequestItems=request_items,
    ReturnConsumedCapacity='INDEXES' # 返回消耗的容量
)

# 检查请求的消耗容量
for table_name, consumed_capacity in response['ConsumedCapacity'].items():
    print(f"Table: {table_name}")
    print(f"Write Capacity Units: {consumed_capacity['CapacityUnits']}")
    print(f"Write Capacity Units (Indexes): {consumed_capacity['Table']['CapacityUnits']}")

在这个示例中，我们首先创建了一个DynamoDB客户端对象。然后，我们指定要写入的表名并设置写入容量单位。接下来，我们构建了一个要写入的项目列表，并将其添加到写入请求中。最后，我们使用batch_write_item方法发起批量写入请求，并通过ReturnConsumedCapacity参数指定要返回的消耗容量。在响应中，我们可以通过ConsumedCapacity字段检查请求的消耗容量。

请注意，以上代码示例仅为了说明如何使用AWS SDK for Python（Boto3）进行批量写入并获取消耗的容量。您需要根据自己的实际需求进行适当的修改。