diff options
| -rw-r--r-- | src/extract_lambda.py | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/src/extract_lambda.py b/src/extract_lambda.py index 24f0981..0e6dd8c 100644 --- a/src/extract_lambda.py +++ b/src/extract_lambda.py @@ -99,7 +99,9 @@ def connect_to_database() -> Connection: raise DBConnectionException("Failed to connect to database") -def extract_bucket(client=boto3.client("s3")): +def extract_bucket(client=None): + if client is None: + client = boto3.client("s3") response = client.list_buckets() extract_bucket_filter = [ bucket["Name"] for bucket in response["Buckets"] if "extract" in bucket["Name"] @@ -108,11 +110,16 @@ def extract_bucket(client=boto3.client("s3")): return extract_bucket_filter[0] -def list_existing_s3_files(bucket_name=extract_bucket(), client=boto3.client("s3")): +def list_existing_s3_files(bucket_name=None, client=None): """Creates a dictionary and populates it with the results of listing the contents of the s3 bucket, then returns the populated dictionary """ + if client is None: + client = boto3.client("s3") + if bucket_name is None: + bucket_name = extract_bucket(client) + logging.info("Listing existing S3 files") existing_files = {} |