shared/data_store/connect.py

@@ -9,6 +9,21 @@ from minio import Minio
 def connect(
 ) -> Minio:
     """Connect to MinIO server"""
+    # ensure necessary env vars available
+    necesasary_var_list = [
+        'MINIO_ENDPOINT',
+        'MINIO_ACCESS_KEY',
+        'MINIO_SECRET_KEY',
+        'MINIO_BUCKET_NAME',
+    ]
+    for env_var in necesasary_var_list:
+        # ensure env var set
+        assert (
+            env_var in os.environ
+        ), (
+            f"environment variable not set: {env_var}"
+        )
+    # prepare arguments
     minio_endpoint = os.getenv('MINIO_ENDPOINT', default=None)
     assert isinstance(minio_endpoint, str)
     minio_access_key = os.getenv('MINIO_ACCESS_KEY', default=None)