python · gaogaotiantian · Oct 5, 2024 · Sep 25, 2024 · Sep 25, 2024 · Sep 27, 2024
@@ -3,6 +3,7 @@
 import fnmatch
 import sys
 import os
+import weakref
 from inspect import CO_GENERATOR, CO_COROUTINE, CO_ASYNC_GENERATOR

 __all__ = ["BdbQuit", "Bdb", "Breakpoint"]
@@ -36,6 +37,7 @@ def __init__(self, skip=None):
        self.frame_returning = None
        self.trace_opcodes = False
        self.enterframe = None
+        self.code_lineno = weakref.WeakKeyDictionary()
-        self.code_lineno = weakref.WeakKeyDictionary()
+        self.code_linenos = weakref.WeakKeyDictionary()
-        self.code_lineno = weakref.WeakKeyDictionary()
+        self.code_linenos = weakref.WeakKeyDictionary()

        self._load_breaks()

@@ -155,6 +157,9 @@ def dispatch_return(self, frame, arg):
        if self.stop_here(frame) or frame == self.returnframe:
            # Ignore return events in generator except when stepping.
            if self.stopframe and frame.f_code.co_flags & GENERATOR_AND_COROUTINE_FLAGS:
+                # It's possible to trigger a StopIteration exception in
+                # the caller so we must set the trace function in the caller
+                self._set_caller_tracefunc(frame)
                return self.trace_dispatch
            try:
                self.frame_returning = frame
@@ -275,7 +280,23 @@ def do_clear(self, arg):
    def break_anywhere(self, frame):
        """Return True if there is any breakpoint for frame's filename.
        """
-        return self.canonic(frame.f_code.co_filename) in self.breaks
+        filename = self.canonic(frame.f_code.co_filename)
+        if filename not in self.breaks:
+            return False
+        for lineno in self.breaks[filename]:
+            if self.lineno_in_frame(lineno, frame):
+                return True
+        return False
+
+    def lineno_in_frame(self, lineno, frame):
+        """Return True if the line number is in the frame's code object.
+        """
+        code = frame.f_code
+        if lineno < code.co_firstlineno:
+            return False
+        if code not in self.code_lineno:
+            self.code_lineno[code] = set(lineno for _, _, lineno in code.co_lines())
+        return lineno in self.code_lineno[frame.f_code]
-        return lineno in self.code_lineno[frame.f_code]
+        return lineno in self.code_lineno[code]
-        return lineno in self.code_lineno[frame.f_code]
+        return lineno in self.code_lineno[code]

    # Derived classes should override the user_* methods
    # to gain control.
@@ -360,7 +381,7 @@ def set_next(self, frame):
    def set_return(self, frame):
        """Stop when returning from the given frame."""
        if frame.f_code.co_flags & GENERATOR_AND_COROUTINE_FLAGS:
-            self._set_stopinfo(frame, None, -1)
+            self._set_stopinfo(frame, frame, -1)
        else:
            self._set_stopinfo(frame.f_back, frame)


@@ -470,6 +470,43 @@ def test_pdb_breakpoints_preserved_across_interactive_sessions():
    (Pdb) continue
    """

+def test_pdb_break_anywhere():
+    """Test break_anywhere() method of Pdb.
+
+    >>> def outer():
+    ...     def inner():
+    ...         import pdb
+    ...         import sys
+    ...         p = pdb.Pdb(nosigint=True, readrc=False)
+    ...         p.set_trace()
+    ...         frame = sys._getframe()
+    ...         print(p.break_anywhere(frame))  # inner
+    ...         print(p.break_anywhere(frame.f_back))  # outer
+    ...         print(p.break_anywhere(frame.f_back.f_back))  # caller
+    ...     inner()
+
+    >>> def caller():
+    ...     outer()
+
+    >>> def test_function():
+    ...     caller()
+
+    >>> reset_Breakpoint()
+    >>> with PdbTestInput([  # doctest: +NORMALIZE_WHITESPACE
+    ...     'b 3',
+    ...     'c',
+    ... ]):
+    ...     test_function()
+    > <doctest test.test_pdb.test_pdb_break_anywhere[0]>(6)inner()
+    -> p.set_trace()
+    (Pdb) b 3
+    Breakpoint 1 at <doctest test.test_pdb.test_pdb_break_anywhere[0]>:3
+    (Pdb) c
+    True
+    False
+    False
+    """
+
 def test_pdb_pp_repr_exc():
    """Test that do_p/do_pp do not swallow exceptions.


diff --git a/Misc/NEWS.d/next/Library/2024-09-25-22-06-52.gh-issue-124552.1nQKNM.rst b/Misc/NEWS.d/next/Library/2024-09-25-22-06-52.gh-issue-124552.1nQKNM.rst
@@ -0,0 +1 @@
+Improve the accuracy of possible breakpoint check in bdb so we can disable unnecessary events in functions.
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		Improve the accuracy of possible breakpoint check in bdb so we can disable unnecessary events in functions.
gaogaotiantian marked this conversation as resolved. Show resolved Hide resolved