aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorAlex Schofield <git@ajschof.me>2025-02-19 03:27:54 +0000
committerAlex Schofield <git@ajschof.me>2025-02-19 03:27:54 +0000
commitad0328b2f292fe438a8a6a1f7ff2d36856dc578d (patch)
tree2c1a19ca1bfd81071d8b0246eb8200c70b9bdec4
parent3c4b66e8490c6fdf93fb8fee735d52c76eb2853b (diff)
downloadgdpr-obfuscator-ad0328b2f292fe438a8a6a1f7ff2d36856dc578d.tar.gz
gdpr-obfuscator-ad0328b2f292fe438a8a6a1f7ff2d36856dc578d.zip
modify logger messages to be more clear
-rw-r--r--cli.py10
-rw-r--r--obfuscator/csv_reader.py4
-rw-r--r--obfuscator/csv_writer.py4
-rw-r--r--obfuscator/obfuscate.py2
4 files changed, 8 insertions, 12 deletions
diff --git a/cli.py b/cli.py
index c03ffd1..40c777f 100644
--- a/cli.py
+++ b/cli.py
@@ -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 [
git.ajschof.me — hosted by ajschofield — powered by cgit