From 4be9585d0d56046a486e58b4c5b559cabe9f5636 Mon Sep 17 00:00:00 2001
From: Kristofer Hallin <kristofer@sunet.se>
Date: Wed, 13 Apr 2022 08:39:55 +0200
Subject: Use logger for all sorts of logging. Also prepare for writing some
 logs to a database for audit reasons.

---
 src/log.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 src/log.py

(limited to 'src/log.py')

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
-- 
cgit v1.1