Catch exception if node is in unexpected statement (#84)

alexmojaki · Aug 29, 2024 · 6428317 · 6428317
1 parent ea6a695
commit 6428317
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/executing/executing.py b/executing/executing.py
@@ -273,16 +273,15 @@ def executing(cls, frame_or_tb):
                             node_finder = NodeFinder(frame, stmts, tree, lasti, source)
                             node = node_finder.result
                             decorator = node_finder.decorator
+
+                    if node:
+                        new_stmts = {statement_containing_node(node)}
+                        assert_(new_stmts <= stmts)
+                        stmts = new_stmts
                 except Exception:
                     if TESTING:
                         raise
 
-                assert stmts is not None
-                if node:
-                    new_stmts = {statement_containing_node(node)}
-                    assert_(new_stmts <= stmts)
-                    stmts = new_stmts
-
             executing_cache[key] = args = source, node, stmts, decorator
 
         return Executing(frame, *args)