diff options
| -rw-r--r-- | obfuscator/csv_reader.py | 9 |
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] |