Problema ao executar o script Python na função AWS Lambda

Aug 26 2020

Tentei executar o script abaixo no AWS Lambda, mas, infelizmente, não foi executado com sucesso. Alguém pode me ajudar a conseguir essa correção ou me corrigir se houver um problema com o script que precisa de alguma alteração para executá-lo no Lambda?

#!/usr/bin/env python3
import boto3
client = boto3.client('athena')
def run_query(query, database, s3_output):
    response = client.start_query_execution(
        QueryString=query,
        QueryExecutionContext={
            'Database': 'database'
            },
        ResultConfiguration={
            'OutputLocation': s3_output,
            }
        )
    print('Execution ID: ' + response['QueryExecutionId'])
    return response     
  
#Athena configuration
s3_input = 's3://smathena/cf-ant-prod/'
s3_ouput = 's3://smathena/athenatest/'
database = 's3_accesslog'
table = 'test_output1'

#Athena database and table definition
create_database = "CREATE DATABASE IF NOT EXISTS %s;" % (database)
delete_table = "drop table %s.%s;" % ( database, table )
create_table = \
  """CREATE EXTERNAL TABLE IF NOT EXISTS %s.%s (
  `Date` DATE,
   ScContentLen BIGINT,
   ScRangeStart BIGINT,
   ScRangeEnd BIGINT
   )
   ROW FORMAT DELIMITED FIELDS TERMINATED BY '\t'
   LOCATION '%s'
   TBLPROPERTIES ('skip.header.line.count' = '2');""" % ( database, table, s3_input )

#Query definitions
query_1 = "SELECT * FROM %s.%s where CAST(status AS VARCHAR) like '404';" % (database, table)

#Execute all queries
queries = [ create_database, delete_table, create_table, query_1 ]
for q in queries:
   print("Executing query: %s" % (q))
   res = 'run_query(q, database, s3_ouput)'

Erro ao testar no AWS Lambda:

  Response:
{
  "errorMessage": "run_query() missing 1 required positional argument: 's3_output'",
  "errorType": "TypeError",
  "stackTrace": [
    "  File \"/var/runtime/bootstrap.py\", line 131, in handle_event_request\n    response = request_handler(event, lambda_context)\n"
  ]
}

ID do pedido: "2cb2175c-8838-470d-a8dd-efdf4c051312"

Registros de funções: START RequestId: 2cb2175c-8838-470d-a8dd-efdf4c051312 Versão: $ LATEST [ERROR] TypeError: run_query () ausente 1 argumento posicional obrigatório: 's3_output' Traceback (última chamada mais recente): Arquivo "/ var / runtime /bootstrap.py ", linha 131, em handle_event_request response = request_handler (event, lambda_context) END RequestId: 2cb2175c-8838-470d-a8dd-efdf4c051312

import boto3
client = boto3.client('athena')
def run_query(event, context):
    response = client.start_query_execution(
        QueryString=query,
        QueryExecutionContext={
            'Database': 'database'
            },
        ResultConfiguration={
            'OutputLocation': s3_output,
            }
        )
    print('Execution ID: ' + response['QueryExecutionId'])
    return event  

Obtendo o erro abaixo:

START RequestId: 55dbf703-f30c-4106-8873-c685f3d06e4d Versão: $ LATEST [ERROR] NameError: o nome 's3_output' não está definido Traceback (última chamada mais recente): Arquivo "/var/task/lambda_function.py", linha 12 , em run_query 'OutputLocation': s3_output, END RequestId: 55dbf703-f30c-4106-8873-c685f3d06e4d REPORT RequestId: 55dbf703-f30c-4106-8873-c685f3d06e4d Duração: 14,30 ms Duração faturada: 128 100 ms Tamanho máximo da memória: : 75 MB Duração inicial: 649,66 ms

Respostas

JohnRotenstein Aug 27 2020 at 11:06

Este é o mesmo problema da sua pergunta anterior: Erro ao executar um script Python no AWS Lambda - Stack Overflow

Uma função AWS Lambda tem este formato:

import boto3

def lambda_handler(event, context):
    
    print(event)
    print(context)

Quando a função Lambda é chamada, ela chama a lambda_handler()função. O nome desta função pode ser alterado se desejado, mas sempre receberá esses dois parâmetros de entrada: eventecontext

Dependendo de como a função Lambda é chamada, o conteúdo do eventcontém informações que são "passadas" para a função Lambda. Por exemplo:

  • Se a função for chamada do Amazon S3, ela conterá o nome do intervalo e o objeto que acionou o evento
  • Se a função for chamada do Amazon SNS, ela conterá a mensagem que foi enviada ao tópico SNS

Parece que você deseja usar três valores na função: query, database, s3_output

Esses valores precisarão ser passados ​​para a função Lambda por meio de qualquer meio que a função Lambda esteja sendo chamada. A função pode então recuperar esses valores por meio do eventparâmetro.