From 04157bd1736b4997f549be8b79713448c2cceea3 Mon Sep 17 00:00:00 2001
From: Alex Schofield <git@ajschof.me>
Date: Thu, 20 Feb 2025 19:16:07 +0000
Subject: return byte stream from read_s3 using create_byte_stream method

---
 gdpr_obfuscator/read.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'gdpr_obfuscator/read.py')

diff --git a/gdpr_obfuscator/read.py b/gdpr_obfuscator/read.py
index b1ffc2f..8dd3e25 100644
--- a/gdpr_obfuscator/read.py
+++ b/gdpr_obfuscator/read.py
@@ -49,7 +49,8 @@ class DataReader:
             response = client.get_object(Bucket=bucket, Key=key)
             self.logger.info("S3 object read successfully")
             content = response["Body"].read().decode("utf-8")
-            return self.read_string(content)
+            read_csv_content = self.read_string(content)
+            return utils.create_byte_stream(read_csv_content)
         except client.exceptions.NoSuchKey:
             self.logger.error(f"Object not found: {bucket}/{key}")
             raise
-- 
cgit v1.2.3