U 4Je)@sHddlmZddlmZddlmZddlmZmZddl m Z m Z m Z ddl mZddlmZmZmZe dZeeZeeZeeZeeZd eeeeefd d d Zd!eeeeed ddZd"eeeeedddZd#eeeed ddZd$eeed ddZee dddZ!ee dddZ"eee dddZ#d S)%) getLogger)compile)MappingProxyType)MappingOptional) create_key get_value set_value)Context)_BAGGAGE_PROPERTY_FORMAT _KEY_FORMAT _VALUE_FORMATbaggageN)contextreturncCs&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)rrrH/tmp/pip-unpacked-wheel-7_167w8m/mysql/opentelemetry/baggage/__init__.pyget_all$s  r)namerrcCst|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 r)rvaluerrcCs$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 r)rrcCstt|dk SN) _KEY_PATTERN fullmatchstr)rrrr _is_valid_keypsr$)rrcCsXt|d}t|ddk }t|dkrT|ddD]}t|dkr8d}qTq8|S)N;rF)r#split_VALUE_PATTERNr"len_PROPERT_PATTERN)rpartsZis_valid_valuepropertyrrr_is_valid_valuets r-)keyrrcCst|ot|Sr )r$r-)r.rrrr_is_valid_pairsr/)N)N)N)N)N)$loggingrrertypesrtypingrrZmysql.opentelemetry.contextrrr Z#mysql.opentelemetry.context.contextr Zmysql.opentelemetry.util.rer r r r__name__Z_loggerr!r(r*r#objectrrrrrboolr$r-r/rrrrs.