fixed argument for check_env function

misc/generate_random_prediction_test.py

@@ -8,6 +8,7 @@ from shared.data_store import connect_minio
 from shared.database import connect_mongodb
 from shared.database.classes import VisualCommunication
 from shared.utils import check_env, setup_logging
+from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 
 if __name__ == '__main__':
     # load in env file
@@ -15,7 +16,7 @@ if __name__ == '__main__':
     assert env_path.exists()
     load_dotenv(env_path)
     # ensure env vars set
-    check_env()
+    check_env(NECESSARY_ENV_VAR_LIST)
     # setup logging
     setup_logging()
     # connect to minIO

misc/get_visual_communication_test.py

@@ -7,6 +7,7 @@ from dotenv import load_dotenv
 from shared.data_store import connect_minio
 from shared.database import connect_mongodb, get_visual_communication
 from shared.utils import check_env, setup_logging
+from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 
 if __name__ == '__main__':
     # load in env file
@@ -14,7 +15,7 @@ if __name__ == '__main__':
     assert env_path.exists()
     load_dotenv(env_path)
     # ensure env vars set
-    check_env()
+    check_env(NECESSARY_ENV_VAR_LIST)
     # setup logging
     setup_logging()
     # connect to minIO

misc/image_upload_test.py

@@ -9,6 +9,7 @@ from shared.data_store import connect_minio
 from shared.database import connect_mongodb
 from shared.database.classes import VisualCommunication
 from shared.utils import check_env, setup_logging
+from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 
 if __name__ == '__main__':
     # load in env file
@@ -16,7 +17,7 @@ if __name__ == '__main__':
     assert env_path.exists()
     load_dotenv(env_path)
     # ensure env vars set
-    check_env()
+    check_env(NECESSARY_ENV_VAR_LIST)
     # setup logging
     setup_logging()
     # connect to minIO

misc/image_upload_to_server_test.py

@@ -8,6 +8,7 @@ from pymongo.errors import DuplicateKeyError
 from shared.data_store import connect_minio
 from shared.database import VisualCommunication, connect_mongodb
 from shared.utils import check_env, setup_logging
+from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 
 if __name__ == '__main__':
     # load in env file
@@ -15,7 +16,7 @@ if __name__ == '__main__':
     assert env_path.exists()
     load_dotenv(env_path)
     # ensure env vars set
-    check_env()
+    check_env(NECESSARY_ENV_VAR_LIST)
     # setup logging
     setup_logging()
     # connect to minIO

misc/prediction_upload_test.py

@@ -8,6 +8,7 @@ from shared.data_store import connect_minio
 from shared.database import connect_mongodb, upsert_prediction
 from shared.database.classes import VisualCommunication
 from shared.utils import check_env, setup_logging
+from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 
 if __name__ == '__main__':
     # load in env file
@@ -15,7 +16,7 @@ if __name__ == '__main__':
     assert env_path.exists()
     load_dotenv(env_path)
     # ensure env vars set
-    check_env()
+    check_env(NECESSARY_ENV_VAR_LIST)
     # setup logging
     setup_logging()
     # connect to minIO

other/transfer_images_mongo_minio.py

@@ -13,6 +13,7 @@ from pymongo.collection import Collection
 from shared.data_store import connect_minio, put
 from shared.database import VisualCommunication, connect_mongodb
 from shared.utils import check_env, setup_logging
+from web_ui.src.main import NECESSARY_ENV_VAR_LIST
 
 
 def list_mongo_document_ids(
@@ -76,7 +77,7 @@ if __name__ == '__main__':
     assert env_path.exists()
     load_dotenv(env_path)
     # ensure env vars set
-    check_env()
+    check_env(NECESSARY_ENV_VAR_LIST)
     # setup logging
     setup_logging()
     # connect to minIO

web_ui/src/main.py

@@ -11,7 +11,7 @@ from shared.utils import check_env, setup_logging
 from .app import init_app
 
 # ensure env vars set
-var_list = {
+NECESSARY_ENV_VAR_LIST = {
     'MONGO_HOST',
     'MONGO_DB',
     'MONGO_COLLECTION',
@@ -25,7 +25,7 @@ var_list = {
     'MINIO_BUCKET_NAME',
     'MINIO_BUCKET_NAME_MODELS',
 }
-check_env(var_list)
+check_env(NECESSARY_ENV_VAR_LIST)
 
 # setup logging stream handler
 setup_logging()