beautiful-traceback 0.3.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

beautiful_traceback/__init__.py

@@ -1,17 +1,7 @@
 from ._extension import load_ipython_extension  # noqa: F401
-from .formatting import LoggingFormatter, LoggingFormatterMixin
-from .hook import install, uninstall
-from .json_formatting import exc_to_json
+from .formatting import LoggingFormatter, LoggingFormatterMixin  # noqa: F401
+from .hook import install, uninstall  # noqa: F401
+from .json_formatting import exc_to_json  # noqa: F401
 
 # retain typo for backward compatibility
 LoggingFormaterMixin = LoggingFormatterMixin
-
-
-__all__ = [
-    "install",
-    "uninstall",
-    "LoggingFormatter",
-    "LoggingFormatterMixin",
-    "LoggingFormaterMixin",
-    "exc_to_json",
-]

beautiful_traceback/formatting.py

@@ -129,6 +129,46 @@ TEST_PATHS: typ.List[str] = []
 PWD = os.getcwd()
 
 
+def _compile_exclude_patterns(
+    exclude_patterns: typ.Sequence[str],
+) -> list[re.Pattern[str]]:
+    if not exclude_patterns:
+        return []
+
+    return [re.compile(pattern) for pattern in exclude_patterns]
+
+
+def _row_matches_exclude_patterns(
+    row: Row, exclude_patterns: typ.Sequence[re.Pattern[str]]
+) -> bool:
+    if not exclude_patterns:
+        return False
+
+    # Try multiple representations so patterns can match different parts:
+    # 1. short_module: "_pytest/runner.py" (aliased path without prefix)
+    # 2. full_module: "/path/to/site-packages/_pytest/runner.py" (absolute path)
+    # 3. haystack_full: "<site> /path/to/site-packages/_pytest/runner.py:353 from_call result: ..."
+    # 4. haystack_short: "<site> _pytest/runner.py:353 from_call result: ..."
+    haystack_full = (
+        f"{row.alias} {row.full_module}:{row.lineno} {row.call} {row.context}"
+    )
+    haystack_short = (
+        f"{row.alias} {row.short_module}:{row.lineno} {row.call} {row.context}"
+    )
+    candidates = [
+        row.short_module,
+        row.full_module,
+        haystack_full,
+        haystack_short,
+    ]
+    for pattern in exclude_patterns:
+        for candidate in candidates:
+            if pattern.search(candidate):
+                return True
+
+    return False
+
+
 def _py_paths() -> typ.List[str]:
     if TEST_PATHS:
         return TEST_PATHS
