o iF.@sddlZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z dd lm Z ddl Z dd l m Z dd lmZdd lmZdd lmZddlmZejddZGddde ZeeeeeeefgefZdede fddZeZedkre dnedkre dne deZ!edkre! dnedkre! dne! deZ"edkre" dnedkre" d ne" d!eedkZ#d"edee ddffd#d$Z$eZ% d%d"edee ddffd&d'Z&d(ed)ede j'fd*d+Z(d,ed(edefd-d.Z)d,ede*fd/d0Z+d,ed(ede*fd1d2Z,d3ed(edefd4d5Z-d,edefd6d7Z.d,ed"eddfd8d9Z/dS):N)CodeType) FunctionType)Any)Callable) Generator)Optional)Protocol)cast)Instr)Assembly)link_function_to_code) wrap_async)wrap_generatorc@sPeZdZUdZdZeeed<dZee e e fed<de de de fdd Z dS) WrappedFunctionzA wrapped function.N__dd_wrapped____dd_wrappers__argskwargsreturncOsdSN)selfrrrrV/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/internal/wrapping/__init__.py__call__szWrappedFunction.__call__) __name__ __module__ __qualname____doc__rrr__annotations__rdictrrrrrrrs rlinenorcCs&tdkr tdtjj|dStd|dS)N BINARY_OPr! INPLACE_ADD)PYr bcBinaryOpADDr&rrr_add$s r,)r# z resume 0 load_const {wrapper} push_null load_const {wrapped} r"z resume 0 push_null load_const {wrapper} load_const {wrapped} z] load_const {wrapper} load_const {wrapped} )r# z copy 1 load_method $update load_fast {varkwargsname} call 1 pop_top z copy 1 load_method $update load_fast {varkwargsname} precall 1 call 1 pop_top z dup_top load_attr $update load_fast {varkwargsname} call_function 1 pop_top zH call {arg} return_value zn precall {arg} call {arg} return_value zH call_function {arg} return_value codec#sj}jttjtjj@}j}|r||j nd}|rNfdd|d|DEdHt d|dV|rLt d|dVt VdSdS|rZt d|dVdSt dddVdS)zAGenerate the opcodes for building the positional arguments tuple.Nc3sD|]}tdkr|jvrtdt|dntd|dVqdS)r" LOAD_DEREFr& LOAD_FASTN)r( co_cellvarsr r)CellVar).0argnamer/r!rr s  z#generate_posargs.. BUILD_TUPLEr&r1r) co_varnamesco_firstlinenoFIRSTLINENO_OFFSETboolco_flagsr) CompilerFlagsVARARGS co_argcountco_kwonlyargcountr r,)r/varnamesvarargsnargs varargsnamerr6rgenerate_posargss$    rFzI load_const {arg} load_fast {arg} c cs|j}|j}|jt}t|tjj@}t|tjj@}|j }|j }|r,||||nd}|ra||||D]} t j d| i|dEdHq8t d||dV|r_tj d|i|dEdHdSdS|rmt d||dVdSt dd|dVdS)zCGenerate the opcodes for building the keyword arguments dictionary.Nargr& BUILD_MAP varkwargsnamer1r)r=r9r:r;r<r)r>r? VARKEYWORDSr@rAPAIRbindr UPDATE_MAP) r/flagsrBr!rC varkwargsrD kwonlyargsrIrGrrrgenerate_kwargss& rQwrapperwrappedcs|j}|jttj||dd}|t||t||tjddidt dkrR|j rAfdd|j D|dd<|j rR| dt d t|j dtjj|j@rhtjj|j@sht|||St|||S) aWrap a function with a wrapper function. The wrapper function expects the wrapped function as the first argument, followed by the tuple of arguments and the dictionary of keyword arguments. The nature of the wrapped function is also honored, meaning that a generator function will return a generator function, and a coroutine function will return a coroutine function, and so on. The signature is also preserved to avoid breaking, e.g., usages of the ``inspect`` module. )rRrSr&rGr#r"cs g|] }tdt|dqS) MAKE_CELLr&)r r)r3)r4_r&rr s z!wrap_bytecode..rCOPY_FREE_VARS)__code__r:r;HEADrLextendrFrQ CALL_RETURNr(r2 co_freevarsinsertr lenr)r> GENERATORr= COROUTINErr )rRrSr/instrsrr&r wrap_bytecodes    rbfc Cst|j}|jd|j|j}ztt|}tt|jtt|_Wn ty(Ynw|j |_ |j }|j }|j }t |tjj@t |tjj@}t||} || _|jd|| _|j| _|j| _|| _|| _|j| _|j| _t dkrx|j!| _"| #|_tt|}||_t$|||S)a>Wrap a function with a wrapper. The wrapper expects the function as first argument, followed by the tuple of positional arguments and the dict of keyword arguments. Note that this changes the behavior of the original function with the wrapper function, instead of creating a new function object. Nr")%rrX __globals__ __defaults__ __closure__r rrAttributeError__kwdefaults__r=r@rAr<r)r>r?rJrbargcountr9argnames co_filenamefilenamer\freevarsrNkwonlyargcountco_namenameco_posonlyargcountposonlyargcountr(r2cellvarsto_coder ) rcrRr/rSwfrNrj kwonlycountrD wrapped_coderrrwrapsN       rycCsDztt|}tt|j}|jdksJdWdSty!YdSw)z0Check if a function is wrapped with any wrapper.rdWrapper has wrapped functionTF)r rrrrrh)rcrvinnerrrr is_wrapped5s   r|cCs\z#tt|}tt|j}|jdksJd||jjvrWdSt||WSty-YdSw)z7Check if a function is wrapped with a specific wrapper.rdrzTF) r rrrrrX co_constsis_wrapped_withrh)rcrRrvr{rrrr~Bs     r~rvcCsztt|j}Wntytt|YSw|jdks Jd|tt|jjvr1ttt||Stt|}|j|_z tt|j|_W|StyP|`Y|Sw)zUnwrap a wrapped function. This is the reverse of :func:`wrap`. In case of multiple wrapping layers, this will unwrap the one that uses ``wrapper``. If the function was not wrapped with ``wrapper``, it will return the first argument. rdrz) r rrrhrrXr}unwrapr)rvrRr{rcrrrrUs"    rcCs t|r tt|jp |jS|jSrr|r rrrX)rcrrrget_function_code|s rcCs&|t|rtt|jp |_dS|_dSrr)rcr/rrrset_function_codes&r)0systypesrrtypingrrrrrr bytecoder)r ddtrace.internal.assemblyr !ddtrace.internal.utils.inspectionr ddtrace.internal.wrapping.asyncsr $ddtrace.internal.wrapping.generatorsr version_infor(rtupler strWrapperintr,rYparserMr[r;rFrKrQBytecoderbryr<r|r~rrrrrrrs                      0= '