File: environment.py

package info (click to toggle)
python-setoptconf 0.3.0-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 252 kB
  • sloc: python: 1,427; sh: 7; makefile: 5
file content (38 lines) | stat: -rw-r--r-- 1,007 bytes parent folder | download | duplicates (5)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
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]