wrap write functions in class and update references

1 files changed, 4 insertions, 2 deletions

diff --git a/cli.py b/cli.py
index 5100e2b..bd49707 100644
--- a/cli.py
+++ b/cli.py
@@ -1,8 +1,8 @@
 import argparse
 from obfuscator.read import DataReader
+from obfuscator.write import DataWriter
 from obfuscator.obfuscate import obfuscate
 from obfuscator.logger import get_logger
-from obfuscator.csv_writer import create_byte_stream
 
 
 def main():
@@ -43,7 +43,9 @@ def main():
 
     obfuscated_data = obfuscate(data, args.pii)
 
-    return create_byte_stream(obfuscated_data)
+    writer = DataWriter()
+
+    return writer.create_byte_stream(obfuscated_data)
 
 
 if __name__ == "__main__":