cleanup logging

dnstapir · Oct 21, 2024 · 667f8f5 · 667f8f5
1 parent 0533f67
commit 667f8f5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/dnstapir/logging.py b/dnstapir/logging.py
@@ -1,3 +1,4 @@
+import logging.config
 from datetime import datetime
 
 from jsonformatter import JsonFormatter as _JsonFormatter
@@ -42,3 +43,7 @@ class JsonFormatter(_JsonFormatter):
     def formatTime(self, record, datefmt=None):
         dt = datetime.fromtimestamp(record.created).astimezone()
         return dt.strftime(TIMESTAMP_FORMAT)
+
+
+def configure_json_logging():
+    logging.config.dictConfig(LOGGING_CONFIG_JSON)
diff --git a/tests/test_logging.py b/tests/test_logging.py
@@ -1,11 +1,9 @@
 import logging
-import logging.config
 
-from dnstapir.logging import LOGGING_CONFIG_JSON
+from dnstapir.logging import configure_json_logging
 
 
 def test_logging():
+    configure_json_logging()
     logger = logging.getLogger(__name__)
-    logging_config = LOGGING_CONFIG_JSON
-    logging.config.dictConfig(logging_config)
-    logger.warning("Hello")
+    logger.warning("Hello world")