@@ -7,21 +7,12 @@ class EnvironmentGetter(Getter[dict]):
77 from environment variables.
88 """
99
10- KEY_MAPPING = {
11- "TRACEPARENT" : "traceparent" ,
12- "TRACESTATE" : "tracestate" ,
13- "BAGGAGE" : "baggage"
14- }
15-
1610 def __init__ (self ):
1711 self .env_copy = dict (os .environ )
1812 self .carrier = {}
1913
2014 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
2516
2617 def get (self , carrier : dict , key : str ) -> typing .Optional [typing .List [str ]]:
2718 """Get a value from the carrier for the given key"""
@@ -40,12 +31,6 @@ class EnvironmentSetter(Setter[dict]):
4031 """This class decorates Setter to enable setting context and baggage
4132 to environment variables.
4233 """
43-
44- KEY_MAPPING = {
45- "TRACEPARENT" : "traceparent" ,
46- "TRACESTATE" : "tracestate" ,
47- "BAGGAGE" : "baggage"
48- }
4934
5035 def set (self , carrier : typing .Optional [dict ], key : str , value : str ) -> None :
5136 """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:
5540 key: The key to set
5641 value: The value to set
5742 """
58- env_key = self . KEY_MAPPING . get ( key , key .upper () )
43+ env_key = key .upper ()
5944
6045 os .environ [env_key ] = value
0 commit comments