o i@sdZddlmZddlmZmZddlmZddlm Z ddl m Z ddl m ZmZmZddlmZdd lZdd lmZeee fZd Zd ddZedd dd!ddZe GdddeZGdddeZe GdddeZd S)"aThis module provides a thin wrapper around the OpenTelemetry propagate API to allow the OpenTelemetry contexts (and therefore Logfire contexts) to be transferred between different code running in different threads, processes or even services. In general, you should not need to use this module since Logfire will automatically patch [`ThreadPoolExecutor`][concurrent.futures.ThreadPoolExecutor] and [`ProcessPoolExecutor`][concurrent.futures.ProcessPoolExecutor] to carry over the context. And existing plugins exist to propagate the context with [requests](https://pypi.org/project/opentelemetry-instrumentation-requests/) and [httpx](https://pypi.org/project/opentelemetry-instrumentation-httpx/). ) annotations)IteratorMapping)contextmanager) dataclass)Any)context propagatetrace)TextMapPropagatorN)warn_at_user_stacklevel) get_contextattach_contextContextCarrierreturnrcCsi}t||S)a<Create a new empty carrier dict and inject context into it. Returns: A new dict with the context injected into it. Usage: ```py import logfire logfire_context = logfire.get_context() ... # later on in another thread, process or service with logfire.attach_context(logfire_context): ... ``` You could also inject context into an existing mapping like headers with: ```py import logfire existing_headers = {'X-Foobar': 'baz'} existing_headers.update(logfire.get_context()) ... ``` )r injectcarrierrW/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/logfire/propagate.pyr s r F) third_party propagatorrrboolrTextMapPropagator | NoneIterator[None]c cst}|dur t}|s t|ttfr |j}t|ttfs|s$J|j|d}zt |dVWt |dSt |w)aAttach a context as generated by [`get_context`][logfire.propagate.get_context] to the current execution context. Since `attach_context` is a context manager, it restores the previous context when exiting. Set `third_party` to `True` if using this inside a library intended to be used by others. This will respect the [`distributed_tracing` argument of `logfire.configure()`][logfire.configure(distributed_tracing)], so users will be warned about unintentional distributed tracing by default and they can suppress it. See [Unintentional Distributed Tracing](https://logfire.pydantic.dev/docs/how-to-guides/distributed-tracing/#unintentional-distributed-tracing) for more information. Nr) otel_context get_currentr get_global_textmap isinstance#WarnOnExtractTraceContextPropagatorNoExtractTraceContextPropagatorwrappedextractattach)rrr old_context new_contextrrrrCs  rc@s:eZdZUdZded<dd d Zdd d Zed dZdS)WrapperPropagatorz-Helper base class to wrap another propagator.r r!argsrkwargsrotel_context.ContextcO|jj|i|SN)r!r"selfr'r(rrrr"izWrapperPropagator.extractcOr*r+)r!rr,rrrrlr.zWrapperPropagator.injectcCs|jjSr+)r!fields)r-rrrr/oszWrapperPropagator.fieldsN)r'rr(rrr))r'rr(r) __name__ __module__ __qualname____doc____annotations__r"rpropertyr/rrrrr&cs   r&cs&eZdZdZ d dfd d ZZS)r zA propagator that ignores any trace context that was extracted by the wrapped propagator. Used when `logfire.configure(distributed_tracing=False)` is called. Nrrrotel_context.Context | Noner'r(rr)cs:tj||g|Ri|}||kr|Stt||Sr+)superr"r set_span_in_contextget_current_span)r-rrr'r(result __class__rrr"zsz'NoExtractTraceContextPropagator.extractr+ rrrr6r'rr(rrr))r0r1r2r3r" __classcell__rrr;rr tsr cs4eZdZUdZdZded< ddfdd ZZS)rzA propagator that warns the first time that trace context is extracted by the wrapped propagator. Used when `logfire.configure(distributed_tracing=None)` is called. This is the default behavior. FrwarnedNrrrr6r'r(rr)cs`tj||g|Ri|}|js.||kr.t|t|kr.d|_d}t|tt||S)NTzFound propagated trace context. See https://logfire.pydantic.dev/docs/how-to-guides/distributed-tracing/#unintentional-distributed-tracing.) r7r"r?r r9r RuntimeWarninglogfirewarn)r-rrr'r(r:messager;rrr"s"  z+WarnOnExtractTraceContextPropagator.extractr+r=)r0r1r2r3r?r4r"r>rrr;rrs  r)rr)rrrrrrrr)r3 __future__rcollections.abcrr contextlibr dataclassesrtypingr opentelemetryrrr r !opentelemetry.propagators.textmapr rAlogfire._internal.stack_infor strr__all__r rr&r rrrrrs,        #