AxL95's picture
Upload 35 files
d155f3f verified
raw
history blame contribute delete
944 Bytes
import json
import psycopg2
from psycopg2 import OperationalError
import os
def handler(event, context):
try:
conn = psycopg2.connect(
host=os.getenv('DB_HOST'),
database=os.getenv('DB_NAME'),
user=os.getenv('DB_USER'),
password=os.getenv('DB_PASSWORD'),
port=os.getenv('DB_PORT'),
sslmode='require'
)
cur = conn.cursor()
cur.execute('SELECT * FROM users;')
users = cur.fetchall()
cur.close()
conn.close()
return {
'statusCode': 200,
'body': json.dumps(users)
}
except OperationalError as e:
return {
'statusCode': 500,
'body': json.dumps({"error": str(e)})
}
except Exception as e:
return {
'statusCode': 500,
'body': json.dumps({"error": str(e)})
}