feat(add logutils):

agentic_security/logutils.py

@@ -0,0 +1,45 @@
+import logging
+from os import getenv
+
+from rich.logging import RichHandler
+
+LOGGER_NAME = "agentic_security"
+
+
+def get_logger(logger_name: str) -> logging.Logger:
+    # https://rich.readthedocs.io/en/latest/reference/logging.html#rich.logging.RichHandler
+    # https://rich.readthedocs.io/en/latest/logging.html#handle-exceptions
+    rich_handler = RichHandler(
+        show_time=False,
+        rich_tracebacks=False,
+        show_path=True if getenv("API_RUNTIME") == "dev" else False,
+        tracebacks_show_locals=False,
+    )
+    rich_handler.setFormatter(
+        logging.Formatter(
+            fmt="%(message)s",
+            datefmt="[%X]",
+        )
+    )
+
+    _logger = logging.getLogger(logger_name)
+    _logger.addHandler(rich_handler)
+    _logger.setLevel(logging.INFO)
+    _logger.propagate = True
+    return _logger
+
+
+logger: logging.Logger = get_logger(LOGGER_NAME)
+
+
+def set_log_level_to_debug():
+    _logger = logging.getLogger(LOGGER_NAME)
+    _logger.setLevel(logging.DEBUG)
+
+
+def set_log_level_to_info():
+    _logger = logging.getLogger(LOGGER_NAME)
+    _logger.setLevel(logging.INFO)
+
+
+set_log_level_to_debug()