o i=@sddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZee Z!e e"e#e$e%e&efZ'e e(e)e"e%e"e$fZ*e"e#e$e%fZ+Gddde Z,Gddde,Z-Gddde e-Z.de(dede(fd d!Z/Gd"d#d#Z0d$ede1e(e0ffd%d&Z2Gd'd(d(eZ3d)S)*) defaultdict)deque)CodeType) FunctionType) ModuleType)Any)Iterator)Optional)Protocol)Union)cast)FunctionWrapper)PYTHON_VERSION_INFO) get_loggerorigin) _isinstance)collect_code_objects)functions_for_code)linenos)resolved_code_origin) undecorated)get_function_codec@s2eZdZUdZdZeeed<dZeeed<dS) FullyNamedzA fully named object.N__name__ __fullname__)r __module__ __qualname____doc__r str__annotations__rr!r!Y/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/debugging/_function/discovery.pyr$s rc@s0eZdZUdZeed<dededefddZdS) FullyNamedFunctionzA fully named function object.rargskwargsreturncOsdSNr!)selfr$r%r!r!r"__call__0szFullyNamedFunction.__call__N)rrrrrr rr)r!r!r!r"r#+s r#c @szeZdZdZ d dedeeedefee e ffddfddZ de eee ffdd Zdeee ffd d ZeZdS) ContainerIteratorzWrapper around different types of function containers. A container comes with an origin, i.e. a parent container and a position within it in the form of a key. N containerrr&cs$tttfrtj|_j|_nPtt r+tt ddD|_d|_n;tt rItfdddD|_j dusCJj j|_ntt tfr_ttjfg|_d|_ntdt|_|dur|djdur|dj}|jrd||jf|_dS||_dS|j|_dS) Ncss|]}|jVqdSr') cell_contents).0_r!r!r" Gsz-ContainerIterator.__init__..c3s"|] \}}|t|fVqdSr')getattr)r-mar+r!r"r/Ks >getterfgetsetterfsetdeleterfdelzUnsupported container type: %sr.) isinstancetyperiter__dict__copyitems_iterrtuple enumeratepropertyr7 classmethod staticmethod__func__ TypeError _containerrjoin)r(r+rorigin_fullnamer!r4r"__init__;s*       & zContainerIterator.__init__cCs|jSr')rEr(r!r!r"__iter__`szContainerIterator.__iter__cCs t|jSr')nextrErQr!r!r"__next__cs zContainerIterator.__next__r')rrrrFunctionContainerTyper r rF ContainerKeyr#rrPrrrRrTrSr!r!r!r"r*4s  %r*namefr&cCs8|j}|dr||r|S||krd||fS|S)N__z ..)r startswithendswithrN)rWrX func_namer!r!r" _local_nameis r]c@sBeZdZdZdZd deedeeddfddZde fd d Z dS) _FunctionCodePairzFunction-Code Pair This class allows us to resolve a code object to a function object by querying the GC on-demand. )functioncodeNr`r_r&cCsF|dur|dur|j|urtd||_|dur|j|_dS||_dS)Nz&Function and code objects do not match)__code__ ValueErrorr_r`)r(r`r_r!r!r"rPsz_FunctionCodePair.__init__cCs|jdur tt|jS|jdurd}t||j}t|}t|}|dkr.d|}t||dkr;d|}t||d|_}z|j|_Wn tyQYnwtt|}|j d|j |_ |S)Nz'Cannot resolve pair with no code objectrz(Cannot resolve code object to function: z)Multiple functions found for code object r>) r_r r#r`rbrlenraAttributeErrorrrr)r(msgr` functionsn_frXr!r!r"resolves.        z_FunctionCodePair.resolve)NN) rrrr __slots__r rrrPr#rjr!r!r!r"r^vs  r^modulec Cst|}|dur iStt|g}i}t}t}|r|}t|j|vr'q|t|j|D]\}}t|t t frBt |ddnd} | durt t |} t |trWt||n|j} ||vrv||t t|}|jrsd|j| fn| |_t |tr|| kr|| fn| fD]-} |jrd|j| fn| } | |vst| |krtt t | |krt| | |n|d|| <qz| jr|t| j|dfdWq1tyYq1wt|trt|trt |jt sq1|t|||fdq1|s|S)zCollect functions from a given module. All the collected functions are augmented with a ``__fullname__`` attribute to disambiguate the same functions assigned to different names. Nrar>r_r0r)rrr*setpopleftidrMaddrrr r1r r?rr]rr#rrNrr^r __closure__appendreCONTAINER_TYPESrHr7)rlpath containersrgseen_containersseen_functionsckor`rX local_namerWfullnamer!r!r"_collect_functionssT    $   (r~cseZdZdZdeddffdd Zdedeefdd Z d e d e defd d Z de defddZ ededdfddZedededefddZZS)FunctionDiscoveryaDiscover all function objects in a module. The discovered functions can be retrieved by line number or by their qualified name. In principle one wants to create a function discovery object per module and then cache the information. For this reason, instances of this class should be obtained with the ``from_module`` class method. This builds the discovery object and caches the information on the module object itself. rlr&Nc s^ttt|}|durdS||_tdkrtt|_i|_t |dr^i|_ |j D]1}t |d}tdkrD|j d|j}||j |<n |j|j|t|D] }|||qQq*dSt||_ t}|j D]A\}}z|} Wn ty~Yqkw| |vrtttt| }|krttt|D]}||t tt| dq|| qkdS)N __dd_code__)r`r>rm)superrPlistr_modulerr _name_index_cachedhasattr_fullname_indexrr^r co_qualnameco_namersrr~rnrDrjrbrrr rrq) r(rl module_pathr`fcpr}linenorxr.r_ __class__r!r"rPsD           zFunctionDiscovery.__init__linec Csr||jvr |j|S||vr7g}||D]}z ||Wqty(Yqw|s0||=|S||j|<|SgS)zGet the functions at the given line. Note that, in general, there can be multiple copies of the same functions. This can happen as a result, e.g., of using decorators. )rrsrjrb)r(rrgrr!r!r"at_line%s     zFunctionDiscovery.at_linepairr}c Csz|WStyp}z_|j}}d}||jddddD]}zt||}Wq"ty4|w||ur;||j}|dusDJt t t |t t |t |}t|t r\|j|us^|||_t t|WYd}~Sd}~ww)Nr>rc)rjrbrreplacersplitr1rer`rr rrrr?rar_r#) r(rr}erltargetpartr`rXr!r!r" _resolve_pairAs,     zFunctionDiscovery._resolve_pairqualnamec Cs|jjd|}z ||j||WStyYnStyntdkrl|jD]@\}}||ks;| d|rkt |D]+}z |||}||j|<| d|j |kr`|WYSWq?tyjYq?wq+Ynwtd|)z'Get the function by its qualified name.r>rrzFunction '%s' not found) rrrrrbKeyErrorrrrDr[rpopr)r(rr}rWfcpsrrXr!r!r"by_nameas.         zFunctionDiscovery.by_namecCs<z|jWSty||}|_t|dr|`|YSw)zReturn a function discovery object from the given module. If this is called on a module for the first time, it caches the information on the module object itself. Subsequent calls will return the cached information. r)__function_discovery__rerr)clsrlfdr!r!r" from_modules   zFunctionDiscovery.from_moduler`cCst||_|Sr')rr)rr`rlr!r!r" transformers zFunctionDiscovery.transformer)rrrrrrPintrr#rr^rrrrIrrr __classcell__r!r!rr"rs 2 ! rN)4 collectionsrrtypesrrrtypingrrr r r r wraptr ddtrace.internal.compatrddtrace.internal.loggerrddtrace.internal.modulerddtrace.internal.safetyr!ddtrace.internal.utils.inspectionrrrrrddtrace.internal.wrappingrrlogr@rHrIrJrFrUrrrVrtrr#r*r]r^dictr~rr!r!r!r"sB                       5 7;