如何从 AWS Lambda 函数访问 YAML 文件中的变量？

Question

我正在使用无服务器框架来测试我的 lamdba 函数，但它要求我访问我的 yaml 文件以获取特定变量。为了清楚起见，我将在下面添加文件和代码：

我需要访问的 YAML 变量是 tableName

environment:
  tableName: notes_project

这是 lambda 函数：

import json
import boto3
import uuid

dynamodb = boto3.resource('dynamodb')

table = dynamodb.Table('notes_project')

def lambda_handler(event, context):

    data = json.loads(event['body'])

    item = {
        'user_id': event['requestContext']['identity']['cognitoIdentityId'],
        'note_id': str(uuid.uuid1()),
        'note_text': data
    }

    table.put_item( Item = item)

目前，您可以看到表名是硬编码的，但我想避免这种情况。

到目前为止我尝试过的：

我查看了Get yaml key value in python 和其他涉及 os.environ 的建议，但如果我要将代码部署到 AWS，我不确定这些是否是正确的选择。我几乎不知道 lambda 代码在部署后如何访问这些 yaml 变量。解释将不胜感激。

Answer 1

serverless.yml 不会上传到 lambda 函数代码，因此您将无权访问该文件。

您可以通过几种方式做到这一点，

将值作为环境变量直接存储在 serverless.yml 中

在serverless.yml 中为 lambda 定义环境变量。然后你就可以访问 lambda 中的环境变量了。

#serverless.yml
service: my-service

provider:
  name: aws
  runtime: python3.8

functions:
  hello:
    handler: handler.hello
    environment:
      # store the value directly
      deliveryMethod: email

将参数存储在 SSM 存储中

将参数存储在SSM参数存储中并在serverless.yml中引用它

#serverless.yml
service: my-service

provider:
  name: aws
  runtime: python3.8

functions:
  hello:
    handler: handler.hello
    environment:
      # store the value directly
      deliveryMethod: ${ssm:/path/to/deliveryMethod}

然后在 lambda 代码中：

# lambda code
def lambda_handler(event, context):
    deliveryMethod = os.environ['deliveryMethod']

希望这会有所帮助。

参考： https://serverless.com/framework/docs/providers/aws/guide/variables/