fixes #3

Author: jph00

DESIGN.md

@@ -108,9 +108,11 @@ This keeps tracing focused to the target function and any nested defs within its
 ### Call stack capture
 
 Each recorded call includes a `stack_str` captured on call entry. The stack is a newline-joined
-list of `func_name (file:line)` frames, filtered so `fn` is the shallowest frame shown (frames
-above `fn` are omitted). For frames under `fn`'s directory, `file` is shown relative to that
-directory; other frames keep full paths. If `target_func` is `None`, `stack_str` is empty.
+list of `qualname (file:line)` frames (so methods include the class name), filtered so `fn` is the
+shallowest frame shown (frames above `fn` are omitted). For frames under `fn`'s directory, `file`
+is shown relative to that directory. Remaining absolute paths are normalized by replacing the
+longest matching prefixes first (e.g., `$SITE_PACKAGES`, `$VIRTUAL_ENV`, `~`, Python prefix paths).
+If `target_func` is `None`, `stack_str` is empty.
 
 ### Mapping execution position to an AST line
 

tests/test_tracefunc.py

@@ -141,6 +141,16 @@ def _wrapper_calls_target_two_places(n):
     return out
 
 
+class _StackClass:
+    def inc(self, x):
+        return x + 1
+
+
+def _wrapper_calls_method(x):
+    obj = _StackClass()
+    return obj.inc(x)
+
+
 # ----------------------------
 # Helpers
 # ----------------------------
@@ -568,3 +578,10 @@ def test_stack_traces_include_target_and_show_two_call_sites():
         assert lines[-1].startswith("_targeted_inner (")
         assert any("_wrapper_calls_target_two_places (" in line for line in lines)
         assert base_dir not in stack
+
+
+def test_stack_traces_include_class_name_for_methods():
+    res_list = tracefunc(_wrapper_calls_method, 3, target_func=_StackClass.inc)
+    assert len(res_list) == 1
+    stack = res_list[0][0]
+    assert any(line.startswith("_StackClass.inc (") for line in stack.splitlines())

tracefunc/core.py

@@ -1,4 +1,4 @@
-import ast, builtins, dis, inspect, os, sys, textwrap
+import ast, builtins, dis, inspect, os, sys, sysconfig, textwrap
 from fastcore.utils import *
 
 comp_node_types = (ast.ListComp, ast.SetComp, ast.DictComp, ast.GeneratorExp)
@@ -86,6 +86,27 @@ def tracefunc(fn, /, *args, target_func=None, **kwargs):
 
     stack_anchor = fn.__code__ if target_func is not None else None
     stack_base = os.path.dirname(stack_anchor.co_filename) if stack_anchor else None
+    repls = []
+    venv = os.environ.get("VIRTUAL_ENV")
+    if venv: repls.append((os.path.realpath(venv), "$VIRTUAL_ENV"))
+    home = os.path.expanduser("~")
+    if home: repls.append((os.path.realpath(home), "~"))
+    for p, label in (
+        (sys.prefix, "$PYTHON_PREFIX"),
+        (getattr(sys, "base_prefix", None), "$PYTHON_BASE"),
+        (getattr(sys, "exec_prefix", None), "$PYTHON_EXEC_PREFIX"),
+        (getattr(sys, "base_exec_prefix", None), "$PYTHON_BASE_EXEC_PREFIX"),
+    ):
+        if p: repls.append((os.path.realpath(p), label))
+    paths = sysconfig.get_paths()
+    for key, label in (
+        ("purelib", "$SITE_PACKAGES"),
+        ("platlib", "$PLAT_SITE_PACKAGES"),
+        ("stdlib", "$PYTHON_STDLIB"),
+        ("platstdlib", "$PYTHON_PLAT_STDLIB"),
+    ):
+        if key in paths: repls.append((os.path.realpath(paths[key]), label))
+    repls.sort(key=lambda x: len(x[0]), reverse=True)
     if target_func is None: target_func = fn
 
     try: src_lines, block_first_lineno = inspect.getsourcelines(target_func)
@@ -284,12 +305,19 @@ def _stack_str(frame):
                 frames = frames[i:]
                 break
         else: return ""
+        def _shorten(path):
+            if stack_base and path.startswith(stack_base + os.sep):
+                return os.path.relpath(path, stack_base)
+            if not os.path.isabs(path): return path
+            for base, label in repls:
+                if path == base or path.startswith(base + os.sep):
+                    return label + path[len(base):]
+            return path
         out = []
         for fr in frames:
-            path = fr.f_code.co_filename
-            if stack_base and path.startswith(stack_base + os.sep):
-                path = os.path.relpath(path, stack_base)
-            out.append(f"{fr.f_code.co_name} ({path}:{fr.f_lineno})")
+            path = _shorten(fr.f_code.co_filename)
+            name = getattr(fr.f_code, "co_qualname", fr.f_code.co_name)
+            out.append(f"{name} ({path}:{fr.f_lineno})")
         return "\n".join(out)
 
     def _snapshot(call_idx, line_id, frame, *, force=False):