Only trigger events on thread calling start_trace

gaogaotiantian · gaogaotiantian · commit 99ea70c21892 · 2024-10-16T19:28:13.000-04:00
diff --git a/Lib/bdb.py b/Lib/bdb.py
@@ -2,6 +2,7 @@
 
 import fnmatch
 import sys
+import threading
 import os
 from inspect import CO_GENERATOR, CO_COROUTINE, CO_ASYNC_GENERATOR
 
@@ -36,9 +37,11 @@ def __init__(self):
         self._name = 'bdbtracer'
         self._tracefunc = None
         self._disable_current_event = False
+        self._tracing_thread = None
 
     def start_trace(self, tracefunc):
         self._tracefunc = tracefunc
+        self._tracing_thread = threading.current_thread()
         curr_tool = sys.monitoring.get_tool(self._tool_id)
         if curr_tool is None:
             sys.monitoring.use_tool_id(self._tool_id, self._name)
@@ -57,6 +60,7 @@ def start_trace(self, tracefunc):
         sys.monitoring.set_events(self._tool_id, all_events)
 
     def stop_trace(self):
+        self._tracing_thread = None
         curr_tool = sys.monitoring.get_tool(self._tool_id)
         if curr_tool != self._name:
             return
@@ -78,6 +82,8 @@ def callback_wrapper(func):
 
         @functools.wraps(func)
         def wrapper(self, *args):
+            if self._tracing_thread != threading.current_thread():
+                return
             try:
                 frame = sys._getframe().f_back
                 ret = func(self, frame, *args)
