Terraform打包Python Lambda代码前如何注入LaunchTemplate资源变量

你之前直接在源码中写入Terraform资源引用字符串不生效的核心原因：archive_file 只会原样读取打包文件内容，不会自动解析文件内的Terraform资源引用语法，必须通过显式处理完成值注入，以下是可落地的方案：

方案1：模板渲染后打包（符合打包前注入代码的要求） #

• 实现逻辑：将Python源码作为模板，Terraform提前把变量替换为实际资源值后，再对渲染完成的文件打包
• 操作步骤：
  1. 把原Python文件中硬编码的启动模板参数改为模板占位符，比如将文件命名为lambda_function.py.tpl，对应位置写为LAUNCH_TEMPLATE_ID = "${launch_template_id}"
  2. Terraform侧通过template_file数据源渲染模板，传入LaunchTemplate的实际资源属性，再将渲染后的代码写入临时打包目录，最后执行打包，参考配置：

# 读取并渲染源码模板
data "template_file" "lambda_render" {
  template = file("${path.module}/src/lambda_function.py.tpl")
  vars = {
    lt_id   = aws_launch_template.ingest_server_lt.id
    lt_name = aws_launch_template.ingest_server_lt.name
  }
}

# 将渲染完成的代码写入打包目录
resource "local_file" "lambda_rendered_code" {
  content  = data.template_file.lambda_render.rendered
  filename = "${path.module}/build/lambda_function.py"
}

# 打包渲染后的代码，通过depends_on保证渲染先执行
data "archive_file" "lambda_package" {
  type        = "zip"
  source_dir  = "${path.module}/build"
  output_path = "${path.module}/build/lambda_deploy.zip"
  depends_on  = [local_file.lambda_rendered_code]
}

方案2：字符串替换预处理（符合打包前注入代码的要求） #

• 实现逻辑：不需要修改源码文件后缀，直接在原始Python代码里用特殊字符串作为占位符，Terraform读取源码内容后做文本替换，再写入临时目录打包，不影响本地IDE对Python文件的语法识别
• 参考配置：

locals {
  # 读取原始Python源码
  raw_code = file("${path.module}/src/lambda_function.py")
  # 替换占位符为实际资源值
  rendered_code = replace(
    replace(local.raw_code, "__REPLACE_LT_ID__", aws_launch_template.ingest_server_lt.id),
    "__REPLACE_LT_NAME__", aws_launch_template.ingest_server_lt.name
  )
}

# 写入渲染后代码、打包的逻辑和方案1一致
resource "local_file" "lambda_rendered_code" {
  content  = local.rendered_code
  filename = "${path.module}/build/lambda_function.py"
}

你只需要在原始Python代码里把硬编码值换成__REPLACE_LT_ID__这类自定义占位符即可。

方案3：Lambda环境变量传值（生产环境推荐，无需修改打包流程） #

如果不强制要求把值硬编码进打包后的代码文件，这个方案维护成本最低，不需要做任何源码预处理，原有打包逻辑可以完全保留：

1. 修改Python代码，从环境变量读取配置值，不要硬编码：

import os
import boto3

# 从环境变量读取启动模板ID
LAUNCH_TEMPLATE_ID = os.environ["LAUNCH_TEMPLATE_ID"]
ec2_client = boto3.client("ec2")

def lambda_handler(event, context):
    # 原有创建EC2实例的逻辑
    response = ec2_client.run_instances(
        LaunchTemplate={"LaunchTemplateId": LAUNCH_TEMPLATE_ID},
        MinCount=1,
        MaxCount=1,
        # 其余实例配置
    )
    return response

2. Terraform配置Lambda函数时，直接把资源值注入环境变量即可：

resource "aws_lambda_function" "ec2_launcher" {
  filename      = data.archive_file.lambda_package.output_path
  function_name = "daily-ec2-launcher"
  role          = aws_iam_role.lambda_exec.arn
  handler       = "lambda_function.lambda_handler"
  runtime       = "python3.12"

  environment {
    variables = {
      LAUNCH_TEMPLATE_ID = aws_launch_template.ingest_server_lt.id
    }
  }
}

这个方案后续如果需要切换LaunchTemplate，不需要重新打包代码，只需要更新Lambda配置即可生效。

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