Add function to setup logging for scripts.

1 files changed, 18 insertions, 0 deletions

diff --git a/scripts/cli/logging.py b/scripts/cli/logging.py
new file mode 100644
index 0000000..30ecf17
--- /dev/null
+++ b/scripts/cli/logging.py
@@ -0,0 +1,18 @@
+"""Logging for scripts."""
+import logging
+
+def setup_logging(
+        script_logger: logging.Logger,
+        loglevel: str,
+        modules: tuple[str, ...] = tuple()
+):
+    """Setup module-level loggers to the same log-level as the application."""
+    logging.basicConfig(
+        encoding="utf-8",
+        format=("%(asctime)s — %(filename)s:%(lineno)s — %(levelname)s: "
+                "%(message)s"),
+        level=logging.INFO)
+    script_logger.setLevel(getattr(logging, loglevel.upper()))
+    effective_loglevel = logging.getLevelName(script_logger.getEffectiveLevel())
+    for module in modules:
+        logging.getLogger(module).setLevel(effective_loglevel)