How to return io.BufferedReader through AWS Lambda(python)?

user_11567961 0
Viewed 315

I am trying to read a file from the S3 bucket, get the file in an io.BufferedReader and return it in via lambda function that would later decode to JSON. I am getting an error message as

Unable to marshal response: Object of type BufferedReader is not JSON serializable

My code is mentioned below.

s3 = boto3.client('s3')
def lambda_handler(event, context):
    bucket = "bucketID"
    body = []
    for record in event['key']:
        with open('/tmp/2021-10-11T06:23:29:691472.pdf', 'wb') as f:
            s3.download_fileobj(bucket, "basedir/file.pdf", f)
        f = open("/tmp/2021-10-11T06:23:29:691472.pdf", "rb")
    body.append(f)
    
    return {
        "statusCode": 200,
        "file":f,
        "content":f.read()
    }

Error Response from lambda

Response
{
  "errorMessage": "Unable to marshal response: Object of type BufferedReader is not JSON serializable",
  "errorType": "Runtime.MarshalError",
  "requestId": "10aea120-kyc-jpk-bnce-7123eTyda64",
  "stackTrace": []
}

I am using the AWS-Lambda Python function

aws-lambda amazon-web-services amazon-s3 python
1 Answers

f.read() return bytes and JSON does not support binary data. Also "file":f is incorrect. Guess it should be a filename. Anyway, usually you would return binary data in JSON as base64:

import base64

s3 = boto3.client('s3')
def lambda_handler(event, context):
    bucket = "document.as.a.service.test"
    body = []
    for record in event['uuid_filepath']:
        with open('/tmp/2021-10-11T06:23:29:691472.pdf', 'wb') as f:
            s3.download_fileobj(bucket, "123TfZ/2021-10-11T06:23:29:691472.pdf", f)
        f = open("/tmp/2021-10-11T06:23:29:691472.pdf", "rb")
    body.append(f)
    
    return {
        "statusCode": 200,
        "file": "2021-10-11T06:23:29:691472.pdf",
        "content": base64.b64encode(f.read())
    }

Then on the client side, you have to decode base64 to binary.

user_2488230
Related