Merge pull request #1 from ajschofield/feat/csv-reader

merge csvreader into stable

3 files changed, 58 insertions, 0 deletions

diff --git a/obfuscator/__init__.py b/obfuscator/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/obfuscator/__init__.py
diff --git a/obfuscator/csv_reader.py b/obfuscator/csv_reader.py
new file mode 100644
index 0000000..b9dccdb
--- /dev/null
+++ b/obfuscator/csv_reader.py
@@ -0,0 +1,34 @@
+import csv
+import io
+from typing import List, Dict
+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}")
+
+        try:
+            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}")
+
+    @staticmethod
+    def read_s3(path) -> List[Dict[str, str]]:
+        return []
+
+    @staticmethod
+    def read_string(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]
diff --git a/obfuscator/logger.py b/obfuscator/logger.py
new file mode 100644
index 0000000..ca41e95
--- /dev/null
+++ b/obfuscator/logger.py
@@ -0,0 +1,24 @@
+import logging
+import os
+
+
+def get_logger(name: str) -> logging.Logger:
+    logger = logging.getLogger(name)
+
+    if not logger.hasHandlers():
+        if os.getenv("DEBUG", "FALSE").upper() == "TRUE":
+            log_level = logging.DEBUG
+        else:
+            log_level = logging.INFO
+
+        logger.setLevel(log_level)
+
+        handler = logging.StreamHandler()
+        formatting = logging.Formatter(
+            "%(asctime)s - %(levelname)s - %(name)s - %(message)s"
+        )
+        handler.setFormatter(formatting)
+
+        logger.addHandler(handler)
+
+    return logger