experiment: testing noxfile banner contextmanager

Author: chalmerlowe

packages/bigframes/noxfile.py

@@ -17,13 +17,14 @@
 from __future__ import absolute_import
 
 import argparse
+import contextlib
 import multiprocessing
 import os
 import pathlib
 import re
 import shutil
 import time
-from typing import Dict, List
+from typing import Dict, Generator, List
 
 import nox
 import nox.sessions
@@ -135,6 +136,25 @@
 # Error if a python version is missing
 nox.options.error_on_missing_interpreters = True
 
+@contextlib.contextmanager
+def log_package_context(session: nox.Session) -> Generator[None, None, None]:
+    """Logs a highly visible package context banner right before a session exits.
+
+    Ensures metadata is printed adjacent to Nox's final status log,
+    even if the session fails or raises an exception.
+    """
+    # Dynamically extract current folder name (e.g., 'bigframes')
+    # Falls back to the root directory name if run from the repo root
+    package_name = os.path.basename(os.getcwd())
+
+    try:
+        # Hands control back to the session code block
+        yield
+    finally:
+        # This executes AFTER test output finishes, immediately above Nox's summary line
+        banner_text = f" Finished session for package: {package_name.upper()} "
+        session.log(banner_text)
+
 
 @nox.session(python=DEFAULT_PYTHON_VERSION)
 def lint(session):
@@ -1086,14 +1106,16 @@ def mypy(session):
 
     session.install(*deps)
     shutil.rmtree(".mypy_cache", ignore_errors=True)
-    session.log("--- Running mypy (DINOSAUR) ---")
-    session.run(
-        "mypy",
-        f"--config-file={MYPY_CONFIG_FILE}",
-        "bigframes",
-        os.path.join("tests", "system"),
-        os.path.join("tests", "unit"),
-        "--check-untyped-defs",
-        "--explicit-package-bases",
-        '--exclude="^third_party"',
-    )
+
+    # Wrap runtime execution steps to inject package name logging
+    with log_package_context(session):
+        session.run(
+            "mypy",
+            f"--config-file={MYPY_CONFIG_FILE}",
+            "bigframes",
+            os.path.join("tests", "system"),
+            os.path.join("tests", "unit"),
+            "--check-untyped-defs",
+            "--explicit-package-bases",
+            '--exclude="^third_party"',
+        )