From a6597abf7b1633c7c72790c805a9690f7d80f866 Mon Sep 17 00:00:00 2001 From: Fredrik Wrede Date: Tue, 27 Feb 2024 15:08:26 +0100 Subject: [PATCH] boolean envs --- fedn/fedn/common/config.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/fedn/fedn/common/config.py b/fedn/fedn/common/config.py index e53d7dea7..7b1643692 100644 --- a/fedn/fedn/common/config.py +++ b/fedn/fedn/common/config.py @@ -18,6 +18,21 @@ def get_env(key, default=None): """ return os.environ.get(key, default) +def get_boolean_env(key, default=False): + """ Get environment variable as boolean. + + :param key: The environment variable key. + :type key: str + :param default: The default value if the environment variable is not set (optional). + :type default: bool + :return: The environment variable value. + :rtype: bool + """ + value = os.environ.get(key, default) + if isinstance(value, str): + return value.lower() in ['true', '1'] + return value + def get_environment_config(): """ Get the configuration from environment variables. @@ -85,7 +100,7 @@ def get_modelstorage_config(file=None): "storage_secret_key": get_env("MODELSTORAGE_SECRET_KEY"), "storage_bucket": get_env("MODELSTORAGE_BUCKET", "fedn-models"), "context_bucket": get_env("MODELSTORAGE_CONTEXT_BUCKET", "fedn-context"), - "storage_secure_mode": bool(get_env("MODELSTORAGE_SECURE_MODE", False)), + "storage_secure_mode": get_boolean_env("MODELSTORAGE_SECURE_MODE", False), } } else: