From c03897d446352c461790ab82c7a2bae85db17e86 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Mon, 17 Feb 2025 01:06:08 +0000 Subject: add method for reading csv content directly from string --- obfuscator/csv_reader.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'obfuscator') 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] -- cgit v1.2.3