diff options
Diffstat (limited to 'src/transform_lambda.py')
| -rw-r--r-- | src/transform_lambda.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/src/transform_lambda.py b/src/transform_lambda.py index defa15d..2cd9272 100644 --- a/src/transform_lambda.py +++ b/src/transform_lambda.py @@ -5,7 +5,7 @@ import logging import pandas as pd import pyarrow as pa import pyarrow.parquet as pq -from src.dataframes import * +from dataframes import * from botocore.exceptions import ClientError from pg8000.native import Connection, InterfaceError from datetime import datetime @@ -207,5 +207,10 @@ def list_existing_s3_files(bucket_name, client=boto3.client("s3")): except ClientError as e: logger.error(f"Error listing S3 objects: {e}") + raise e return existing_files + + +if __name__ == "__main__": + lambda_handler({}, "") |