Update environment variable carrier

Author: adrielp

opentelemetry-api/src/opentelemetry/propagators/envcarrier.py

@@ -1,19 +1,63 @@
+# Copyright The OpenTelemetry Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 import os
 import typing
-from opentelemetry.propagators.textmap import Getter, Setter
+from opentelemetry.propagators.textmap import Getter, Setter, TextMapPropagator
+from typing_extensions import deprecated
+
+class EnvironmentVariableCarrier(dict):
+    """
+    A carrier for environment variables.
+
+    This class provides a dictionary-like interface for accessing and modifying
+    environment variables. It is used by the `EnvironmentVariablePropagator`
+    to inject and extract context information from the environment.
+    """
+
+    def __init__(self):
+        super().__init__()
+        self._environ = os.environ
+
+    def __getitem__(self, key: str) -> str:
+        return self._environ.get(key, super().__getitem__(key))
+
+    def __setitem__(self, key: str, value: str) -> None:
+        super().__setitem__(key, value)
+
+    def __delitem__(self, key: str) -> None:
+        super().__delitem__(key)
 
+    def __iter__(self) -> typing.Iterator[str]:
+        return iter(set(self._environ.keys()) | set(super().keys()))
+
+
+@deprecated(
+    "opentelemetry.propagators.envcarrier.EnvironmentGetter will be removed in a future version.",
+)
 class EnvironmentGetter(Getter[dict]):
     """This class decorates Getter to enable extracting from context and baggage
     from environment variables.
     """
-    
+
     def __init__(self):
         self.env_copy = dict(os.environ)
         self.carrier = {}
-        
+
         for env_key, env_value in self.env_copy.items():
             self.carrier[env_key.lower()] = env_value
-    
+
     def get(self, carrier: dict, key: str) -> typing.Optional[typing.List[str]]:
         """Get a value from the carrier for the given key"""
         val = self.carrier.get(key, None)
@@ -22,11 +66,15 @@ def get(self, carrier: dict, key: str) -> typing.Optional[typing.List[str]]:
         if isinstance(val, typing.Iterable) and not isinstance(val, str):
             return list(val)
         return [val]
-    
+
     def keys(self, carrier: dict) -> typing.List[str]:
         """Get all keys from the carrier"""
         return list(self.carrier.keys())
 
+
+@deprecated(
+    "opentelemetry.propagators.envcarrier.EnvironmentSetter will be removed in a future version.",
+)
 class EnvironmentSetter(Setter[dict]):
     """This class decorates Setter to enable setting context and baggage
     to environment variables.
@@ -41,5 +89,5 @@ def set(self, carrier: typing.Optional[dict], key: str, value: str) -> None:
             value: The value to set
         """
         env_key = key.upper()
-        
+
         os.environ[env_key] = value