debug: more convenient run_trace.py

Author: nedbat

lab/run_trace.py

@@ -14,13 +14,14 @@ def trace(frame, event, arg):
         # This can happen when Python is shutting down.
         return None
 
-    print("%s%s %s %d @%d" % (
-        "    " * nest,
-        event,
-        os.path.basename(frame.f_code.co_filename),
-        frame.f_lineno,
-        frame.f_lasti,
-    ))
+    if the_program in frame.f_code.co_filename:
+        print("%s%s %s %d @%d" % (
+            "    " * nest,
+            event,
+            os.path.basename(frame.f_code.co_filename),
+            frame.f_lineno,
+            frame.f_lasti,
+        ))
 
     if event == 'call':
         nest += 1
@@ -33,5 +34,6 @@ def trace(frame, event, arg):
 the_program = sys.argv[1]
 
 code = open(the_program, encoding="utf-8").read()
+code_obj = compile(code, the_program, mode="exec")
 sys.settrace(trace)
-exec(code)
+exec(code_obj)