1 files changed, 47 insertions, 0 deletions
diff --git a/src/utils/log_util.py b/src/utils/log_util.py
new file mode 100644
index 0000000..7c31a32
--- /dev/null
+++ b/src/utils/log_util.py
@@ -0,0 +1,47 @@
+import logging.config
+
+
+# TODO: can't configure queue_handler easily as of 3.11 python only available on 3.12
+logger = logging.getLogger("WorkLogger")
+
+
+def setup_logging(log_lvl_stdout='WARNING') -> None:
+    logging_config: dict = {
+        "version": 1,
+        "disable_existing_loggers": False,
+        "formatters": {
+            "simple": {
+                "format": "%(levelname)s: %(message)s",
+            },
+            "detailed": {
+                "format": "[%(levelname)s|%(module)s|L%(lineno)d] %(asctime)s: %(message)s",
+                "datefmt": "%Y-%m-%dT%H:%M:%S%z",
+            },
+        },
+        "handlers": {
+            "stdout": {
+                "class": "logging.StreamHandler",
+                "level": log_lvl_stdout,
+                "formatter": "simple",
+                "stream": "ext://sys.stdout",
+            },
+            "file": {
+                "class": "logging.handlers.RotatingFileHandler",
+                "level": "DEBUG",
+                "formatter": "detailed",
+                "filename": "logs/WorkLogger.log",
+                "maxBytes": 5000000,  # 5 mb
+                "backupCount": 5,
+            },
+        },
+        "loggers": {
+            "root": {
+                "level": "DEBUG",
+                "handlers": [
+                    "stdout",
+                    "file",
+                ],
+            },
+        },
+    }
+    logging.config.dictConfig(logging_config)