o gz @sddlZddlZddlmZddlmZddlmZmZddl m Z ddl m Z e eZdefdd ZeZd edefd d Zdd edejeddfddZ dd edddejedefddZdefddZdedefddZdeddfddZedZedZgdZdS) N)environ)uuid4)Context_RuntimeContext)OTEL_PYTHON_CONTEXT) entry_pointsreturnc Cshd}tt|}ztttd|dWSty3t d||tttd|dYSw)zWInitialize the RuntimeContext Returns: An instance of RuntimeContext. contextvars_contextopentelemetry_context)groupnamez*Failed to load context: %s, fallback to %s) rgetrnextiterrload Exceptionlogger exception)default_contextconfigured_contextrU/var/www/visachat/venv/lib/python3.10/site-packages/opentelemetry/context/__init__.py_load_runtime_contexts<  rkeynamecCs|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)rrrr create_keyFs rkeycontextobjectcCs|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)rrrrr get_valueRs r!valuecCs&|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)r copyr)rr"r new_valuesrrr set_valueas r%cCstS)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. )_RUNTIME_CONTEXTr rrrrr xsr cCs 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. )r&attach)rrrrr's r'tokencCs0zt|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)r&detachrrr)r(rrrr)s  r)suppress_instrumentationsuppress_http_instrumentation)rr'rr)r r!r%)N)loggingtypingosruuidropentelemetry.context.contextrr#opentelemetry.environment_variablesr&opentelemetry.util._importlib_metadatar getLogger__name__rrr&rrOptionalr!r%r rr'r)_SUPPRESS_INSTRUMENTATION_KEY"_SUPPRESS_HTTP_INSTRUMENTATION_KEY__all__rrrrs<     '