通过 Lambda 运行 aws Athena 查询：未定义错误名称“response”

Question

我创建了一个 AWS lambda function 和 python 3.9 来运行 Athena 查询并获取查询结果

import time
import boto3

# create Athena client

client = boto3.client('athena')

# create Athena query varuable

query = 'select * from mydatabase.mytable limit 8'
DATABASE = 'mydatabase'
output='s3://mybucket/'

def lambda_handler(event, context):
    # Execution
    response = client.start_query_execution(
        QueryString=query,
        QueryExecutionContext={
            'Database': DATABASE
        },
        ResultConfiguration={
            'OutputLocation': output,
        }
    )
    
query_execution_id = response['QueryExecutionId']
    
time.sleep(10)
    
result = client.get_query_results(QueryExecutionId=query_execution_id)
    
for row in results['ResultSet']['Rows']:
    print(row)

我在测试时收到此错误消息“[ERROR] NameError：未定义名称'response'”

Answer 1

您在lambda_handler function 中定义response变量。但您在全局 scope 中引用它，在 function 之外，此处：

query_execution_id = response['QueryExecutionId']

该变量未在该 scope 上定义，因此出现错误消息。 看来您可能只是缺少所有这些行的缩进：

query_execution_id = response['QueryExecutionId']
    
time.sleep(10)
    
result = client.get_query_results(QueryExecutionId=query_execution_id)
    
for row in results['ResultSet']['Rows']:
    print(row)

在 Python 缩进是语法！ 如果您希望这些行成为lambda_handler function 的一部分，那么它们需要有正确的缩进才能将它们放在 function 的 scope 中，如下所示：

def lambda_handler(event, context):
    # Execution
    response = client.start_query_execution(
        QueryString=query,
        QueryExecutionContext={
            'Database': DATABASE
        },
        ResultConfiguration={
            'OutputLocation': output,
        }
    )
    
    query_execution_id = response['QueryExecutionId']
    
    time.sleep(10)
    
    result = client.get_query_results(QueryExecutionId=query_execution_id)
    
    for row in results['ResultSet']['Rows']:
        print(row)