Using psycopg2 for PostgreSQL connections via Python on Lambda functions

Download the psycopg2 library from this link https://github.com/jkehler/awslambda-psycopg2. Follow the instructions in the link . Essentially , you'll need to place the correct folder alongside the lamba_function.py file .  

Initially - I was uploading this package into AWS Lambda and getting an error. Then I found some useful details on https://awstip.com/access-an-amazon-rds-database-using-lambda-function-python-293a014d1cd5 .   

The critical information in the link was "rename _psycopg.cpython-39-x86_64-linux-gnu.so to _psycopg.so" .    Once I renamed the file , uploaded the psycopg2 folder - the python script was able to connect ok

Make sure you reconfigure the default timeout value - read more Task timed out after 3.02 seconds aws lambda

This is the Python sample I executed in the AWS Lambda context 

import json
import boto3
import psycopg2

username = "my_username"
password = "my_password"
target_host = "my_aws_db_instance_"
db = "postgres"
target_port = "1234"
  
  
 

def lambda_handler(event, context):
  connection = psycopg2.connect(user=username, password=password, host=target_host, database=db, port=target_port)
  cursor = connection.cursor()
  query = "SELECT version() AS version"
  cursor.execute(query)
  results = cursor.fetchone()
  cursor.close()
  connection.commit()
  return results