Skip to content

Commit 8cbc6ed

Browse files
author
Eric Meadows
committed
enable more settings
1 parent e2a2dc0 commit 8cbc6ed

File tree

1 file changed

+25
-11
lines changed

1 file changed

+25
-11
lines changed

python/seldon_core/wrapper.py

Lines changed: 25 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -176,17 +176,31 @@ def _set_flask_app_configs(app):
176176
:param app:
177177
:return:
178178
"""
179-
env_to_config_map = {
180-
"FLASK_JSONIFY_PRETTYPRINT_REGULAR": "JSONIFY_PRETTYPRINT_REGULAR",
181-
"FLASK_JSON_SORT_KEYS": "JSON_SORT_KEYS",
182-
"FLASK_PROPAGATE_EXCEPTIONS": "PROPAGATE_EXCEPTIONS",
183-
}
184-
185-
for env_var, config_name in env_to_config_map.items():
186-
if os.environ.get(env_var):
187-
# Environment variables come as strings, convert them to boolean
188-
bool_env_value = os.environ.get(env_var).lower() == "true"
189-
app.config[config_name] = bool_env_value
179+
FLASK_CONFIG_IDENTIFIER = "FLASK_"
180+
FLASK_CONFIGS_BOOL = [
181+
"DEBUG",
182+
"EXPLAIN_TEMPLATE_LOADING",
183+
"TESTING",
184+
"PROPAGATE_EXCEPTIONS",
185+
"PRESERVE_CONTEXT_ON_EXCEPTION",
186+
"SESSION_COOKIE_HTTPONLY",
187+
"SESSION_COOKIE_SECURE",
188+
"SESSION_REFRESH_EACH_REQUEST",
189+
"TEMPLATES_AUTO_RELOAD",
190+
"TRAP_HTTP_EXCEPTIONS",
191+
"TRAP_BAD_REQUEST_ERRORS",
192+
"USE_X_SENDFILE",
193+
]
194+
195+
for env_var, value in os.environ.items():
196+
if not env_var.startswith(FLASK_CONFIG_IDENTIFIER):
197+
continue
198+
flask_config = env_var.replace(FLASK_CONFIG_IDENTIFIER, "")
199+
if flask_config not in FLASK_CONFIGS_BOOL:
200+
continue
201+
# Environment variables come as strings, convert them to boolean
202+
app.config[flask_config] = value == "true"
203+
logger.info(f"App Config: {app.config}")
190204

191205

192206
# ----------------------------

0 commit comments

Comments
 (0)