o #f@s6ddlmZddlmZmZGdddeZdgZdS)) ContextVar)Context_RuntimeContextc@sPeZdZdZdZdddZdedefdd Zdefd d Z d eddfd dZ dS)ContextVarsRuntimeContextzAn implementation of the RuntimeContext interface which wraps ContextVar under the hood. This is the preferred implementation for usage with Python 3.5+ current_contextreturnNcCst|jtd|_dS)N)default)r _CONTEXT_KEYr_current_contextselfr e/home/ubuntu/dev/venv/lib/python3.10/site-packages/mysql/opentelemetry/context/contextvars_context.py__init__sz"ContextVarsRuntimeContext.__init__contextcCs |j|S)zSets the current `Context` object. Returns a token that can be used to reset to the previous `Context`. Args: context: The Context to set. )r set)r rr r rattachs z ContextVarsRuntimeContext.attachcCs |jS)z%Returns the current `Context` object.)r getr r r r get_current&s z%ContextVarsRuntimeContext.get_currenttokencCs|j|dS)zpResets Context to a previous value Args: token: A reference to a previous Context. N)r reset)r rr r rdetach*sz ContextVarsRuntimeContext.detach)rN) __name__ __module__ __qualname____doc__r rrobjectrrrr r r rrs  rN) contextvarsr#mysql.opentelemetry.context.contextrrr__all__r r r rs