o #f @sLddlZddlZddlZddlmZddlmZddlmZddl m Z m Z ddl m Z ddlmZeeZdaeZejdejd ejfd Zd ed efd dZded efddZd#dedeje d dfddZ d#dedddeje d e fddZed e fddZ ede d e!fddZ"ede!d dfdd Z#ed!Z$ed"Z%dS)$N)wraps)environ)uuid4)Context_RuntimeContext)OTEL_PYTHON_CONTEXT) entry_points_F.)boundfuncreturncsRtdtjtjtjfdtjtjtjfdtjtjffdd }tt|S)zvA decorator used to initialize the global RuntimeContext Returns: A wrapper of the decorated method. argskwargsr c st3tdur.d}tt|}ztttd|daWnt y-t d|YnwWdn1s8wY|i|S)Ncontextvars_contextopentelemetry_context)groupnamezFailed to load context: %s) _RUNTIME_CONTEXT_LOCK_RUNTIME_CONTEXTrgetrnextiterrload Exceptionlogger exception)r rdefault_contextconfigured_contextr Z/home/ubuntu/dev/venv/lib/python3.10/site-packages/mysql/opentelemetry/context/__init__.pywrapper*s. z&_load_runtime_context..wrapper)rtypingTupleAnyDictOptionalcastr )r r!rrr _load_runtime_context#s  r(keynamecCs|dttS)aeTo allow cross-cutting concern to control access to their local state, the RuntimeContext API provides a function which takes a keyname as input, and returns a unique key. Args: keyname: The key name is for debugging purposes and is not required to be unique. Returns: A unique string representing the newly created key. -)strr)r)rrr create_keyLs r,keycontextobjectcCs|dur ||St|S)avTo access the local state of a concern, the RuntimeContext API provides a function which takes a context and a key as input, and returns a value. Args: key: The key of the value to retrieve. context: The context from which to retrieve the value, if None, the current context is used. Returns: The value associated with the key. N)r get_current)r-r.rrr get_valueXs r1valuecCs&|durt}|}|||<t|S)aTo record the local state of a cross-cutting concern, the RuntimeContext API provides a function which takes a context, a key, and a value as input, and returns an updated context which contains the new value. Args: key: The key of the entry to set. value: The value of the entry to set. context: The context to copy, if None, the current context is used. Returns: A new `Context` containing the value set. N)r0copyr)r-r2r. new_valuesrrr set_valuegs r5cCstS)zTo access the context associated with program execution, the Context API provides a function which takes no arguments and returns a Context. Returns: The current `Context` object. )rr0rrrr r0~s r0cCs t|S)aAssociates a Context with the caller's current execution unit. Returns a token that can be used to restore the previous Context. Args: context: The Context to set as current. Returns: A token that can be used with `detach` to reset the context. )rattach)r.rrr r6s r6tokencCs0zt|WdStytdYdSw)zResets the Context associated with the caller's current execution unit to the value it had before attaching a specified Context. Args: token: The Token that was returned by a previous call to attach a Context. zFailed to detach contextN)rdetachrrr)r7rrr r8s  r8suppress_instrumentationsuppress_http_instrumentation)N)&logging threadingr" functoolsrosruuidr#mysql.opentelemetry.context.contextrr)mysql.opentelemetry.environment_variablesr,mysql.opentelemetry.util._importlib_metadatar getLogger__name__rrLockrTypeVarCallabler$r r(r+r,r&r1r5r0r/r6r8_SUPPRESS_INSTRUMENTATION_KEY"_SUPPRESS_HTTP_INSTRUMENTATION_KEYrrrr sD      )