U 4Je@szddlmZddlmZmZmZddlmZddlm Z m Z dZ edZ d e eeeddd Z deee d d d ZdS))Optional) create_key get_value set_value)Context) INVALID_SPANSpanz current-spanN)spancontextreturncCstt||d}|S)zSet the span in the given context. Args: span: The Span to set. context: a Context object. if one is not passed, the default current context is used instead. r )r _SPAN_KEY)r r ctxrR/tmp/pip-unpacked-wheel-7_167w8m/mysql/opentelemetry/trace/propagation/__init__.pyset_span_in_contextsr)r r cCs&tt|d}|dkst|ts"tS|S)zRetrieve the current span. Args: context: A Context object. If one is not passed, the default current context is used instead. Returns: The Span set in the context if it exists. INVALID_SPAN otherwise. r N)rr isinstancerr)r r rrrget_current_span$s r)N)N)typingrZmysql.opentelemetry.contextrrrZ#mysql.opentelemetry.context.contextrZmysql.opentelemetry.trace.spanrrZSPAN_KEYr rrrrrrs