Merge pull request #8 from ajschofield/refining-phase

mostly minor changes (fixing things up)

1 files changed, 28 insertions, 0 deletions

diff --git a/obfuscator/write.py b/obfuscator/write.py
new file mode 100644
index 0000000..451b073
--- /dev/null
+++ b/obfuscator/write.py
@@ -0,0 +1,28 @@
+import csv
+import io
+from typing import List, Dict
+from obfuscator.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")