diff options
| -rw-r--r-- | main.py | 20 | ||||
| -rw-r--r-- | src/main.py | 8 |
2 files changed, 20 insertions, 8 deletions
@@ -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) |
