aboutsummaryrefslogtreecommitdiffstats
path: root/obfuscator
diff options
context:
space:
mode:
Diffstat (limited to 'obfuscator')
-rw-r--r--obfuscator/csv_reader.py12
1 files changed, 3 insertions, 9 deletions
diff --git a/obfuscator/csv_reader.py b/obfuscator/csv_reader.py
index 55fb892..3b4496b 100644
--- a/obfuscator/csv_reader.py
+++ b/obfuscator/csv_reader.py
@@ -5,25 +5,19 @@ from obfuscator.logger import get_logger
logger = get_logger("CSVReader")
-
class CSVReader:
@staticmethod
def read_local(path) -> List[Dict[str, str]]:
logger.debug(f"Reading local CSV from: {path}")
- data = []
try:
- with open(path, mode="r", encoding="utf-8") as file:
- reader = csv.DictReader(file)
- for row in reader:
- data.append(dict(row))
+ with open(path, mode="r", encoding="utf-8") as f:
+ reader = csv.DictReader(f)
+ return [dict(row) for row in reader]
except FileNotFoundError:
logger.error(f"File not found: {path}")
except Exception as e:
logger.error(f"Error reading file: {e}")
-
- logger.debug(f"Total rows read: {len(data)}")
- return data
@staticmethod
def read_s3(path) -> List[Dict[str, str]]:
git.ajschof.me — hosted by ajschofield — powered by cgit