如何从Lambda执行多行Athena查询及解决Lambda语法错误问题

嘿，我来帮你排查这个问题！首先要明确：Lambda报的Syntax error in module 'lambda_function'是Python代码本身的语法问题，不是Athena查询的问题——毕竟你在Athena编辑器里能正常跑查询对吧？咱们一步步来解决：

1. 先修复查询字符串的定义问题 #

你给出的query_1字符串里出现了. .，如果这是你写代码时的省略标记（实际是其他列名），那没问题，但如果直接写在代码里会导致SQL语法错误；另外，单行字符串如果太长或包含换行，很容易触发Python语法问题。建议用**三重引号（"""）**定义多行查询字符串，既清晰又避免出错，比如：

query_1 = """
SELECT REGEXP_EXTRACT(data,'[a-z]*[0-9]') as datacenter,
       data, pool_id  -- 替换成你实际需要的列
FROM DATABASE.DATATABLE 
WHERE pool_id LIKE 'POOL_ID%';
"""

2. 补全代码中省略的部分 #

你代码里的database = "DATABASE" ...如果实际代码里就是用...省略内容，这会直接触发Python语法错误——...在这里不是有效的语句。你需要补全完整的逻辑，比如初始化Athena客户端、配置S3结果存储路径等。

3. 完整的可运行示例代码 #

给你一个修正后的完整Lambda代码，你可以参考调整：

import boto3

def lambda_handler(event, context):
    # 用三重引号定义多行查询，避免语法问题
    query_1 = """
    SELECT REGEXP_EXTRACT(data,'[a-z]*[0-9]') as datacenter,
           data, pool_id  -- 替换成你实际需要的列
    FROM DATABASE.DATATABLE 
    WHERE pool_id LIKE 'POOL_ID%';
    """
    database_name = "DATABASE"
    # 替换成你的S3存储结果的路径
    s3_output_path = "s3://your-result-bucket/athena-output/"
    
    # 初始化Athena客户端
    athena = boto3.client('athena')
    
    # 发起查询执行请求
    execution_response = athena.start_query_execution(
        QueryString=query_1,
        QueryExecutionContext={
            'Database': database_name
        },
        ResultConfiguration={
            'OutputLocation': s3_output_path
        }
    )
    
    # 返回查询ID，方便后续查看结果状态
    return {
        'statusCode': 200,
        'message': '查询已触发',
        'queryExecutionId': execution_response['QueryExecutionId']
    }

4. 额外检查点 #

• 确认Lambda的Python运行时版本和你本地测试的一致（比如用Python 3.9+，避免版本兼容问题）
• 检查Lambda的执行角色是否有访问Athena和S3的权限（这不是语法错误，但后续可能会遇到权限问题）

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