Improve perf of stack inspection

blockbuster/blockbuster.py

@@ -52,13 +52,17 @@ def wrapper(*args: Any, **kwargs: Any) -> _T:
             asyncio.get_running_loop()
         except RuntimeError:
             return func(*args, **kwargs)
-        for filename, functions in can_block_functions:
-            for frame_info in inspect.stack():
-                if (
-                    frame_info.filename.endswith(filename)
-                    and frame_info.function in functions
-                ):
-                    return func(*args, **kwargs)
+        if can_block_functions:
+            frame = inspect.currentframe()
+            while frame:
+                frame_info = inspect.getframeinfo(frame)
+                for filename, functions in can_block_functions:
+                    if (
+                        frame_info.filename.endswith(filename)
+                        and frame_info.function in functions
+                    ):
+                        return func(*args, **kwargs)
+                frame = frame.f_back
         if can_block_predicate(*args, **kwargs):
             return func(*args, **kwargs)
         raise _blocking_error(func)

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "blockbuster"
-version = "1.3.1"
+version = "1.3.2"
 description = "Utility to detect blocking calls in the async event loop"
 readme = "README.md"
 keywords = ["async", "block", "detect", "event loop", "asyncio"]