AWS CDK中如何将YAML配置的字符串转为RemovalPolicy类型？

问题场景 #

使用AWS CDK创建S3 Bucket时，尝试通过字符串插值传递removal_policy参数，代码如下：

object_storage_bucket = s3.Bucket(
            self,
            "object-storage-bucket",
            removal_policy=f"RemovalPolicy.{props.s3_removal_policy}",
            auto_delete_objects=True,
            object_ownership=s3.ObjectOwnership.BUCKET_OWNER_ENFORCED,
            bucket_name=f"{props.customer}-{props.region}-object-storage",
        )

运行时触发错误：

TypeError: type of argument removal_policy must be one of (aws_cdk.RemovalPolicy, NoneType); got str instead

已知直接写removal_policy=RemovalPolicy.DESTROY可正常工作，但希望从以下YAML配置文件中传入DESTROY或RETAIN值：

customer: taylor swift
region: us-west-2
s3_removal_policy: DESTROY

解决方案 #

问题核心是removal_policy要求传入**aws_cdk.RemovalPolicy枚举类型**，而非字符串。需将YAML中读取的字符串值转换为对应的枚举实例，以下是几种可行实现方式：

方法1：字典映射（直观可控） #

创建字符串到枚举值的映射字典，根据配置值匹配对应枚举：

from aws_cdk import RemovalPolicy
import yaml

# 读取YAML配置
with open("config.yaml", "r") as f:
    props = yaml.safe_load(f)

# 建立字符串与RemovalPolicy的映射关系
removal_policy_map = {
    "DESTROY": RemovalPolicy.DESTROY,
    "RETAIN": RemovalPolicy.RETAIN,
    "SNAPSHOT": RemovalPolicy.SNAPSHOT
}

# 创建Bucket时使用映射后的枚举值
object_storage_bucket = s3.Bucket(
    self,
    "object-storage-bucket",
    removal_policy=removal_policy_map[props["s3_removal_policy"]],
    auto_delete_objects=True,
    object_ownership=s3.ObjectOwnership.BUCKET_OWNER_ENFORCED,
    bucket_name=f"{props['customer']}-{props['region']}-object-storage",
)

该方式可明确限定允许的取值，避免非法字符串传入。

方法2：用getattr动态获取枚举 #

利用Python的getattr函数直接从RemovalPolicy类中读取对应属性：

from aws_cdk import RemovalPolicy
import yaml

# 读取YAML配置
with open("config.yaml", "r") as f:
    props = yaml.safe_load(f)

# 动态转换字符串为枚举值
try:
    removal_policy = getattr(RemovalPolicy, props["s3_removal_policy"])
except AttributeError:
    # 处理非法配置值，可自定义默认行为或抛出明确错误
    raise ValueError(f"无效的s3_removal_policy值：{props['s3_removal_policy']}，允许值为DESTROY、RETAIN、SNAPSHOT")

# 创建Bucket
object_storage_bucket = s3.Bucket(
    self,
    "object-storage-bucket",
    removal_policy=removal_policy,
    auto_delete_objects=True,
    object_ownership=s3.ObjectOwnership.BUCKET_OWNER_ENFORCED,
    bucket_name=f"{props['customer']}-{props['region']}-object-storage",
)

方法3：CDK上下文配置（推荐最佳实践） #

将配置存入CDK上下文（如cdk.json或命令行参数），通过self.node.try_get_context读取后转换为枚举：

// cdk.json
{
  "context": {
    "customer": "taylor swift",
    "region": "us-west-2",
    "s3_removal_policy": "DESTROY"
  }
}

代码实现：

from aws_cdk import RemovalPolicy

# 从CDK上下文读取配置
customer = self.node.try_get_context("customer")
region = self.node.try_get_context("region")
s3_removal_policy_str = self.node.try_get_context("s3_removal_policy")

# 转换为枚举值
removal_policy = getattr(RemovalPolicy, s3_removal_policy_str)

# 创建Bucket
object_storage_bucket = s3.Bucket(
    self,
    "object-storage-bucket",
    removal_policy=removal_policy,
    auto_delete_objects=True,
    object_ownership=s3.ObjectOwnership.BUCKET_OWNER_ENFORCED,
    bucket_name=f"{customer}-{region}-object-storage",
)

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