add method for reading csv content directly from string

1 files changed, 9 insertions, 0 deletions

diff --git a/obfuscator/csv_reader.py b/obfuscator/csv_reader.py
index cbd18c1..cf81a30 100644
--- a/obfuscator/csv_reader.py
+++ b/obfuscator/csv_reader.py
@@ -1,4 +1,5 @@
 import csv
+import io
 from typing import List, Dict
 from obfuscator.logger import get_logger
 
@@ -28,3 +29,11 @@ class CSVReader:
 
     def read_s3(self) -> List[Dict[str, str]]:
         return []
+    
+    def read_string(self, content: str) -> List[Dict[str, str]]:
+        if not content.strip():
+            return []
+        
+        f = io.StringIO(content)
+        reader = csv.DictReader(f)
+        return [dict(row) for row in reader]