Project

General

Profile

1
# Program options processing
2

    
3
import os
4

    
5
def env_flag(name): return name in os.environ and os.environ[name] != ''
6

    
7
def get_env_var(name, default, env_names_ref=None):
8
    if env_names_ref != None: env_names_ref.append(name)
9
    return os.getenv(name, default)
10

    
11
def get_env_vars(names, prefix=None, env_names_ref=None):
12
    vars_ = {}
13
    for name in names:
14
        env_name = name
15
        if prefix != None: env_name = prefix+'_'+env_name
16
        if env_names_ref != None: env_names_ref.append(env_name)
17
        try: vars_[name] = os.environ[env_name]
18
        except KeyError: pass
19
    return vars_
20

    
21
def env_usage(env_names, optional=False):
22
    str_ = 'env'
23
    for name in env_names:
24
        str_ += ' '
25
        if optional: str_ += '['
26
        str_ += name+'=...'
27
        if optional: str_ += ']'
28
    return str_
(7-7/14)