Spaces:
Running
on
Zero
Running
on
Zero
File size: 813 Bytes
01e655b 627d3d7 01e655b 627d3d7 |
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 39 |
import os
from modules import config
def get_env_val(key, val_type):
env_val = os.getenv(key.upper())
if env_val is not None and env_val != "":
if val_type == bool:
return env_val.lower() in ("true", "1", "t")
return val_type(env_val)
if env_val == "":
return None
return env_val
def get_env_or_arg(args, arg_name, default, arg_type):
arg_val = getattr(args, arg_name)
env_val = get_env_val(arg_name, arg_type)
if arg_type == bool and env_val is not None:
return env_val
if arg_val is not None:
return arg_val
elif env_val is not None:
return env_val
return default
def get_and_update_env(*args):
val = get_env_or_arg(*args)
key = args[1]
config.runtime_env_vars[key] = val
return val
|