rename write function & update references

1 files changed, 23 insertions, 0 deletions

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