AWS Lambda Python Code

0

Hi Dears, Hope all is good

I have lambda function written in python (event driven ETL), I use two packaged functions to handel one csv file. I want to use the input csv file (S3 object uploaded to the bucket) in the two functions and put two json files in two different s3 buckets as an output.

But When I used the code as below, It convert the first function(df_out_1) successfully but in second function (df_out_2) give me error message that " No columns to parse from file " for the second function :

The code as below:

def lambda_handler(event, context):

s3_client = boto3.client('s3')
s3 = boto3.resource('s3')

if event:
    try:
        obj = event['Records'][0]
        bucket_name = obj['s3']['bucket']['name']
        file = obj['s3']['object']['key']
        csv_object = s3_client.get_object(Bucket = bucket_name , Key = file)
        csv_content = csv_object['Body'].read()
        bite = io.BytesIO(csv_content)

        df_out_1 = csv_json_function_alex_dep(bite)
        s3.Object('midibus-sagemaker-ai-engine-2' , jsonkey ).put(Body=json.dumps(df_out_1))
        print('The file is converted successfully for model-1 input!')
        
        #change input later to be jsonkey
        df_out_2 = csv_json_function_alex_arri(bite)
        s3.Object('midibus-sagemaker-ai-engine-2',  jsonkey ).put(Body=json.dumps(df_out_2))
        print('The file is converted successfully for model-2 input!')

    
        return {
        'statusCode': 200,
        'body': json.dumps('The file is converted successfully!')
        }
    except BaseException as err:
        print("Oops! Something went wrong ! \\n")
        print(err)
        print(err.args)
        traceback.print_exc(limit=5)

        return {
        'statusCode': 400,
        'body': json.dumps('Something went wrong!')
        }

thanks, Basem

2 回答
1
已接受的回答

Have you tried doing bite = io.BytesIO(csv_content) again before the second function call?

profile pictureAWS
专家
已回答 2 年前
  • Thanks!! It works well, Why we put it again with the second function?

0

If your first function uses bite.read() to get the content of the buffer, the internal buffer cursor will be advanced to the end of the buffer and a subsequent read() call will return an empty buffer. You can reset the cursor to the beginning of the buffer using bite.seek(0). Alternatively, if you can control how the functions you invoke are reading data from the buffer, you could usebite.getvalue().

AWS
专家
已回答 2 年前
  • Thanks for your proposed solution, I used the above answer and it works well now.

您未登录。 登录 发布回答。

一个好的回答可以清楚地解答问题和提供建设性反馈,并能促进提问者的职业发展。

回答问题的准则