From a5696fe5939119d091117b0d81976ac1dd708692 Mon Sep 17 00:00:00 2001 From: Alex Schofield Date: Wed, 12 Feb 2025 01:49:24 +0000 Subject: add logging & move main.py to project root --- main.py | 20 ++++++++++++++++++++ src/main.py | 8 -------- 2 files changed, 20 insertions(+), 8 deletions(-) create mode 100644 main.py delete mode 100644 src/main.py diff --git a/main.py b/main.py new file mode 100644 index 0000000..8ee40e9 --- /dev/null +++ b/main.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) + + diff --git a/src/main.py b/src/main.py deleted file mode 100644 index 23afc66..0000000 --- a/src/main.py +++ /dev/null @@ -1,8 +0,0 @@ -import csv -from io import StringIO -from typing import List, Dict - -def csv_reader(content: str) -> List[Dict[str, str]]: - f = StringIO(content) - reader = csv.DictReader(f) - return list(reader) -- cgit v1.2.3