add logging & move main.py to project root

2 files changed, 20 insertions, 8 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)
+
+
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)