Get warning line same as in python impl

khanlab · Feb 23, 2024 · 87367f3 · 87367f3
1 parent 96b13c0
commit 87367f3
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/snakebids/_warningformat.py b/snakebids/_warningformat.py
@@ -2,7 +2,6 @@
 
 import textwrap
 import warnings
-from pathlib import Path
 
 from colorama import Fore, Style
 
@@ -23,9 +22,15 @@ def formatwarning(
 ):
     """Format warning messages."""
     if line is None:
-        with Path(filename).open() as f:
-            for _ in range(lineno):
-                line = f.readline().strip()
+        try:
+            import linecache
+
+            line = linecache.getline(filename, lineno)
+        except Exception:  # noqa: BLE001
+            # When a warning is logged during Python shutdown, linecache
+            # and the import machinery don't work anymore
+            line = None
+            linecache = None
 
     return WARN_TEMPLATE.format(
         message=textwrap.indent(