o #f) @sdddlmZddlmZddlmZddlmZmZddl m Z m Z m Z ddl mZddlmZmZmZe dZeeZeeZeeZeeZ dd eed eeeffd d Zdded eed eefddZddeded eed efddZdded eed efddZdd eed efddZded e fddZ!ded e fddZ"deded e fddZ#d S) ) getLogger)compile)MappingProxyType)MappingOptional) create_key get_value set_value)Context)_BAGGAGE_PROPERTY_FORMAT _KEY_FORMAT _VALUE_FORMATbaggageNcontextreturncCs&tt|d}t|trt|StiS)zReturns the name/value pairs in the Baggage Args: context: The Context to use. If not set, uses current Context Returns: The name/value pairs in the Baggage r)r _BAGGAGE_KEY isinstancedictr)rrrZ/home/ubuntu/dev/venv/lib/python3.10/site-packages/mysql/opentelemetry/baggage/__init__.pyget_all$s rnamecCst|d|S)a=Provides access to the value for a name/value pair in the Baggage Args: name: The name of the value to retrieve context: The Context to use. If not set, uses current Context Returns: The value associated with the given name, or null if the given name is not present. r)rget)rrrrr get_baggage5s rvaluecCs$tt|d}|||<tt||dS)zSets a value in the Baggage Args: name: The name of the value to set value: The value to set context: The Context to use. If not set, uses current Context Returns: A Context with the value updated r)rrr r)rrrrrrr set_baggageDs rcCs(tt|d}||dtt||dS)zRemoves a value from the Baggage Args: name: The name of the value to remove context: The Context to use. If not set, uses current Context Returns: A Context with the name/value removed rN)rrpopr r)rrrrrrremove_baggageTs rcCstti|dS)zRemoves all values from the Baggage Args: context: The Context to use. If not set, uses current Context Returns: A Context with all baggage entries removed r)r rrrrrcleards rcCstt|duSN) _KEY_PATTERN fullmatchstr)rrrr _is_valid_keypsr$cCsZt|d}t|ddu}t|dkr+|ddD]}t|dur*d}|Sq|S)N;rF)r#split_VALUE_PATTERNr"len_PROPERT_PATTERN)rpartsis_valid_valuepropertyrrr_is_valid_valuets r.keycCst|ot|Sr )r$r.)r/rrrr_is_valid_pairsr0r )$loggingrrertypesrtypingrrmysql.opentelemetry.contextrrr #mysql.opentelemetry.context.contextr mysql.opentelemetry.util.rer r r r__name___loggerr!r(r*r#objectrrrrrboolr$r.r0rrrrs2