| import os | |
| 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 | |