如何在Python/boto3中全局配置DynamoDB Endpoint URL适配本地测试？

我目前在基于Python/boto3的应用中使用DynamoDB Local做本地单元测试，现在的做法是通过test_mode变量分支创建客户端和资源：

if test_mode:
    client = boto3.client("dynamodb", endpoint_url="localhost:8000")
    resource = boto3.resource("dynamodb", endpoint_url="localhost:8000")
else:
    client = boto3.client("dynamodb")
    resource = boto3.resource("dynamodb")

但我想去掉这种分支判断，有没有办法全局配置boto3的DynamoDB endpoint URL？

当然可以！有几种优雅的方式能帮你摆脱这种分支判断，全局统一配置boto3的DynamoDB endpoint，下面是最常用的几种方案：

1. 配置boto3全局默认会话 #

你可以在应用初始化阶段（比如测试环境的启动脚本、测试用例的前置代码里）调用boto3.setup_default_session()，一次性设置DynamoDB的endpoint，后续所有创建的client/resource都会自动继承这个配置。

示例代码：

import boto3
import os

# 仅在测试环境执行这段配置（通过环境变量控制）
if os.getenv("TEST_MODE") == "true":
    boto3.setup_default_session(
        dynamodb={"endpoint_url": "http://localhost:8000"}
    )

# 之后业务代码里直接创建客户端/资源，无需再指定endpoint
client = boto3.client("dynamodb")
resource = boto3.resource("dynamodb")

这种方式只需要一次配置，后续所有boto3调用都会自动使用你设置的endpoint，完全不用在业务代码里加判断。

2. 通过环境变量控制（推荐） #

boto3原生支持通过环境变量来配置DynamoDB的endpoint，这种方式完全不需要修改代码，只需要在测试环境设置对应的环境变量即可。

设置环境变量 #

• Linux/macOS：在终端执行

  export AWS_DYNAMODB_ENDPOINT_URL=http://localhost:8000
  

• Windows：在命令提示符执行

  set AWS_DYNAMODB_ENDPOINT_URL=http://localhost:8000
  

代码里直接使用 #

设置完环境变量后，业务代码里直接创建client/resource即可，boto3会自动读取这个环境变量并应用：

client = boto3.client("dynamodb")
resource = boto3.resource("dynamodb")

生产环境不设置这个环境变量，boto3就会默认连接AWS官方的DynamoDB服务，完美实现无代码侵入的环境切换。

3. 封装自定义Session（适合复杂场景） #

如果需要更精细的配置控制（比如同时配置其他AWS服务的参数），可以封装一个自定义的Session工厂函数，所有DynamoDB操作都基于这个Session创建：

import boto3
import os

def get_dynamodb_session():
    session = boto3.Session()
    # 根据环境变量判断是否使用本地DynamoDB
    if os.getenv("TEST_MODE") == "true":
        session._session.set_config_variable(
            "dynamodb_endpoint_url", "http://localhost:8000"
        )
    return session

# 初始化一次Session
dynamodb_session = get_dynamodb_session()

# 后续通过这个Session创建客户端/资源
client = dynamodb_session.client("dynamodb")
resource = dynamodb_session.resource("dynamodb")

这种方式把配置逻辑集中在一处，代码更整洁，也方便后续扩展其他配置。

4. 结合pytest Fixture（单元测试场景） #

如果你的单元测试用pytest框架，可以写一个全局自动执行的fixture，在测试会话开始时自动配置boto3指向DynamoDB Local：

import boto3
import pytest

@pytest.fixture(scope="session", autouse=True)
def setup_dynamodb_local():
    """全局配置测试环境使用DynamoDB Local"""
    boto3.setup_default_session(
        dynamodb={"endpoint_url": "http://localhost:8000"}
    )

这个fixture会自动在所有测试用例执行前运行，所有测试代码里创建的DynamoDB client/resource都会默认指向本地实例，完全不用在测试用例里加任何判断。

内容的提问来源于stack exchange，提问作者Nathan