o Á¿iíã@sŠddlZddlZddlmZddlmZddlmZejZejj Z ejj Z de fdd„Zdee e ffd d „Zd d „Zd d„Zdd„ZdS)éNé)ÚTracedGreenlet)Ú TracedIMap)ÚTracedIMapUnorderedÚreturncCs ttddƒS)NÚ __version__Ú)ÚgetattrÚgevent©r r úY/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/contrib/internal/gevent/patch.pyÚ get_versions r cCsddiS)Nr z>=21.1.2r r r r r Ú_supported_versionssrcCs&ttddƒrdSdt_ttttƒdS)a  Patch the gevent module so that all references to the internal ``Greenlet`` class points to the ``DatadogGreenlet`` class. This action ensures that if a user extends the ``Greenlet`` class, the ``TracedGreenlet`` is used as a parent class. Ú__datadog_patchFNT)r r rÚ_replacerrrr r r r Úpatchs rcCs&ttddƒsdSdt_ttttƒdS)zÀ Restore the original ``Greenlet``. This function must be invoked before executing application code, otherwise the ``DatadogGreenlet`` class may be used during initialization. rFN)r r rrÚ __GreenletÚ__IMapÚ__IMapUnorderedr r r r Úunpatch&s rcCsh|tj_|tjj_|tj_|tj_tjjt_tjjjt_tjjjt_|tj_ |tj_ |tj_ |tj_ dS)zU Utility function that replace the gevent Greenlet class with the given one. N) r ÚgreenletÚGreenletÚpoolÚGroupÚgreenlet_classÚ_imapÚspawnÚ spawn_laterÚIMapÚ IMapUnordered)Úg_classÚ imap_classÚimap_unordered_classr r r r3s     r)r Ú gevent.poolrrrrrrrrrrrÚstrr Údictrrrrr r r r Ús