Python 操作 DynamoDB 数据库：从连接到基本操作

Amazon DynamoDB是一种高度托管的 NoSQL 数据库服务，适用于处理大规模数据集的实时应用程序。本篇博客将详细介绍如何使用 Python 通过 AWS SDK（Boto3）连接、创建表、插入数据和查询 DynamoDB 数据库。

1. 安装 Boto3

在开始之前，确保你已经安装了 Boto3 库，它是 Amazon Web Services（AWS）的官方 SDK。

pip install boto3

2. 配置 AWS 认证信息

在使用 Boto3 之前，需要配置 AWS 认证信息。你可以通过设置环境变量或使用 AWS CLI 配置文件来实现。

• 设置环境变量

export AWS_ACCESS_KEY_ID="your_access_key"
export AWS_SECRET_ACCESS_KEY="your_secret_key"
export AWS_DEFAULT_REGION="your_region"

• 使用 AWS CLI 配置文件

运行以下命令并按照提示提供 AWS 访问密钥和默认区域。

aws configure

3. 连接到 DynamoDB

import boto3

# 创建 DynamoDB 客户端
dynamodb = boto3.resource('dynamodb')

# 选择表
table_name = 'YourTableName'
table = dynamodb.Table(table_name)

4. 创建 DynamoDB 表

在 DynamoDB 中，你需要在使用之前创建表。以下是一个简单的例子：

# 定义表的主键
key_schema = [
    {'AttributeName': 'user_id', 'KeyType': 'HASH'},  # Partition key
    {'AttributeName': 'timestamp', 'KeyType': 'RANGE'}  # Sort key
]

# 定义属性的数据类型
attribute_definitions = [
    {'AttributeName': 'user_id', 'AttributeType': 'S'},  # S for String
    {'AttributeName': 'timestamp', 'AttributeType': 'N'}  # N for Number
]

# 配置读写容量单位
provisioned_throughput = {
    'ReadCapacityUnits': 5,
    'WriteCapacityUnits': 5
}

# 创建表
table.create_table(
    TableName=table_name,
    KeySchema=key_schema,
    AttributeDefinitions=attribute_definitions,
    ProvisionedThroughput=provisioned_throughput
)

# 等待表创建完成
table.meta.client.get_waiter('table_exists').wait(TableName=table_name)

5. 插入数据

插入数据是 DynamoDB 中的一个重要操作。以下是一个简单的插入数据的例子：

from datetime import datetime

# 插入数据
user_id = 'user123'
timestamp = int(datetime.now().timestamp())

item = {
    'user_id': user_id,
    'timestamp': timestamp,
    'data': 'Your Data Here'
}

table.put_item(Item=item)

6. 查询数据

DynamoDB 提供了灵活的查询功能。以下是一个按照 user_id 查询的例子：

from boto3.dynamodb.conditions import Key

response = table.query(
    KeyConditionExpression=Key('user_id').eq('user123')
)

items = response['Items']
for item in items:
    print(item)

7. 更新和删除数据

更新和删除数据也是常见的操作，使用 update_item 和 delete_item 方法即可。

# 更新数据
update_expression = 'SET #attr = :val'
expression_attribute_values = {':val': 'Updated Data'}
expression_attribute_names = {'#attr': 'data'}

table.update_item(
    Key={'user_id': user_id, 'timestamp': timestamp},
    UpdateExpression=update_expression,
    ExpressionAttributeValues=expression_attribute_values,
    ExpressionAttributeNames=expression_attribute_names
)

# 删除数据
table.delete_item(
    Key={'user_id': user_id, 'timestamp': timestamp}
)

8. 清理资源

在结束使用 DynamoDB 后，确保关闭连接，并根据需要删除表。

# 删除表
table.delete()

结语

通过 Boto3，Python 提供了方便而强大的工具来操作 DynamoDB 数据库。通过连接、创建表、插入数据、查询和其他操作，你可以充分利用 DynamoDB 的能力。希望这篇博客能够帮助你更深入地了解如何在 Python 中操作 DynamoDB 数据库。