Add cache_dir checking

kthcloud · Oct 6, 2023 · 5cb80b1 · 5cb80b1
1 parent 53c0776
commit 5cb80b1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "secd"
-version = "0.0.5"
+version = "0.0.6"
 authors = [
   { name="Emil Karlsson", email="emilk2@kth.se" },
   { name="Pierre Le Fevre", email="pierrelf@kth.se" }

diff --git a/src/secd/secd.py b/src/secd/secd.py
@@ -2,8 +2,17 @@
 import os
 import hashlib
 import inspect
+import yaml
 
-CACHE_DIR = "cache"
+def get_cache_dir():
+    with open("secd.yml", "r") as f:
+        run_meta = yaml.load(f)
+        if "cache_dir" in run_meta:
+            return run_meta["cache_dir"]
+
+    return "cache"
+
+CACHE_DIR = get_cache_dir()
 
 def generate_cache_key(func, args, kwargs):
     # Create a unique cache key based on the function name, args, kwargs, and source code