从 API 授权方访问 lambda 函数 (python) 中的 principalId

Question

我有一个 API 授权器，它工作正常，但我想在我的 lambda 函数（用 python 编写）中访问获得的 principalId。我在其他答案中看到他们建议使用模板映射，但我根本无法让它工作。我用以下内容创建了一个简单的映射：

{
    "userId" : "$context.authorizer.principalId"
}

对于Content-Type：application/json 并尝试使用这两个选项：

当没有模板匹配请求时Content-Type header
没有定义模板时（推荐）

但无论如何，当我尝试访问函数中的变量时，它会给我一个 keyError。我也尝试使用默认为您提供的映射并通过“上下文”访问它，但我也得到一个 keyError。

要访问它，我只需使用：

event["userId"]

或

event["context"]

但在这两种情况下我都得到 keyError。

更新 1： 这是仅供参考的代码，函数和授权方在API中都可以正常工作，唯一的问题是当我尝试在函数中获取principalId时。

lambda 函数代码：

def handler(event, context):
    print(event['userId'])
    # Your code goes here!
    documento = event.get("documento")
    tipo = event.get("tipo")
    documento_bytes = base64.decodebytes(documento.encode('utf-8'))
    if tipo == "ine":
        return get_info_ine(documento_bytes)
    else:
        respuesta = {
            "estatus" : "ERROR",
            "mensaje" : "El tipo de documento <%s> enviado no existe" % tipo,
            "claveMensaje" : 2
        }
        return respuesta

授权码：

import psycopg2
import base64
import re
from perfil import *


def handler(event, context):
    login_string = event["headers"]["Authorization"]
    
    login_string = re.sub('^Basic ', '', login_string)
    username, password = base64.b64decode(
        login_string).decode("UTF-8").split(':')

    connection, cursor = get_connection(hostname, dbuser, dbpass, database)

    cursor.execute("a query here")
    id = 0
    effect = "Deny"
    for record in cursor:
        id = record[0]
        effect = "Allow"

    return {
        "principalId": str(id),
        "policyDocument": {
            "Version": "2012-10-17",
            "Statement": [
                {
                    "Action": "execute-api:Invoke",
                    "Effect": effect,
                    "Resource": "arn:aws:execute-api:us-west-2:12345678910:xyz/*/POST/*"
                }
            ]
        }
    }

Answer 1

根据我的理解，授权方将来自数据库的用户 ID 放在生成的策略的 PrincipalId 中。

目标是在函数中访问这个id。

可以在具有键属性requestContext 的输入event 对象上找到授权方附加的策略。完整路径是：event['requestContext']['authorizer']['principalId']

这是一个简单的 hello 函数，它从其授权方读取（并输出）PrincipalId 值：

def hello(event, context):
    print('event', event)
    userId = None
    if 'requestContext' in event and 'authorizer' in event['requestContext']:
        userId = event['requestContext']['authorizer']['principalId']
    body = {
        "userId": userId,
        "event": event
    }

    response = {
        "statusCode": 200,
        "headers": {
            # Required for CORS support to work
            'Access-Control-Allow-Origin': '*',
            # Required for cookies, authorization headers with HTTPS
            'Access-Control-Allow-Credentials': True,
        },
        "body": json.dumps(body)
    }

    return response

从 API 授权方访问 lambda 函数 (python) 中的 principalId

Access principalId in lambda function (python) from API Authorizer

aws-lambda

lambda-authorizer