create byte stream within main process_s3 & process_local methods

1 files changed, 6 insertions, 2 deletions

diff --git a/gdpr_obfuscator/__init__.py b/gdpr_obfuscator/__init__.py
index 38680e8..6de7145 100644
--- a/gdpr_obfuscator/__init__.py
+++ b/gdpr_obfuscator/__init__.py
@@ -1,14 +1,18 @@
 from .read import DataReader
 from .obfuscate import obfuscate_data
+from .utils import Utilities
 from typing import List, Dict
 
 
 class Obfuscator:
     def __init__(self):
         self.reader = DataReader()
+        self.utils = Utilities()
 
     def process_s3(self, path: str, pii_fields: List[str]) -> bytes:
-        return obfuscate_data(self.reader.read_s3(path), pii_fields)
+        obfuscated_data = obfuscate_data(self.reader.read_s3(path), pii_fields)
+        return self.utils.create_byte_stream(obfuscated_data)
 
     def process_local(self, path: str, pii_fields: List[str]) -> bytes:
-        return obfuscate_data(self.reader.read_local(path), pii_fields)
+        obfuscated_data = obfuscate_data(self.reader.read_local(path), pii_fields)
+        return self.utils.create_byte_stream(obfuscated_data)