aboutsummaryrefslogtreecommitdiffstats
path: root/obfuscator
diff options
context:
space:
mode:
authorAlex Schofield <git@ajschof.me>2025-02-19 03:50:09 +0000
committerAlex Schofield <git@ajschof.me>2025-02-19 03:50:09 +0000
commit422acef7a0762089298e9eae9944877e788fd94d (patch)
tree3538d69a19e20dad074a1a8600df08d82448a7b4 /obfuscator
parent648c3785071adfae163e7ac1321f19188a5e4921 (diff)
downloadgdpr-obfuscator-422acef7a0762089298e9eae9944877e788fd94d.tar.gz
gdpr-obfuscator-422acef7a0762089298e9eae9944877e788fd94d.zip
fix some tests (some are still broken)
Diffstat (limited to 'obfuscator')
-rw-r--r--obfuscator/csv_writer.py1
-rw-r--r--obfuscator/obfuscate.py4
2 files changed, 2 insertions, 3 deletions
diff --git a/obfuscator/csv_writer.py b/obfuscator/csv_writer.py
index 2bff6e0..de7cd4b 100644
--- a/obfuscator/csv_writer.py
+++ b/obfuscator/csv_writer.py
@@ -9,7 +9,6 @@ logger = get_logger("CSVWRITER")
def create_byte_stream(data: List[Dict[str, str]]) -> bytes:
if not data:
logger.error("Invalid or empty data was provided to write")
- raise
output = io.StringIO()
diff --git a/obfuscator/obfuscate.py b/obfuscator/obfuscate.py
index 73448ce..9cd3a03 100644
--- a/obfuscator/obfuscate.py
+++ b/obfuscator/obfuscate.py
@@ -13,10 +13,10 @@ def obfuscate(
"""
if not data:
logger.error("Invalid or empty was provided to obfuscate")
- raise
+ raise ValueError("Invalid data provided to obfuscate")
if not pii_fields:
logger.error("No PII fields provided to obfuscate")
- raise
+ raise ValueError("No PII fields provided to obfuscate")
return [
{k: ("***" if k in pii_fields else v) for k, v in record.items()}
git.ajschof.me — hosted by ajschofield — powered by cgit