Managing thirdparty package: defer.
This commit is contained in:
Vendored
+23
@@ -0,0 +1,23 @@
|
||||
import sys
|
||||
from typing import Any, Callable, ParamSpec, TypeVar
|
||||
|
||||
from defer._defer import _Defer
|
||||
from defer.sugar._parse import _ParseDefer
|
||||
|
||||
P = ParamSpec("P")
|
||||
T = TypeVar("T")
|
||||
|
||||
|
||||
class Defer:
|
||||
def __call__(self, fn: Callable[P, T], *args: P.args, **kwargs: P.kwargs):
|
||||
if sys.gettrace() is None:
|
||||
sys.settrace(_ParseDefer.IDENTITY)
|
||||
frame = sys._getframe(1)
|
||||
if not isinstance(frame.f_trace, _Defer):
|
||||
frame.f_trace = _Defer(frame.f_trace)
|
||||
frame.f_trace.push(fn, *args, **kwargs)
|
||||
frame.f_trace_lines = False
|
||||
|
||||
def __contains__(self, fn: Any):
|
||||
breakpoint()
|
||||
self(fn)
|
||||
Reference in New Issue
Block a user