diff options
| -rw-r--r-- | gdpr_obfuscator/read.py | 3 |
1 files changed, 2 insertions, 1 deletions
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 |