o i @sddlZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd l mZd efd d Zd eeeffd dZddZddZddZdS)N)Any)Pin)core)get_argument_value)set_argument_value)unwrap)wrap)tracerreturncCsdS)Nr r r Z/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/contrib/internal/asyncio/patch.py get_version srcCsddiS)Nasyncio*r r r r r _supported_versionssrcCs4ttddrdSdt_ttttjjtdS)zpPatches current loop `create_task()` method to enable spawned tasks to parent to the base task context. _datadog_patchFNT) getattrrrrontor BaseEventLoop create_task_wrapped_create_taskr r r r patchs  rcCs(ttddrdSdt_ttjjtdS)z$Remove tracing from patched modules.rFN)rrrrrrrr r r r unpatch s rcstt}|r |s||i|Sitdfts(||i|St||ddfdd}|}t drCj |_ t drLj |_ t ||dd|\}}||i|S)zThis function ensures the current active trace context is propagated to scheduled tasks. By default the trace context is propagated when a task is executed and NOT when it is created. zasyncio.create_taskcorocs2tkr tjtdfIdHS)Nzasyncio.execute_task)r current_trace_contextcontext_provideractivaterdispatch)args_ckwargs_cr dd_active task_datar r traced_coro>s    z)_wrapped_create_task..traced_coro__name__ __qualname__) rget_fromrenabledrrr rrhasattrr&r'r)wrappedargskwargspinr%tcr r"r r)s"    r)rtypingrddtrace._trace.pinrddtrace.internalrddtrace.internal.utilsrrddtrace.internal.wrappingrr ddtrace.tracer strrdictrrrrr r r r s