diff options
| author | Alex Schofield <git@ajschof.me> | 2025-02-19 03:27:54 +0000 |
|---|---|---|
| committer | Alex Schofield <git@ajschof.me> | 2025-02-19 03:27:54 +0000 |
| commit | ad0328b2f292fe438a8a6a1f7ff2d36856dc578d (patch) | |
| tree | 2c1a19ca1bfd81071d8b0246eb8200c70b9bdec4 | |
| parent | 3c4b66e8490c6fdf93fb8fee735d52c76eb2853b (diff) | |
| download | gdpr-obfuscator-ad0328b2f292fe438a8a6a1f7ff2d36856dc578d.tar.gz gdpr-obfuscator-ad0328b2f292fe438a8a6a1f7ff2d36856dc578d.zip | |
modify logger messages to be more clear
| -rw-r--r-- | cli.py | 10 | ||||
| -rw-r--r-- | obfuscator/csv_reader.py | 4 | ||||
| -rw-r--r-- | obfuscator/csv_writer.py | 4 | ||||
| -rw-r--r-- | obfuscator/obfuscate.py | 2 |
4 files changed, 8 insertions, 12 deletions
@@ -31,23 +31,19 @@ def main(): args = parser.parse_args() log_level = "DEBUG" if args.verbose else "INFO" - logger = get_logger("CLI", log_level) reader = CSVReader(log_level) if args.local and not args.s3: - logger.debug("User chose to read CSV from local path") + logger.debug("Read data from local path") data = reader.read_local(args.local) - logger.debug("Contents: " + str(data)) else: - logger.debug("User chose to read CSV from S3") - + logger.debug("Read data from S3") data = reader.read_s3(args.s3) - logger.debug("Contents: " + str(data)) obfuscated_data = obfuscate(data, args.pii) - logger.debug("Obfuscated data (JSON): " + json.dumps(obfuscated_data, indent=4)) + return create_byte_stream(obfuscated_data) diff --git a/obfuscator/csv_reader.py b/obfuscator/csv_reader.py index 2b099c8..3649681 100644 --- a/obfuscator/csv_reader.py +++ b/obfuscator/csv_reader.py @@ -45,7 +45,9 @@ class CSVReader: if os.getenv("LOCALSTACK", "FALSE").upper() == "TRUE": localstack_endpoint = "http://localhost.localstack.cloud:4566" - self.logger.debug("Using LocalStack endpoint for S3") + self.logger.debug( + "Using LocalStack endpoint for S3 - ensure LocalStack is running" + ) client = boto3.client( "s3", endpoint_url=localstack_endpoint, diff --git a/obfuscator/csv_writer.py b/obfuscator/csv_writer.py index 56b3f1f..de7cd4b 100644 --- a/obfuscator/csv_writer.py +++ b/obfuscator/csv_writer.py @@ -8,8 +8,7 @@ logger = get_logger("CSVWRITER") def create_byte_stream(data: List[Dict[str, str]]) -> bytes: if not data: - logger.info("No valid data was provided to write") - return b"" + logger.error("Invalid or empty data was provided to write") output = io.StringIO() @@ -20,6 +19,5 @@ def create_byte_stream(data: List[Dict[str, str]]) -> bytes: writer.writerows(data) csv_string = output.getvalue() - logger.debug(f"CSV data: {csv_string}") return csv_string.encode("utf-8") diff --git a/obfuscator/obfuscate.py b/obfuscator/obfuscate.py index e964433..9d43975 100644 --- a/obfuscator/obfuscate.py +++ b/obfuscator/obfuscate.py @@ -12,7 +12,7 @@ def obfuscate( sensitive values with a string of asterisks. """ if not data: - logger.info("No valid data was provided to obfuscate") + logger.error("Invalid or empty was provided to obfuscate") return [] return [ |
