Refactoring: Shuffle logging utils around

Author: joshuatz

django_utils_lib/logger.py

@@ -1,34 +1,6 @@
 import logging
-from os import get_terminal_size
-from typing import List, Union
 
 from django_utils_lib.constants import PACKAGE_NAME
 
 pkg_logger = logging.getLogger(PACKAGE_NAME)
 pkg_logger.setLevel(logging.INFO)
-
-
-def build_heading_block(heading: Union[str, List[str]], border_width=2) -> str:
-    """
-    Generate a heading, like:
-    ```
-    ===============
-    ==   Hello   ==
-    ===============
-    ```
-    """
-    terminal_width = get_terminal_size().columns
-    heading_delim = "".ljust(terminal_width, "=")
-    heading_lines = [heading_delim]
-
-    # Leave space for border, plus one space on each side
-    border = "=" * border_width
-    border_width_total = border_width + 1
-
-    for line in heading if isinstance(heading, list) else heading.splitlines():
-        left_inner_padding = int((terminal_width - len(line) - (border_width_total * 2)) / 2)
-        right_inner_padding = int(terminal_width - left_inner_padding - len(line) - (border_width_total * 2))
-        heading_lines.append(f"{border} {' ' * left_inner_padding}{line}{' ' * right_inner_padding} {border}")
-    heading_lines.append(heading_delim)
-
-    return "\n".join(heading_lines)

django_utils_lib/logging_utils.py

@@ -0,0 +1,28 @@
+from os import get_terminal_size
+from typing import List, Union
+
+
+def build_heading_block(heading: Union[str, List[str]], border_width=2) -> str:
+    """
+    Generate a heading, like:
+    ```
+    ===============
+    ==   Hello   ==
+    ===============
+    ```
+    """
+    terminal_width = get_terminal_size().columns
+    heading_delim = "".ljust(terminal_width, "=")
+    heading_lines = [heading_delim]
+
+    # Leave space for border, plus one space on each side
+    border = "=" * border_width
+    border_width_total = border_width + 1
+
+    for line in heading if isinstance(heading, list) else heading.splitlines():
+        left_inner_padding = int((terminal_width - len(line) - (border_width_total * 2)) / 2)
+        right_inner_padding = int(terminal_width - left_inner_padding - len(line) - (border_width_total * 2))
+        heading_lines.append(f"{border} {' ' * left_inner_padding}{line}{' ' * right_inner_padding} {border}")
+    heading_lines.append(heading_delim)
+
+    return "\n".join(heading_lines)

django_utils_lib/testing/pytest_plugin.py

@@ -25,7 +25,8 @@
 from typing_extensions import NotRequired, TypedDict
 
 from django_utils_lib.constants import PACKAGE_NAME
-from django_utils_lib.logger import build_heading_block, pkg_logger
+from django_utils_lib.logger import pkg_logger
+from django_utils_lib.logging_utils import build_heading_block
 from django_utils_lib.testing.utils import PytestNodeID, is_main_pytest_runner, validate_requirement_tagging
 
 BASE_DIR = Path(__file__).resolve().parent