| 123456789101112131415161718192021222324252627282930313233343536 |
- 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]
|
