moved files
This commit is contained in:
@@ -1,3 +0,0 @@
|
|||||||
from __future__ import annotations
|
|
||||||
|
|
||||||
from .check_env import check_env
|
|
||||||
@@ -1,25 +0,0 @@
|
|||||||
"""Definition of check_env function."""
|
|
||||||
from __future__ import annotations
|
|
||||||
|
|
||||||
import os
|
|
||||||
|
|
||||||
|
|
||||||
def check_env() -> None:
|
|
||||||
"""Check necessary environment variables are set."""
|
|
||||||
necesasary_var_list = {
|
|
||||||
'MONGO_HOST',
|
|
||||||
'MONGO_DB',
|
|
||||||
'MONGO_COLLECTION',
|
|
||||||
'MONGO_USER',
|
|
||||||
'MONGO_PASSWORD',
|
|
||||||
'MINIO_ENDPOINT',
|
|
||||||
'MINIO_ACCESS_KEY',
|
|
||||||
'MINIO_SECRET_KEY',
|
|
||||||
}
|
|
||||||
for env_var in necesasary_var_list:
|
|
||||||
# ensure env var set
|
|
||||||
assert (
|
|
||||||
env_var in os.environ
|
|
||||||
), (
|
|
||||||
f"environment variable not set: {env_var}"
|
|
||||||
)
|
|
||||||
@@ -1,4 +1,5 @@
|
|||||||
"""Definition of check_env function."""
|
"""Definition of check_env function."""
|
||||||
|
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
import os
|
import os
|
||||||
@@ -18,11 +19,8 @@ def check_env() -> None:
|
|||||||
'MINIO_ACCESS_KEY',
|
'MINIO_ACCESS_KEY',
|
||||||
'MINIO_SECRET_KEY',
|
'MINIO_SECRET_KEY',
|
||||||
'MINIO_BUCKET_NAME',
|
'MINIO_BUCKET_NAME',
|
||||||
|
'MINIO_BUCKET_NAME_MODELS',
|
||||||
}
|
}
|
||||||
for env_var in necesasary_var_list:
|
for env_var in necesasary_var_list:
|
||||||
# ensure env var set
|
# ensure env var set
|
||||||
assert (
|
assert env_var in os.environ, f"environment variable not set: {env_var}"
|
||||||
env_var in os.environ
|
|
||||||
), (
|
|
||||||
f"environment variable not set: {env_var}"
|
|
||||||
)
|
|
||||||
|
|||||||
Reference in New Issue
Block a user