Use logger for all sorts of logging. Also prepare for writing some logs to a database for audit reasons.feature.scanners

1 files changed, 26 insertions, 0 deletions

diff --git a/src/log.py b/src/log.py
new file mode 100644
index 0000000..ec63106
--- /dev/null
+++ b/src/log.py
@@ -0,0 +1,26 @@
+import logging
+
+import uvicorn
+
+log_format = "%(levelprefix)s %(asctime)s | %(message)s"
+
+
+def get_handler():
+    formatter = uvicorn.logging.DefaultFormatter(log_format)
+    handler = logging.StreamHandler()
+    handler.setFormatter(formatter)
+
+    return handler
+
+
+def get_logger(name=None):
+    logger = logging.getLogger(name)
+    if logger.handlers:
+        return logger
+
+    logger = logging.getLogger(name)
+    handler = get_handler()
+    logger.addHandler(handler)
+    logger.setLevel(logging.DEBUG)
+
+    return logger