28 lines
763 B
Python
28 lines
763 B
Python
import sys
|
|
from typing import Any, Callable, ParamSpec, TypeVar
|
|
|
|
from defer._defer import _Defer
|
|
|
|
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(_DeferTrace())
|
|
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)
|
|
|
|
|
|
class _DeferTrace:
|
|
"""Minimal no-op trace for when @defer is used but no other trace is active."""
|
|
def __call__(self, frame, event, arg):
|
|
return self |