def db_conn(): # Pass setup_container
return Postgres()
import library.db_queries as db_queries
from library.exceptions import DatabaseReturnedNone
from library.db_connections import Postgres, get_conn
from library.utils import awshandler, aws_get_path_parameter
conn = Postgres()
@awshandler
def histories(event, context):
organization_id = event["pathParameters"]["organization_id"]
history_list = db_queries.get_histories(conn, organization_id)
return [hist.to_dict() for hist in history_list]
@awshandler
def get_history(event, context):
organization_id = aws_get_path_parameter(event, "organization_id")
history_id = aws_get_path_parameter(event, "history_id")
history = db_queries.get_history(connection=conn, organization_id=organization_id, history_id=history_id)
if history is None:
raise DatabaseReturnedNone(f"Check object id: {history_id}")
return history.to_dict()