From 3d73e7498f06017b06af12263b324e053d8ba7dc Mon Sep 17 00:00:00 2001 From: Litchi Pi Date: Tue, 13 Feb 2024 20:30:07 +0100 Subject: [PATCH] feat: allow overriding of some absolute paths using environment variables (#3102) * Allow overriding of alembic config file path using environment variable Signed-off-by: Litchi Pi * Allow overriding of MODEL_PATH using environment variable Signed-off-by: Litchi Pi --------- Signed-off-by: Litchi Pi --- mealie/db/init_db.py | 8 +++++++- mealie/services/parser_services/crfpp/processor.py | 3 ++- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/mealie/db/init_db.py b/mealie/db/init_db.py index 1fcefe1c21bf..6f94ee962023 100644 --- a/mealie/db/init_db.py +++ b/mealie/db/init_db.py @@ -1,3 +1,4 @@ +import os from collections.abc import Callable from pathlib import Path from time import sleep @@ -87,7 +88,12 @@ def main(): if max_retry == 0: raise ConnectionError("Database connection failed - exiting application.") - alembic_cfg = Config(str(PROJECT_DIR / "alembic.ini")) + alembic_cfg_path = os.getenv("ALEMBIC_CONFIG_FILE", default=str(PROJECT_DIR / "alembic.ini")) + + if not os.path.isfile(alembic_cfg_path): + raise Exception("Provided alembic config path doesn't exist") + + alembic_cfg = Config(alembic_cfg_path) if db_is_at_head(alembic_cfg): logger.debug("Migration not needed.") else: diff --git a/mealie/services/parser_services/crfpp/processor.py b/mealie/services/parser_services/crfpp/processor.py index 5c114732f31a..8ca936885386 100644 --- a/mealie/services/parser_services/crfpp/processor.py +++ b/mealie/services/parser_services/crfpp/processor.py @@ -1,3 +1,4 @@ +import os import subprocess import tempfile from fractions import Fraction @@ -13,7 +14,7 @@ from . import utils from .pre_processor import pre_process_string CWD = Path(__file__).parent -MODEL_PATH = CWD / "model.crfmodel" +MODEL_PATH = os.getenv("CRF_MODEL_PATH", default=CWD / "model.crfmodel") class CRFConfidence(BaseModel):