o i=4 @sddlmZddlZddlZddlmZddlmZddlm Z ddlm Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZd dlmZd dlmZd dlmZd dlmZeeZ iddddddddddddddddddddddddddd dd!dd"dd#did$dd%dd&dd'dd(dd)dd*dd+dd,dd-dd.dd/dd0dd1dd2dd3dd4did5dd6dd7dd8dd9dd:d;ddd?dd@ddAd;dBd;dCd;dDddEddFdidGddHddIddJd;dKddLddMddNddOddPddQddRddSddTddUddVddWdidXddYddZdd[dd\d;d]dd^dd_dd`ddaddbddcddddded;dfddgddhdd;ej!ddiZ"d\djiZ#e$Z%id dkd1dld:dmdUdnd-dod"dpdEdqdXdrdYdsdZdtd[dudBdvd)dwd#dxd$dyd^dzd*d{dhd|iZ&dhZ'Gd}d~d~e(Z)Gddde)Z*Gddde)Z+de,de,de-fddZ.dede,de e,dffddZ/dede,de,de e,dffddZ0dede,de,ddfddZ1 dde,de,de-de e-e2e,fdee gdff ddZ3de-ddfddZ4de-ddfddZ5dde-de e2e,e-fddfddZ6de$e,fddZ7dS))CallableN)Path) ModuleType)Any)Union) when_imported)config)TELEMETRY_NAMESPACE) deprecate) SpecifierSet)Version) telemetry) get_logger)formats)DDTraceDeprecationWarningaiokafkaTaiomysqlaredisasyncioavrobotobotocorebottleceleryconsul ddtrace_apidjangodramatiq elasticsearch algoliasearchfutures google_adk google_genaigeventgraphqlgrpchttpxkafka langgraphlitellmmysqlmysqldbpymysqlmariadbmcppsycopgpylibmc pymemcachepymongoredis redisclusterrequestsrqsanic snowflakeF sqlalchemysqlite3aiohttpaiohttp_jinja2aiopg aiobotocorehttpliburllib3vertexaiverticamoltenjinja2makoflaskkombu starlettefalconpyramidlogbookloggingloguru structlogpynamodbpyodbcfastapi dogpile_cacheyaaredisasyncpg aws_lambdaazure_eventhubsazure_functionsazure_servicebustornadoopenai langchain anthropiccrewai pydantic_aivllm subprocessunittestcoverageseleniumvalkey openai_agents)rayprotobufclaude_agent_sdk)r!)relasticsearch1elasticsearch2elasticsearch5elasticsearch6elasticsearch7elastic_transport opensearchpy)r0psycopg2)zsnowflake.connector)z dogpile.cache)MySQLdb)zconcurrent.futures.thread)vertica_python)datadog_lambda)zazure.eventhub)zazure.functions)zazure.servicebus)z http.client)confluent_kafka)z google.adk)z google.genai)langchain_core)r)zlanggraph.graphzlanggraph.prebuilt)agentsc@seZdZdZdS)PatchExceptionz5Wraps regular `Exception` class when patching modulesN)__name__ __module__ __qualname____doc__r}r}C/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/_monkey.pyrxsrxc@s eZdZdS)ModuleNotFoundExceptionN)ryrzr{r}r}r}r~rsrcs0eZdZddedeedfffdd ZZS)IncompatibleModuleExceptionNmessageinstalled_versioncst|||_dSN)super__init__r)selfrr __class__r}r~rs  z$IncompatibleModuleException.__init__r)ryrzr{strrr __classcell__r}r}rr~rs(rversionsupported_versions_specreturncCs@|sdS|dkr dSz t|}t||vWStyYdSw)zeReturns whether a given package version is compatible with the integration's supported version range.F*T)r r Exception)rr specifier_setr}r}r~is_version_compatibles rimported_modulehooked_module_namecCs.t|dr ||St|dr|SdS)z*Returns the installed version of a module. get_versions get_versionN)hasattrrgetr)rrr}r}r~_get_installed_module_versions  rintegration_patch_moduleintegration_namecCs:t|dsdS|}||vr||S||vr||SdS)z7Returns the supported version range for an integration._supported_versionsN)rr)rrrsupported_versionsr}r}r~#_get_integration_supported_versionss rcCsXt||}|s dSt|||}|sdSt||s*d|d|d|d}t||ddS)zrDetermines if a module should be patched based on installed version and the integration's supported version range.NzSkipped patching 'z"' integration, installed version: z2 is not compatible with integration support spec: .)r)rrrr)rrrrsupported_version_specrr}r}r~check_module_compatibilitys   rmodulepath_f raise_errorspatch_indicatorcsfdd}|S)z=Factory to create an import hook for the provided module namec sz"tf}tjrt||j|t|dr!|Wnrt yQ}z#t dt |t jjdtdut ||jdWYd}~dSd}~wty}z8r[t j dt |ddt jdtdut |t jtjddd fd t|jffWYd}~dSd}~wwt|d r|}|D]\}}t jj|dtdud |dqdSt|d r|}t jjdtdud |ddSdS)Npatch_submodulesz+failed to enable ddtrace support for %s: %sFT)r)exc_infointegration_errorsr r error_typerr) importlib import_moduler#_trace_safe_instrumentation_enabledrrypatchrrrlogerrorrrtelemetry_writeradd_integration PATCH_MODULESrrradd_count_metricr TRACERStyperitemsr)hookreversionsnamevrrrrrr}r~ on_importsb      z%_on_import_factory..on_importr})rrrrrr}rr~_on_import_factorys5r patch_modulescKs tddtdtdi|dS)aEnables ddtrace library instrumentation. In addition to ``patch_modules``, an override can be specified via an environment variable, ``DD_TRACE__ENABLED`` for each module. ``patch_modules`` have the highest precedence for overriding. :param dict patch_modules: Override whether particular modules are patched or not. >>> _patch_all(redis=False) zNpatch_all is deprecated and will be removed in a future version of the tracer.z|patch_all is deprecated in favor of ``import ddtrace.auto`` and ``DD_PATCH_MODULES`` environment variable if needed.)rcategoryNr})r r _patch_all)rr}r}r~ patch_all?s  rcKst}|D].\}}d|}|tvr%|tjvr%ttj|||<||r6t |dD]}d||<q/q| |t dddi|dS)NzDD_TRACE_%s_ENABLEDr}TrF) rcopyrupper_NOT_PATCHABLE_VIA_ENVVARosenvironrasboolCONTRIB_DEPENDENCIESrupdater)rmodulesr_enabledenv_vardepr}r}r~rTs   rc Ksdd|D}|D]9\}}ttjdd|ds)|r)t|dt||f}|D]}t|t |d||dq2t |q t d t|d |d S) zPatch only a set of given modules. :param bool raise_errors: Raise error if one patch fail. :param dict patch_modules: List of modules to patch. >>> patch(psycopg=True, elasticsearch=True) cSsi|] \}}|r||qSr}r}).0crr}r}r~ pszpatch..contribinternalzpatch.pyz( does not have automatic instrumentationz!ddtrace.contrib.internal.%s.patch)rrzeConfigured ddtrace instrumentation for %s integration(s). The following modules have been patched: %s,N)rr__file__parentexistsr_MODULES_FOR_CONTRIBrrr_PATCHED_MODULESaddrinfolenjoin)rrcontribsrrmodules_to_patchrr}r}r~rhs* rcCstS)zGet the list of patched modules)rr}r}r}r~_get_patched_modulessr)TT)T)8collections.abcrrrpathlibrtypesrtypingrrwrapt.importerr!ddtrace.internal.settings._configr$ddtrace.internal.telemetry.constantsr ddtrace.vendor.debtcollectorr #ddtrace.vendor.packaging.specifiersr ddtrace.vendor.packaging.versionr rrinternal.loggerrinternal.utilsrinternal.utils.deprecationsrryr_data_streams_enabledrrsetrrrrrxrrrboolrrrrlistrrrrrr}r}r}r~s"                     !"#$%&'()*+,-./012345678:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWa $'      =$$