File size: 944 Bytes
d155f3f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
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)})
} |