|
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)})
|
|
} |