Mount file rather than dir

toolforge · Oct 23, 2023 · 05a605a · 05a605a
1 parent 4116b89
commit 05a605a
Show file tree

Hide file tree

Showing 5 changed files with 30 additions and 34 deletions.
diff --git a/helm-quarry/templates/web_deployment.yaml b/helm-quarry/templates/web_deployment.yaml
@@ -24,7 +24,8 @@ spec:
             - mountPath: "/results"
               name: results
             - name: config
-              mountPath: /config/
+              mountPath: /app/quarry/config.yaml
+              subPath: config.yaml
           resources:
             requests:
               memory: {{ .Values.web.memory }}

diff --git a/helm-quarry/templates/worker_deployment.yaml b/helm-quarry/templates/worker_deployment.yaml
@@ -24,7 +24,8 @@ spec:
             - mountPath: "/results"
               name: results
             - name: config
-              mountPath: /config/
+              mountPath: /app/quarry/config.yaml
+              subPath: config.yaml
           resources:
             requests:
               memory: {{ .Values.worker.memory }}

diff --git a/quarry/web/app.py b/quarry/web/app.py
@@ -24,14 +24,16 @@
 
 def get_config():
     conf = {}
-    if os.path.isfile("/config/config.yaml"):
-        # k8s
-        config_path = "/config/config.yaml"
-    elif os.path.isfile("../config.yaml"):
-        # VM
-        config_path = "../config.yaml"
-
-    conf.update(yaml.safe_load(open(os.path.join(__dir__, config_path))))
+    conf.update(
+        yaml.safe_load(open(os.path.join(__dir__, "../default_config.yaml")))
+    )
+    try:
+        conf.update(
+            yaml.safe_load(open(os.path.join(__dir__, "../config.yaml")))
+        )
+    except IOError:
+        # Is ok if we can't load config.yaml
+        pass
 
     return conf
 

diff --git a/quarry/web/killer.py b/quarry/web/killer.py
@@ -6,17 +6,12 @@
 from connections import Connections
 
 __dir__ = os.path.dirname(__file__)
-if os.path.isfile("/config/config.yaml"):
-    # k8s
-    config_path = "/config/config.yaml"
-elif os.path.isfile("../config.yaml"):
-    # VM
-    config_path = "../config.yaml"
-else:
-    # test
-    config_path = "../default_config.yaml"
-
-config = yaml.safe_load(open(os.path.join(__dir__, config_path)))
+config = yaml.safe_load(open(os.path.join(__dir__, "../default_config.yaml")))
+try:
+    config.update(yaml.safe_load(open(os.path.join(__dir__, "../config.yaml"))))
+except IOError:
+    # is ok if we do not have config.yaml
+    pass
 
 logging.basicConfig(
     filename=config["KILLER_LOG_PATH"],

diff --git a/quarry/web/worker.py b/quarry/web/worker.py
@@ -22,19 +22,16 @@
 celery_log = get_task_logger(__name__)
 
 celery = Celery("quarry.web.worker")
-
-if os.path.isfile("/config/config.yaml"):
-    # k8s
-    config_path = "/config/config.yaml"
-elif os.path.isfile("../config.yaml"):
-    # VM
-    config_path = "../config.yaml"
-else:
-    # for pytest
-    config_path = "../default_config.yaml"
-
-celery.conf.update(yaml.safe_load(open(os.path.join(__dir__, config_path))))
-
+celery.conf.update(
+    yaml.safe_load(open(os.path.join(__dir__, "../default_config.yaml")))
+)
+try:
+    celery.conf.update(
+        yaml.safe_load(open(os.path.join(__dir__, "../config.yaml")))
+    )
+except IOError:
+    # Is ok if we can not load config.yaml
+    pass
 
 conn = None