o i@szUddlZddlZddlmZddlmZddlmZej ej eefZ ej dddZ ej e ed<Gdd d eZdS) N)Context)DefaultContextProvider)Spandatadog_civisibility_contextvar)default_DD_CI_CONTEXTVARcsLeZdZdZfddZddZdeddffd d Zdefd d ZZ S) CIContextProviderzContext provider that retrieves contexts from a context variable. It is suitable for synchronous programming and for asynchronous executors that support contextvars. cstt|tddS)N)superr__init__rset)self __class__Z/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/internal/ci_visibility/context.pyr szCIContextProvider.__init__cCst}|duS)zDReturns whether there is an active context in the current execution.N)rgetr ctxrrr_has_active_contextsz%CIContextProvider._has_active_contextrreturnNcst|tt||dS)z8Makes the given context active in the current execution.N)rr r ractivaterr rrr$s zCIContextProvider.activatecCs t}t|tr||S|S)z=Returns the active span or context for the current execution.)rr isinstancer_update_active)r itemrrractive)s  zCIContextProvider.active) __name__ __module__ __qualname____doc__r rContextTypeValuerr __classcell__rrr rrs  r) contextvarstypingtddtrace._trace.contextrddtrace._trace.providerrddtrace._trace.spanrOptionalUnionr ContextVarr__annotations__rrrrrs