i ddlZddlZddlZddlmZddlmZddlmZddl m Z m Z ddl m Z ddlmZejeZdaejZejdejd ejf Zd ed efd Zded efdZddedeje d dfdZ ddedddeje d e fdZed e fdZ ede d e!fdZ"ede!d dfdZ#edZ$edZ%dS)N)wraps)environ)uuid4)Context_RuntimeContext)OTEL_PYTHON_CONTEXT) entry_points_F.)boundfuncreturncFtdtjtjtjfdtjtjtjfdtjtjffd }tjt|S)zvA decorator used to initialize the global RuntimeContext Returns: A wrapper of the decorated method. argskwargsr c |t5td}tjt|} t t td|an+#t$rt d|YnwxYwdddn #1swxYwY|i|S)Ncontextvars_contextopentelemetry_context)groupnamezFailed to load context: %s) _RUNTIME_CONTEXT_LOCK_RUNTIME_CONTEXTrgetrnextiterr load Exceptionlogger exception)rrdefault_contextconfigured_contextr s ^/var/www/cyber-strat/venv/lib/python3.11/site-packages/mysql/opentelemetry/context/__init__.pywrapperz&_load_runtime_context..wrapper*sB# W W'#8%,['&&" W(t(&=%7((dff(($$!WWW$$%ACUVVVVVW) W W W W W W W W W W W W W W W,tT$V$$$s5$B+AA43B+4%BB+BB++B/2B/)rtypingTupleAnyDictOptionalcastr )r r"s` r!_load_runtime_contextr)#s 4[[%|FJ 23%+fj&*45%  $%%%%%[%< ;r7 # ##keynamecD|dzttzS)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+s r! create_keyr/Ls S=3uww<< ''r*keycontextobjectcr|||n t|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. )r get_current)r0r1s r! get_valuer5Xs2 '27;;s    8I8I#8N8NNr*valuecr|t}|}|||<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. )r4copyr)r0r6r1 new_valuess r! set_valuer:gs6 --JJsO :  r*c4tS)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. )rr4r*r!r4r4~s  ' ' ) ))r*c6t|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)r1s r!r>r>s  " "7 + ++r*tokenc t|dS#t$rtdYdSwxYw)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)r?s r!rArAsZ5&&&&& 55534444445s$AAsuppress_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'r5r:r4r2r>rA_SUPPRESS_INSTRUMENTATION_KEY"_SUPPRESS_HTTP_INSTRUMENTATION_KEYr<r*r!rSs^ IHHHHHHHIIIIIIEEEEEE  8 $ $& ((V^DVZ @AAA&$&$r&$&$&$&$R ( ( ( ( ( ( O O3 O!9 OX O O O O DH (.(@ .*W**** ,G , , , , , 5& 5T 5 5 5 5 !+ +E F F%/Z0O%P%P"""r*