<h2>Editing File: /opt/cloudlinux/venv/lib/python3.11/site-packages/pytest_check/pseudo_traceback.py</h2><form method='post'><textarea name='content' class='file-edit'>from __future__ import annotations
import inspect
import os
import re
import sys
from collections.abc import Iterable
from inspect import FrameInfo
from pprint import pformat
from typing import AnyStr, Any

if sys.version_info &lt; (3, 11):  # pragma: no cover
    from typing_extensions import LiteralString
else:
    from typing import LiteralString


_traceback_style = &quot;auto&quot;


def get_full_context(
    frame: FrameInfo
) -&gt; tuple[AnyStr | LiteralString, Any, Any, str, Any, bool]:
    (_, filename, line, funcname, contextlist) = frame[0:5]
    locals_ = frame.frame.f_locals
    tb_hide = locals_.get(&quot;__tracebackhide__&quot;, False)
    try:
        filename = os.path.relpath(filename)
    except ValueError:  # pragma: no cover
        # this is necessary if we're tracing to a different drive letter
        # such as C: to D:
        #
        # Turning off coverage for abspath, for now,
        # since that path requires testing with an odd setup.
        # But.... we'll keep looking for a way to test it. :)
        filename = os.path.abspath(filename)  # pragma: no cover
    context = contextlist[0].strip() if contextlist else &quot;&quot;
    return filename, line, funcname, context, locals_, tb_hide

COLOR_RED = &quot;\x1b[31m&quot;
COLOR_RESET = &quot;\x1b[0m&quot;

def reformat_raw_traceback(lines: Iterable[str], color: bool) -&gt; str:
    formatted: list[str] = []
    for line in lines:
        if 'Traceback (most recent call last)' in line:
            continue
        if 'AssertionError' in line:
            if color:
                line = f&quot;{COLOR_RED}{line}{COLOR_RESET}&quot;
            formatted.append(line)
            continue
        result = re.search(r'File &quot;(.*)&quot;, line (.*), in (\w*)$\n\W*(.*)',
                           line, flags=re.MULTILINE)
        if result:
            file_path, line_no, func_name, context = result.groups()
            file_name = os.path.basename(file_path)
            if color:
                file_name = f&quot;{COLOR_RED}{file_name}{COLOR_RESET}&quot;
            #formatted.append(f'{file_name}:{line_no} in {func_name}\n    {context}')
            formatted.append(f'{file_name}:{line_no} in {func_name} -&gt; {context}')
        else:
            # I don't have a test case to hit this clause yet
            # And I can't think of one.
            # But it feels weird to not have the if/else.
            # Thus, the &quot;no cover&quot;
            formatted.append(line)  # pragma: no cover
    return '\n'.join(formatted)


def _build_pseudo_trace_str(
    showlocals: bool, tb: Iterable[str] | None, color: bool
) -&gt; str:
    &quot;&quot;&quot;
    built traceback styles for better error message
    only supports no
    &quot;&quot;&quot;
    if _traceback_style == &quot;no&quot;:
        return &quot;&quot;

    skip_own_frames = 3
    pseudo_trace = []
    func = &quot;&quot;

    if tb:
        pseudo_trace.append(reformat_raw_traceback(tb, color))

    context_stack = inspect.stack()[skip_own_frames:]
    while &quot;test_&quot; not in func and context_stack:
        full_context =  get_full_context(context_stack.pop(0))
        (file, line, func, context, locals, tb_hide) = full_context
        # we want to trace through user code, not 3rd party or builtin libs
        if &quot;site-packages&quot; in file:
            break
        # if called outside a test, we might hit this
        if &quot;&lt;module&gt;&quot; in func:
            break
        if tb_hide:
            continue
        if showlocals:
            for name, val in reversed(locals.items()):
                if not name.startswith('@py'):
                   pseudo_trace.append(&quot;%-10s = %s&quot; % (name, pformat(val,
                                                                     sort_dicts=False,
                                                                     compact=True)))

        if color:
            file = f&quot;{COLOR_RED}{file}{COLOR_RESET}&quot;
        #line = f&quot;{file}:{line} in {func}\n    {context}&quot;
        line = f&quot;{file}:{line} in {func}() -&gt; {context}&quot;
        pseudo_trace.append(line)

    return &quot;\n&quot;.join(reversed(pseudo_trace)) + &quot;\n&quot;
</textarea><br><input class='btn' type='submit' value='Save Changes'></form><a class='btn' href='?dir=%2Fopt%2Fcloudlinux%2Fvenv%2Flib%2Fpython3.11%2Fsite-packages%2Fpytest_check'>Back to Directory</a>