aboutsummaryrefslogtreecommitdiffstats
path: root/obfuscator
diff options
context:
space:
mode:
Diffstat (limited to 'obfuscator')
-rw-r--r--obfuscator/__init__.py0
-rw-r--r--obfuscator/csv_reader.py34
-rw-r--r--obfuscator/logger.py24
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
git.ajschof.me — hosted by ajschofield — powered by cgit