Source code for multistorageclient.telemetry.attributes.environment_variables

 1# SPDX-FileCopyrightText: Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 2# SPDX-License-Identifier: Apache-2.0
 3#
 4# Licensed under the Apache License, Version 2.0 (the "License");
 5# you may not use this file except in compliance with the License.
 6# You may obtain a copy of the License at
 7#
 8# http://www.apache.org/licenses/LICENSE-2.0
 9#
10# Unless required by applicable law or agreed to in writing, software
11# distributed under the License is distributed on an "AS IS" BASIS,
12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13# See the License for the specific language governing permissions and
14# limitations under the License.
15
16import os
17from collections.abc import Mapping
18
19import opentelemetry.util.types as api_types
20
21from .base import AttributesProvider
22
23

[docs]
24class EnvironmentVariablesAttributesProvider(AttributesProvider):
25    """
26    Provides :py:type:`opentelemetry.util.types.Attributes` from environment variables.
27    """
28
29    #: Attribute key to environment variable key map.
30    _attributes: Mapping[str, str]
31
32    def __init__(self, attributes: Mapping[str, str]):
33        """
34        :param attributes: Map of attribute key to environment variable key.
35        """
36        self._attributes = attributes
37

[docs]
38    def attributes(self) -> api_types.Attributes:
39        return {
40            attribute_key: os.environ[environment_variable_key]
41            for attribute_key, environment_variable_key in self._attributes.items()
42            if environment_variable_key in os.environ
43        }