Current File : /opt/cloudlinux/venv/lib/python3.11/site-packages/setoptconf/source//environment.py
import os
from ..config import Configuration
from ..setting import ListSetting
from ..util import csv_to_list
from .base import Source
__all__ = ("EnvironmentVariableSource",)
class EnvironmentVariableSource(Source):
def __init__(self, prefix=None):
super(EnvironmentVariableSource, self).__init__()
self.prefix = prefix
def get_config(self, settings, manager=None, parent=None):
if manager and not self.prefix:
self.prefix = manager.name
for setting in settings:
self.get_setting(setting)
return Configuration(settings=settings, parent=parent)
def get_setting(self, setting):
name = setting.name
if self.prefix:
name = "%s_%s" % (self.prefix, name)
name = name.upper()
if name in os.environ:
if isinstance(setting, ListSetting):
setting.value = csv_to_list(os.environ[name])
else:
setting.value = os.environ[name]