@@ -227,7 +267,9 @@ def _iter_entry_rows(
 
 
 def _init_entries_context(
-    entries: StackFrameEntryList, term_width: typ.Optional[int] = None
+    entries: StackFrameEntryList,
+    term_width: typ.Optional[int] = None,
+    exclude_patterns: typ.Sequence[str] = (),
 ) -> Context:
     if term_width is None:
         _term_width = _get_terminal_width()
@@ -246,6 +288,19 @@ def _init_entries_context(
     max_row_width = _term_width - 10
 
     rows = list(_iter_entry_rows(aliases, entry_paths, entries))
+    compiled_exclude_patterns = _compile_exclude_patterns(exclude_patterns)
+    if compiled_exclude_patterns:
+        rows = [
+            row
+            for row in rows
+            if not _row_matches_exclude_patterns(row, compiled_exclude_patterns)
+        ]
+
+    used_aliases = {row.alias for row in rows if row.alias}
+    if used_aliases:
+        aliases = [alias for alias in aliases if alias[0] in used_aliases]
+    else:
+        aliases = []
 
     if rows:
         max_short_module_len = max(
@@ -426,9 +481,15 @@ def _format_traceback(
 
 
 def format_traceback(
-    traceback: ExceptionTraceback, color: bool = False, local_stack_only: bool = False
+    traceback: ExceptionTraceback,
+    color: bool = False,
+    local_stack_only: bool = False,
+    exclude_patterns: typ.Sequence[str] = (),
 ) -> str:
-    ctx = _init_entries_context(traceback.stack_frames)
+    ctx = _init_entries_context(
+        traceback.stack_frames,
+        exclude_patterns=exclude_patterns,
+    )
     return _format_traceback(ctx, traceback, color, local_stack_only)
 
 
@@ -436,6 +497,7 @@ def format_tracebacks(
     tracebacks: typ.List[ExceptionTraceback],
     color: bool = False,
     local_stack_only: bool = False,
+    exclude_patterns: typ.Sequence[str] = (),
 ) -> str:
     traceback_strs: typ.List[str] = []
 
@@ -447,7 +509,12 @@ def format_tracebacks(
             # traceback_strs.append("vvv happend after ^^^ - ")
             traceback_strs.append(CONTEXT_HEAD + os.linesep)
 
-        traceback_str = format_traceback(tb_tup, color, local_stack_only)
+        traceback_str = format_traceback(
+            tb_tup,
+            color,
+            local_stack_only,
+            exclude_patterns=exclude_patterns,
+        )
         traceback_strs.append(traceback_str)
 
     return os.linesep.join(traceback_strs).strip()
@@ -463,6 +530,7 @@ def exc_to_traceback_str(
     color: bool = False,
     local_stack_only: bool = False,
     exc_msg_override: str | None = None,
+    exclude_patterns: typ.Sequence[str] = (),
 ) -> str:
     # NOTE (mb 2020-08-13): wrt. cause vs context see
     #   https://www.python.org/dev/peps/pep-3134/#enhanced-reporting
@@ -511,7 +579,12 @@ def exc_to_traceback_str(
 
     tracebacks = list(reversed(tracebacks))
 
-    return format_tracebacks(tracebacks, color, local_stack_only)
+    return format_tracebacks(
+        tracebacks,
+        color,
+        local_stack_only,
+        exclude_patterns=exclude_patterns,
+    )
 
 
 class LoggingFormatterMixin:

beautiful_traceback/hook.py

@@ -8,7 +8,11 @@ import colorama
 from beautiful_traceback import formatting
 
 
-def init_excepthook(color: bool, local_stack_only: bool) -> typ.Callable:
+def init_excepthook(
+    color: bool,
+    local_stack_only: bool,
+    exclude_patterns: typ.Sequence[str],
+) -> typ.Callable:
     def excepthook(
         exc_type: typ.Type[BaseException],
         exc_value: BaseException,
@@ -17,7 +21,11 @@ def init_excepthook(color: bool, local_stack_only: bool) -> typ.Callable:
         # pylint:disable=unused-argument
         tb_str = (
             formatting.exc_to_traceback_str(
-                exc_value, traceback, color, local_stack_only
+                exc_value,
+                traceback,
+                color,
+                local_stack_only,
+                exclude_patterns=exclude_patterns,
             )
             + "\n"
         )
@@ -39,6 +47,7 @@ def install(
     only_tty: bool = True,
     only_hook_if_default_excepthook: bool = True,
     local_stack_only: bool = False,
+    exclude_patterns: typ.Sequence[str] = (),
 ) -> None:
     """Hook the current excepthook to the beautiful_traceback.
 
@@ -68,7 +77,11 @@ def install(
     if only_hook_if_default_excepthook and not is_default_exepthook:
         return
 
-    sys.excepthook = init_excepthook(color=color, local_stack_only=local_stack_only)
+    sys.excepthook = init_excepthook(
+        color=color,
+        local_stack_only=local_stack_only,
+        exclude_patterns=exclude_patterns,
+    )
 
 
 def uninstall() -> None:

beautiful_traceback/json_formatting.py

@@ -46,6 +46,7 @@ def exc_to_json(
     exc_value: BaseException,
     traceback: types.TracebackType | None,
     local_stack_only: bool = False,
+    exclude_patterns: typ.Sequence[str] = (),
 ) -> dict[str, typ.Any]:
     """Convert an exception to a JSON-serializable dictionary.
 
@@ -112,7 +113,11 @@ def exc_to_json(
             "frames": [],
         }
     else:
-        ctx = fmt._init_entries_context(entries, term_width=fmt.DEFAULT_COLUMNS)
+        ctx = fmt._init_entries_context(
+            entries,
+            term_width=fmt.DEFAULT_COLUMNS,
+            exclude_patterns=exclude_patterns,
+        )
         result = _format_traceback_json(
             ctx.rows,
             main_tb.exc_name,
@@ -132,7 +137,11 @@ def exc_to_json(
                     "frames": [],
                 }
             else:
-                ctx = fmt._init_entries_context(entries, term_width=fmt.DEFAULT_COLUMNS)
+                ctx = fmt._init_entries_context(
+                    entries,
+                    term_width=fmt.DEFAULT_COLUMNS,
+                    exclude_patterns=exclude_patterns,
+                )
                 chain_item = _format_traceback_json(
                     ctx.rows,
                     tb.exc_name,

beautiful_traceback/pytest_assertion.py

@@ -0,0 +1,94 @@
+"""Helpers for extracting pytest's rewritten assertion details."""
+
+import os
+
+import pytest
+
+
+def get_exception_message_override(excinfo: pytest.ExceptionInfo) -> str | None:
+    """Return pytest's verbose exception message when rewriting adds detail.
+
+    The plugin overrides pytest's longrepr rendering, which skips the
+    ExceptionInfo repr where pytest stores rewritten assertion diffs. Without
+    this, AssertionError messages collapse to str(exc) and omit left/right
+    details. Pulling reprcrash.message preserves that verbose message when
+    pytest provides one.
+    """
+    try:
+        repr_info = excinfo.getrepr(style="long")
+    except Exception:
+        return None
+
+    reprcrash = getattr(repr_info, "reprcrash", None)
+    if reprcrash is None:
+        return None
+
+    message = getattr(reprcrash, "message", None)
+    if not message:
+        return None
+
+    exc_name = type(excinfo.value).__name__
+    prefix = f"{exc_name}: "
+    if message.startswith(prefix):
+        message = message.removeprefix(prefix)
+
+    if not message or message == exc_name:
+        return None
+
+    exc_message = str(excinfo.value)
+    if message == exc_message:
+        return None
+
+    return message
+
+
+def get_pytest_assertion_details(excinfo: pytest.ExceptionInfo) -> str | None:
+    """Return pytest's rewritten assertion lines for AssertionError.
+
+    Pytest only provides left/right diffs when a module is rewritten during
+    import. For helper modules, call pytest.register_assert_rewrite before
+    importing them, otherwise the assertion message is plain and there are no
+    diff lines to extract.
+    """
+    if not isinstance(excinfo.value, AssertionError):
+        return None
+
+    try:
+        # pytest stores assertion diffs on its own repr object, not the exception.
+        # Reference: https://github.com/pytest-dev/pytest/blob/main/src/_pytest/_code/code.py
+        repr_info = excinfo.getrepr(style="long")
+    except Exception:
+        return None
+
+    reprtraceback = getattr(repr_info, "reprtraceback", None)
+    if reprtraceback is None:
+        chain = getattr(repr_info, "chain", None)
+        if chain:
+            reprtraceback = chain[-1][0]
+
+    if reprtraceback is None:
+        return None
+
+    reprentries = getattr(reprtraceback, "reprentries", None)
+    if not reprentries:
+        return None
+
+    last_entry = reprentries[-1]
+    entry_lines = getattr(last_entry, "lines", None)
+    if not entry_lines:
+        return None
+
+    lines = []
+    for line in entry_lines:
+        stripped = line.lstrip()
+        if stripped.startswith("E"):
+            lines.append(stripped)
+            continue
+
+        if stripped.startswith(">"):
+            lines.append(stripped)
+
+    if not lines:
+        return None
+
+    return os.linesep.join(lines)

beautiful_traceback/pytest_plugin.py

@@ -1,10 +1,24 @@
-from . import formatting
-import pytest
+"""Pytest plugin that preserves rewritten assertion details in tracebacks.
+
+Pytest rewrites assert statements at import time for test modules (and any
+modules registered via pytest.register_assert_rewrite). The rewritten asserts
+raise AssertionError instances that include rich explanation text and left/right
+diffs inside pytest's repr objects, not on the exception itself. This plugin
+extracts those repr details and appends them to beautiful_traceback output.
+"""
 
+import os
+from typing import Any, Generator
+
+import pytest
 from pytest import Config
 
+from . import formatting
+from .pytest_assertion import get_exception_message_override
+from .pytest_assertion import get_pytest_assertion_details
+
 
-def _get_option(config: Config, key: str):
+def _get_option(config: Config, key: str) -> Any:
     val = None
 
     # will throw an exception if option is not set
@@ -19,44 +33,30 @@ def _get_option(config: Config, key: str):
     return val
 
 
-def _get_exception_message_override(excinfo: pytest.ExceptionInfo) -> str | None:
-    """Return pytest's verbose exception message when rewriting adds detail.
-
-    The plugin overrides pytest's longrepr rendering, which skips the
-    ExceptionInfo repr where pytest stores rewritten assertion diffs. Without
-    this, AssertionError messages collapse to str(exc) and omit left/right
-    details. Pulling reprcrash.message preserves that verbose message when
-    pytest provides one.
-    """
-    try:
-        repr_info = excinfo.getrepr(style="long")
-    except Exception:
-        return None
-
-    reprcrash = getattr(repr_info, "reprcrash", None)
-    if reprcrash is None:
-        return None
-
-    message = getattr(reprcrash, "message", None)
-    if not message:
-        return None
-
-    exc_name = type(excinfo.value).__name__
-    prefix = f"{exc_name}: "
-    if message.startswith(prefix):
-        message = message.removeprefix(prefix)
-
-    if not message or message == exc_name:
-        return None
+def _format_traceback(excinfo: pytest.ExceptionInfo, config: Config) -> str:
+    """Format a traceback with beautiful_traceback styling and pytest details."""
+    message_override = get_exception_message_override(excinfo)
+    assertion_details = get_pytest_assertion_details(excinfo)
+    exclude_patterns = _get_option(config, "beautiful_traceback_exclude_patterns")
+
+    formatted_traceback = formatting.exc_to_traceback_str(
+        excinfo.value,
+        excinfo.tb,
+        color=True,
+        local_stack_only=_get_option(
+            config, "enable_beautiful_traceback_local_stack_only"
+        ),
+        exc_msg_override=message_override,
+        exclude_patterns=exclude_patterns,
+    )
 
-    exc_message = str(excinfo.value)
-    if message == exc_message:
-        return None
+    if assertion_details:
+        formatted_traceback += os.linesep + assertion_details + os.linesep
 
-    return message
+    return formatted_traceback
 
 
-def pytest_addoption(parser):
+def pytest_addoption(parser) -> None:
     parser.addini(
         "enable_beautiful_traceback",
         "Enable the beautiful traceback plugin",
@@ -71,56 +71,33 @@ def pytest_addoption(parser):
         default=True,
     )
 
+    parser.addini(
+        "beautiful_traceback_exclude_patterns",
+        "Exclude traceback frames that match regex patterns",
+        type="linelist",
+        default=[],
+    )
+
 
 @pytest.hookimpl(hookwrapper=True)
-def pytest_runtest_makereport(item, call):
-    """
-    Pytest stack traces are challenging to work with by default. This plugin allows beautiful_traceback to be used instead.
-
-    This little piece of code was hard-won:
+def pytest_runtest_makereport(item, call) -> Generator[None, None, None]:
+    """Format test execution tracebacks with beautiful_traceback.
 
-    https://grok.com/share/bGVnYWN5_951be3b1-6811-4fda-b220-c1dd72dedc31
+    This hook runs during the test execution phase and replaces pytest's
+    default traceback formatting with beautiful_traceback's output.
     """
-    outcome = yield
-    report = outcome.get_result()  # Get the generated TestReport object
+    outcome = yield  # type: ignore[misc]
+    report = outcome.get_result()  # type: ignore[attr-defined]
 
-    # Check if the report is for the 'call' phase (test execution) and if it failed
     if _get_option(item.config, "enable_beautiful_traceback") and report.failed:
-        value = call.excinfo.value
-        tb = call.excinfo.tb
+        report.longrepr = _format_traceback(call.excinfo, item.config)
 
-        message_override = _get_exception_message_override(call.excinfo)
 
-        formatted_traceback = formatting.exc_to_traceback_str(
-            value,
-            tb,
-            color=True,
-            local_stack_only=_get_option(
-                item.config, "enable_beautiful_traceback_local_stack_only"
-            ),
-            exc_msg_override=message_override,
-        )
-        report.longrepr = formatted_traceback
-
-
-def pytest_exception_interact(node, call, report):
-    """
-    This can run during collection, not just test execution.
+def pytest_exception_interact(node, call, report) -> None:
+    """Format collection-phase tracebacks with beautiful_traceback.
 
-    So, if there's an import or other pre-run error in pytest, this will apply the correct formatting.
+    This hook runs during collection (e.g., import errors, fixture errors)
+    and ensures those errors also use beautiful_traceback formatting.
     """
-    if report.failed:
-        value = call.excinfo.value
-        tb = call.excinfo.tb
-        message_override = _get_exception_message_override(call.excinfo)
-
-        formatted_traceback = formatting.exc_to_traceback_str(
-            value,
-            tb,
-            color=True,
-            local_stack_only=_get_option(
-                node.config, "enable_beautiful_traceback_local_stack_only"
-            ),
-            exc_msg_override=message_override,
-        )
-        report.longrepr = formatted_traceback
+    if _get_option(node.config, "enable_beautiful_traceback") and report.failed:
+        report.longrepr = _format_traceback(call.excinfo, node.config)

{beautiful_traceback-0.3.0.dist-info → beautiful_traceback-0.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: beautiful-traceback
-Version: 0.3.0
+Version: 0.5.0
 Summary: Beautiful, readable Python tracebacks with colors and formatting
 Keywords: traceback,error,debugging,formatting
 Author: Michael Bianco
@@ -142,6 +142,9 @@ Customize the plugin in your `pytest.ini` or `pyproject.toml`:
 [tool.pytest.ini_options]
 enable_beautiful_traceback = true                    # Enable/disable the plugin
 enable_beautiful_traceback_local_stack_only = true   # Show only local code (filter libraries)
+beautiful_traceback_exclude_patterns = [             # Regex patterns to drop frames
+  "click/core\\.py",
+]
 ```
 
 Or in `pytest.ini`:
@@ -150,8 +153,29 @@ Or in `pytest.ini`:
 [pytest]
 enable_beautiful_traceback = true
 enable_beautiful_traceback_local_stack_only = true
+beautiful_traceback_exclude_patterns =
+    click/core\.py
 ```
 
+Example: filter out pytest, pluggy, and playwright frames from CI tracebacks:
+
+```toml
+[tool.pytest.ini_options]
+beautiful_traceback_exclude_patterns = [
+  "^_pytest/",
+  "^pluggy/",
+  "^playwright/",
+]
+```
+
+**Pattern Matching:** Patterns are tested against multiple representations of each frame:
+- `_pytest/runner.py` (short module path)
+- `/path/to/site-packages/_pytest/runner.py` (full module path)
+- `<site> _pytest/runner.py:353 from_call result: ...` (formatted line with short path)
+- `<site> /path/to/.../runner.py:353 from_call result: ...` (formatted line with full path)
+
+This allows you to write simpler patterns like `^_pytest/` instead of needing to match the full site-packages path.
+
 ## Examples
 
 Check out the [examples/](examples/) directory for detailed usage examples including basic usage, exception chaining, logging integration, and more.
@@ -176,6 +200,7 @@ beautiful_traceback.install(
     only_tty=True,                         # Only activate for TTY output
     only_hook_if_default_excepthook=True,  # Only install if default hook
     local_stack_only=False,                # Filter to show only local code
+    exclude_patterns=["click/core\\.py"],  # Regex patterns to drop frames
     envvar='ENABLE_BEAUTIFUL_TRACEBACK'    # Optional environment variable gate
 )
 ```

beautiful_traceback-0.5.0.dist-info/RECORD

@@ -0,0 +1,14 @@
+beautiful_traceback/__init__.py,sha256=ZDE4AcfeeT420hCntDfdhnp4uhXm1YwCQHsTuvGc0L4,332
+beautiful_traceback/_extension.py,sha256=klyo3XL4q3-Wdy4Lt6JYdh-Cfh_SkRCI7jCcQCwfWTM,311
+beautiful_traceback/cli.py,sha256=M4EWW9SUNiGH2VCpssxRDMNXNjrVpy_8t9T7jf66RKE,2405
+beautiful_traceback/common.py,sha256=Dg6J4rLdX9uKM6LxJaqSwtLkUjggZk-KrR5kpAIA4uo,2208
+beautiful_traceback/formatting.py,sha256=GI6F_l0KyHbMztCwq009oQ2uhAVRX-iDVUklODoQ5rc,17658
+beautiful_traceback/hook.py,sha256=8f4njtVq8tlO8NXBcKV11paNmmfwYz13Nh6eWw6E9JE,2342
+beautiful_traceback/json_formatting.py,sha256=qCGyOsHB4aAlxC2uSv2Lo-DfNAMf0EDeyN7AEAb9_Ts,6313
+beautiful_traceback/parsing.py,sha256=39GvHo6kx5dyzTMfRjUTc8H9tRGIhFT8RcjxUvbZBZY,3094
+beautiful_traceback/pytest_assertion.py,sha256=vyNTOsOzzR70T2oYvzytrvW0R4gCsO7pBkud8d85arw,2784
+beautiful_traceback/pytest_plugin.py,sha256=aVKNoKfI4UmwXvRAV0dPTuYJuLk-rUYOZP_2y_pVteM,3395
+beautiful_traceback-0.5.0.dist-info/WHEEL,sha256=XV0cjMrO7zXhVAIyyc8aFf1VjZ33Fen4IiJk5zFlC3g,80
+beautiful_traceback-0.5.0.dist-info/entry_points.txt,sha256=EXsu7N89wqDpZPEwLHgYVncZJ3y-lbCFZC5fKBZZDac,138
+beautiful_traceback-0.5.0.dist-info/METADATA,sha256=BJYOONwKSbUybFFSo_QP8UgaAbjfSJyA6tmFmqnj4UE,9501
+beautiful_traceback-0.5.0.dist-info/RECORD,,

beautiful_traceback-0.3.0.dist-info/RECORD

@@ -1,13 +0,0 @@
-beautiful_traceback/__init__.py,sha256=e6zDdmcyc3H2aAmY5_9hAeMtCU_b3xyXwGhqrbOa-qo,438
-beautiful_traceback/_extension.py,sha256=klyo3XL4q3-Wdy4Lt6JYdh-Cfh_SkRCI7jCcQCwfWTM,311
-beautiful_traceback/cli.py,sha256=M4EWW9SUNiGH2VCpssxRDMNXNjrVpy_8t9T7jf66RKE,2405
-beautiful_traceback/common.py,sha256=Dg6J4rLdX9uKM6LxJaqSwtLkUjggZk-KrR5kpAIA4uo,2208
-beautiful_traceback/formatting.py,sha256=s4tjdejhYBXBmHlHiUa3bm9j0I6B2hka46JfHNdgKLk,15501
-beautiful_traceback/hook.py,sha256=6vYpqA-mD4G32HkX5WSyCMzkvMwB4RXP6wbVEIU6oaY,2078
-beautiful_traceback/json_formatting.py,sha256=WcgA6rk6YkFDRLXOTk95yVrK_HxNjvHHUqgr2RlK_O4,6071
-beautiful_traceback/parsing.py,sha256=39GvHo6kx5dyzTMfRjUTc8H9tRGIhFT8RcjxUvbZBZY,3094
-beautiful_traceback/pytest_plugin.py,sha256=IwZh5yUUsbLpUWZuVxVqSIGUVPO30_0xIHThc3xor54,3672
-beautiful_traceback-0.3.0.dist-info/WHEEL,sha256=XV0cjMrO7zXhVAIyyc8aFf1VjZ33Fen4IiJk5zFlC3g,80
-beautiful_traceback-0.3.0.dist-info/entry_points.txt,sha256=EXsu7N89wqDpZPEwLHgYVncZJ3y-lbCFZC5fKBZZDac,138
-beautiful_traceback-0.3.0.dist-info/METADATA,sha256=Z5pu8bStps0bEkS8f53AJ5I7krXVHpu2WVdibKPco_s,8558
-beautiful_traceback-0.3.0.dist-info/RECORD,,