From bc3e976ff9b51ca47fa7ec6465b40e2e41ed6d0d Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Wed, 19 Feb 2025 03:32:58 +0000 Subject: improve error handling in read_s3() --- obfuscator/csv_reader.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'obfuscator/csv_reader.py') diff --git a/obfuscator/csv_reader.py b/obfuscator/csv_reader.py index 3649681..97f67b7 100644 --- a/obfuscator/csv_reader.py +++ b/obfuscator/csv_reader.py @@ -63,6 +63,15 @@ class CSVReader: self.logger.info("S3 object read successfully") content = response["Body"].read().decode("utf-8") return CSVReader.read_string(content) + except client.exceptions.NoSuchKey: + self.logger.error(f"Object not found: {bucket}/{key}") + raise + except client.exceptions.ClientError as e: + self.logger.error(f"Error reading S3 object: {e}") + raise + except UnicodeDecodeError as e: + self.logger.error(f"Error decoding S3 object: {e}") + raise except Exception as e: self.logger.error(f"Error reading S3 object: {e}") raise -- cgit v1.2.3