Elide task.context.run() and contextvars.callable() frames from tracebacks

trio/_core/_run.py

@@ -3,6 +3,7 @@
 import os
 import random
 import select
+import sys
 import threading
 from collections import deque
 import collections.abc
@@ -1376,7 +1377,13 @@ def run_impl(runner, async_fn, args):
             except StopIteration as stop_iteration:
                 final_result = Value(stop_iteration.value)
             except BaseException as task_exc:
-                final_result = Error(task_exc)
+                # Store for later, removing uninteresting top frames:
+                #   1. trio._core._run.run_impl()
+                #   2. contextvars.Context.run() (< Python 3.7 only)
+                tb_next = task_exc.__traceback__.tb_next
+                if sys.version_info < (3, 7):  # pragma: no cover
+                    tb_next = tb_next.tb_next
+                final_result = Error(task_exc.with_traceback(tb_next))
 
             if final_result is not None:
                 # We can't call this directly inside the except: blocks above,

trio/_core/tests/test_run.py

@@ -1,9 +1,11 @@
 import contextvars
 import functools
 import platform
+import re
 import sys
 import threading
 import time
+import traceback
 import warnings
 from contextlib import contextmanager
 from math import inf
@@ -1897,6 +1899,27 @@ def handle(exc):
     assert result == [[0, 0], [1, 1]]
 
 
+async def test_run_impl_traceback_frame_removal():
+    async def my_child_task():
+        raise KeyError()
+
+    try:
+        # Trick: For now cancel/nursery scopes still leave a bunch of tb gunk
+        # behind. But if there's a MultiError, they leave it on the MultiError,
+        # which lets us get a clean look at the KeyError itself. Someday I
+        # guess this will always be a MultiError (#611), but for now we can
+        # force it by raising two exceptions.
+        async with _core.open_nursery() as nursery:
+            nursery.start_soon(my_child_task)
+            nursery.start_soon(my_child_task)
+    except _core.MultiError as exc:
+        first_exc = exc.exceptions[0]
+        assert isinstance(first_exc, KeyError)
+        tb_text = ''.join(traceback.format_tb(first_exc.__traceback__))
+        for r in ('/trio/_core/.* in run_impl$', '/contextvars/.* in run$'):
+            assert not re.search(r, tb_text, re.MULTILINE)
+
+
 def test_contextvar_support():
     var = contextvars.ContextVar("test")
     var.set("before")