moved hardcoded values into main script

shared/utils/check_env.py

@@ -1,26 +1,14 @@
 """Definition of check_env function."""
 
-from __future__ import annotations
-
 import os
 
 
-def check_env() -> None:
+def check_env(
+    var_list: set[str],
+) -> None:
     """Check necessary environment variables are set."""
-    necesasary_var_list = {
+    assert isinstance(var_list, set)
-        'MONGO_HOST',
+    assert all(isinstance(elem, str) for elem in var_list)
-        'MONGO_DB',
+    # check that env vars are set
-        'MONGO_COLLECTION',
+    for env_var in var_list:
-        'MONGO_USER',
-        'MONGO_PASSWORD',
-        'DASH_AUTH_USERNAME',
-        'DASH_AUTH_PASSWORD',
-        'MINIO_ENDPOINT',
-        'MINIO_ACCESS_KEY',
-        'MINIO_SECRET_KEY',
-        'MINIO_BUCKET_NAME',
-        'MINIO_BUCKET_NAME_MODELS',
-    }
-    for env_var in necesasary_var_list:
-        # ensure env var set
         assert env_var in os.environ, f"environment variable not set: {env_var}"

web_ui/src/main.py

@@ -11,7 +11,21 @@ from shared.utils import check_env, setup_logging
 from .app import init_app
 
 # ensure env vars set
-check_env()
+var_list = {
+    'MONGO_HOST',
+    'MONGO_DB',
+    'MONGO_COLLECTION',
+    'MONGO_USER',
+    'MONGO_PASSWORD',
+    'DASH_AUTH_USERNAME',
+    'DASH_AUTH_PASSWORD',
+    'MINIO_ENDPOINT',
+    'MINIO_ACCESS_KEY',
+    'MINIO_SECRET_KEY',
+    'MINIO_BUCKET_NAME',
+    'MINIO_BUCKET_NAME_MODELS',
+}
+check_env(var_list)
 
 # setup logging stream handler
 setup_logging()