o Á¿i¸ã@sTUddlZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l mZe eƒZiZeeefed <d ed ed dfdd„Zded ededefd dfdd„Z  d'd ed ededefdeedfdeeeefd ef dd„Zdededed dfdd„Zd eed efd!d"„Zd eed#ed$ed dfd%d&„ZdS)(éN)ÚAny)ÚCallable)ÚOptional)ÚFunctionWrapper)Ú resolve_path)Ú_gc)Ú get_logger)ÚModuleWatchdogÚ_DD_ORIGINAL_ATTRIBUTESÚmoduleÚnameÚreturnc Csrz%t||ƒ\}}}||ftvr#t||f}t|||ƒt||f=WdSWdSttfy8t d||¡YdSw)NzERROR unwrapping %s.%s )rr Ú apply_patchÚModuleNotFoundErrorÚAttributeErrorÚlogÚdebug)r r ÚparentÚ attributeÚ_Úoriginal©rúO/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/appsec/_patch_utils.pyÚ try_unwraps   ýÿrÚ module_nameÚwrapper.cs(t ˆ¡dtddf‡‡‡fdd„ ƒ}dS)Nr r c s>z t|ˆtˆfƒWdSttfyt dˆˆ¡YdSw)NzModule %s.%s does not exist)Ú wrap_objectrÚ ImportErrorrrr)r ©rr rrrrs ÿz$try_wrap_function_wrapper.._)r Úafter_module_importedr)rr rrrrrÚtry_wrap_function_wrappers r rÚfactoryÚargsÚkwargscsP|duri}t||ƒ\}}}||g|¢Ri|¤Ž‰t||ˆƒ‡fdd„ˆ_ˆS)NcsˆS)Nr)Úmemo©rrrÚ2szwrap_object..)rrÚ __deepcopy__)r r r!r"r#rrrrr%rr&s rrrÚ replacementc Cs‚z.t||ƒ}t|tƒs|t||f<nt|tƒr&t|ddƒt|ddƒur&WdSt|||ƒWdSttfy@t|||ƒYdSw)NÚ _self_wrapper)ÚgetattrÚ isinstancerr ÚsetattrÚ TypeErrorrÚpatch_builtins)rrr(Úcurrent_attributerrrr6s   ÿrÚklasscCst |j¡}|dS)Nr)ÚgcÚ get_referentsÚ__dict__)r0ÚrefsrrrÚpatchable_builtinFs r5ÚattrÚvaluec Cs’t|ƒ}| |d¡}d|}|||<|r>|||<z|j||_Wn ttfy+Ynwz|j||_Wn ty=Ynwtj t  |¡¡dS)a@Based on forbiddenfruit package: https://github.com/clarete/forbiddenfruit/blob/master/forbiddenfruit/__init__.py#L421 --- Patch a built-in `klass` with `attr` set to `value` This function monkey-patches the built-in python object `attr` adding a new attribute to it. You can add any kind of argument to the `class`. It's possible to attach methods as class methods, just do the following: >>> def myclassmethod(cls): ... return cls(1.5) >>> curse(float, "myclassmethod", classmethod(myclassmethod)) >>> float.myclassmethod() 1.5 Methods will be automatically bound, so don't forget to add a self parameter to them, like this: >>> def hello(self): ... return self * 2 >>> curse(str, "hello", hello) >>> "yo".hello() "yoyo" Nz_c_%s) r5ÚgetÚ__name__rr-Ú __qualname__ÚctypesÚ pythonapiÚPyType_ModifiedÚ py_object)r0r6r7ÚdiktÚ old_valueÚold_namerrrr.Ks" ÿ ÿr.)rN)r;ÚtypingrrrÚwraptrrÚddtrace.internal._unpatchedrr1Úddtrace.internal.loggerrÚddtrace.internal.moduler r9rr ÚdictÚ__annotations__Ústrrr ÚtuplerrÚtyper5r.rrrrÚs>        "  ûÿþ ý üû ú"