o i@sUdZddlZddlZddlmZiaejejeefej fe d<ej a Gddde ZddZd d Zejd d Zd ej fddZdS)aUtility for monkey patching typing.overload to allow run time retrieval overloads Requires any @typing.overload to happen within the patched_overload contextmanager, e.g.: ```python with patched_overload(): # the following could be imported from some other module (as long as it wasn't already loaded), or inlined: @typing.overload def foo(a: int) -> float: ... def foo(a: typing.Union[bool, int]) -> typing.Union[bool, float]: if isinstance(a, bool): return a return float(a) # returns reference to the overloads of foo (the int -> float one in this case) # in the order they are declared foo_overloads = get_overloads(foo) N)mock overloadsc@s eZdZdS) UntrackableN)__name__ __module__ __qualname__rrS/home/ubuntu/.local/lib/python3.10/site-packages/synchronicity/overload_tracking.pyrsrcCs<t|ttfr t|jSz|j|jfWStytwN) isinstance staticmethod classmethod_function_locator__func__rrAttributeErrorrfrrr r#s  rcCsLzt|}t|g|Wt|Sty%td|Yt|Sw)Nz#WARNING: can't track overloads for )rr setdefaultappendrprintoriginal_overload)rlocatorrrr _tracking_overload-s rccs:tdt dVWddS1swYdS)Nztyping.overload)rpatchrrrrr patched_overload9s"rreturncCs*z tt|gWStygYSwr )rgetrrrrrr get_overloads?s  r)__doc__ contextlibtypingunittestrrDictTuplestrList__annotations__overloadr Exceptionrrrcontextmanagerrrrrrr s "