Psycopg2 Aurora Connection Example

connection-sample.py

import psycopg2
Import os

DB_HOST = os.getenv("DB_HOST")
DB_PORT = os.getenv("DB_PORT")
DB_USER = os.getenv("DB_USER")
DB_PW = os.getenv("DB_PW")
DB_NAME = os.getenv("DB_NAME")

def lambda_handler(event, context):

    global db_conn
    try:
	  # use very short connect_timeout!
        db_conn = psycopg2.connect(host=DB_HOST, port=DB_PORT,
                                 user=DB_USER, password=DB_PW,
                                 database=DB_NAME, connect_timeout=2)
    except Exception as err:
        logger.exception("Could not connect to database! %s", err)
        # errors result in lambda failures to apply retry strategies
        sys.exit()
	
    with db_conn.cursor() as cursor:
            cursor.execute("SELECT 1")
            result = cursor.fetchone()
            print(result)