diff options
Diffstat (limited to 'gdpr_obfuscator/write.py')
| -rw-r--r-- | gdpr_obfuscator/write.py | 28 |
1 files changed, 0 insertions, 28 deletions
diff --git a/gdpr_obfuscator/write.py b/gdpr_obfuscator/write.py deleted file mode 100644 index dae66f4..0000000 --- a/gdpr_obfuscator/write.py +++ /dev/null @@ -1,28 +0,0 @@ -import csv -import io -from typing import List, Dict -from .logger import get_logger - -logger = get_logger("CSVWRITER") - - -class DataWriter: - def __init__(self): - pass - - def create_byte_stream(self, data: List[Dict[str, str]]) -> bytes: - if not data: - logger.error("Invalid or empty data was provided to write") - return b"" - - output = io.StringIO() - - headers = list(data[0].keys()) - - writer = csv.DictWriter(output, fieldnames=headers) - writer.writeheader() - writer.writerows(data) - - csv_string = output.getvalue() - - return csv_string.encode("utf-8") |
