From 27f89b78775f9b6fd8d3d560689c53db2beb1b64 Mon Sep 17 00:00:00 2001 From: Ellie Date: Tue, 27 Aug 2024 16:39:38 +0100 Subject: add logger error to lambda handler --- src/load_lambda.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) (limited to 'src') diff --git a/src/load_lambda.py b/src/load_lambda.py index 39fa27d..9e15af3 100644 --- a/src/load_lambda.py +++ b/src/load_lambda.py @@ -5,6 +5,7 @@ import pyarrow.parquet as pq from io import BytesIO import logging import json +import traceback from sqlalchemy import create_engine @@ -28,8 +29,7 @@ def lambda_handler(event, context): "statusCode": 200, "body": json.dumps("No dataframes were uploaded."), } - - if uploaded_tables["uploaded"]: + elif uploaded_tables["uploaded"]: return { "statusCode": 200, "body": json.dumps( @@ -37,10 +37,12 @@ def lambda_handler(event, context): {uploaded_tables["uploaded"]} .""" ), } - + else: + logger.error(f"error") + return {"error"} except Exception as e: - logger.error(f"Error: {e}", exc_info=True) - return {"statusCode": 500, "body": json.dumps("Internal server error.")} + logger.error({e}) + return {"statusCode": 500, "body": {e}} def retrieve_secrets(client=None, secret_name=None): -- cgit v1.2.3