add logging & move main.py to project root

1 files changed, 20 insertions, 0 deletions

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)
+
+