(key, default=False)
| 106 | |
| 107 | |
| 108 | def parse_flag_from_env(key, default=False): |
| 109 | try: |
| 110 | value = os.environ[key] |
| 111 | except KeyError: |
| 112 | # KEY isn't set, default to `default`. |
| 113 | _value = default |
| 114 | else: |
| 115 | # KEY is set, convert it to True or False. |
| 116 | try: |
| 117 | _value = strtobool(value) |
| 118 | except ValueError: |
| 119 | # More values are supported, but let's keep the message simple. |
| 120 | raise ValueError(f"If set, {key} must be yes or no.") |
| 121 | return _value |
| 122 | |
| 123 | |
| 124 | _run_slow_tests = parse_flag_from_env("RUN_SLOW", default=False) |