diff options
| -rw-r--r-- | cli.py | 4 | ||||
| -rw-r--r-- | obfuscator/csv_reader.py | 14 |
2 files changed, 9 insertions, 9 deletions
@@ -14,8 +14,8 @@ def main(): if args.local and not args.s3: logger.debug("User chose to read CSV from local path") - reader = CSVReader(args.local) - data = reader.read_local() + reader = CSVReader() + data = reader.read_local(args.local) print(data) else: logger.debug("User chose to read CSV from S3") diff --git a/obfuscator/csv_reader.py b/obfuscator/csv_reader.py index 42e8f8a..901d396 100644 --- a/obfuscator/csv_reader.py +++ b/obfuscator/csv_reader.py @@ -1,24 +1,24 @@ import csv from typing import List, Dict -from logger import get_logger +from obfuscator.logger import get_logger logger = get_logger("CSVReader") class CSVReader: - def __init__(self, path: str): - self.path = path + def __init__(self): + pass - def read_local(self) -> List[Dict[str, str]]: - logger.debug(f"Reading local CSV from: {self.path}") + def read_local(self, path) -> List[Dict[str, str]]: + logger.debug(f"Reading local CSV from: {path}") data = [] try: - with open(self.path, mode="r", encoding="utf-8") as file: + with open(path, mode="r", encoding="utf-8") as file: reader = csv.DictReader(file) for row in reader: data.append(dict(row)) except FileNotFoundError: - logger.error(f"File not found: {self.path}") + logger.error(f"File not found: {path}") except Exception as e: logger.error(f"Error reading file: {e}") |