From 5e125e6c561c81526c2f4c4c7895fe720ba64d90 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Wed, 12 Feb 2025 01:56:14 +0000 Subject: re-organise project structure & update pyproject.toml --- obfuscator/csv_reader.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 obfuscator/csv_reader.py (limited to 'obfuscator/csv_reader.py') diff --git a/obfuscator/csv_reader.py b/obfuscator/csv_reader.py new file mode 100644 index 0000000..8ee40e9 --- /dev/null +++ b/obfuscator/csv_reader.py @@ -0,0 +1,20 @@ +import csv +from io import StringIO +from typing import List, Dict +import logging + +logger = logging.getLogger(__name__) +logger.setLevel(logging.INFO) +if not logger.handlers: + handler = logging.StreamHandler() + formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s') + handler.setFormatter(formatter) + logger.addHandler(handler) + +def csv_reader(content: str) -> List[Dict[str, str]]: + f = StringIO(content) + reader = csv.DictReader(f) + logger.info("Finished reading CSV!") + return list(reader) + + -- cgit v1.2.3