@@ -7,21 +7,12 @@ class EnvironmentGetter(Getter[dict]):
7
7
from environment variables.
8
8
"""
9
9
10
- KEY_MAPPING = {
11
- "TRACEPARENT" : "traceparent" ,
12
- "TRACESTATE" : "tracestate" ,
13
- "BAGGAGE" : "baggage"
14
- }
15
-
16
10
def __init__ (self ):
17
11
self .env_copy = dict (os .environ )
18
12
self .carrier = {}
19
13
20
14
for env_key , env_value in self .env_copy .items ():
21
- if env_key in self .KEY_MAPPING :
22
- self .carrier [self .KEY_MAPPING [env_key ]] = env_value
23
- else :
24
- self .carrier [env_key ] = env_value
15
+ self .carrier [env_key .lower ()] = env_value
25
16
26
17
def get (self , carrier : dict , key : str ) -> typing .Optional [typing .List [str ]]:
27
18
"""Get a value from the carrier for the given key"""
@@ -40,12 +31,6 @@ class EnvironmentSetter(Setter[dict]):
40
31
"""This class decorates Setter to enable setting context and baggage
41
32
to environment variables.
42
33
"""
43
-
44
- KEY_MAPPING = {
45
- "TRACEPARENT" : "traceparent" ,
46
- "TRACESTATE" : "tracestate" ,
47
- "BAGGAGE" : "baggage"
48
- }
49
34
50
35
def set (self , carrier : typing .Optional [dict ], key : str , value : str ) -> None :
51
36
"""Set a value in the environment for the given key.
@@ -55,6 +40,6 @@ def set(self, carrier: typing.Optional[dict], key: str, value: str) -> None:
55
40
key: The key to set
56
41
value: The value to set
57
42
"""
58
- env_key = self . KEY_MAPPING . get ( key , key .upper () )
43
+ env_key = key .upper ()
59
44
60
45
os .environ [env_key ] = value
0 commit comments