Configurable storage

cli/medperf/config.py

@@ -41,7 +41,8 @@
 keyring_access_token_service_name = "medperf_access_token"
 keyring_refresh_token_service_name = "medperf_refresh_token"
 
-storage = str(Path.home().resolve() / ".medperf")
+base_storage = str(Path.home().resolve() / ".medperf")
+storage = base_storage
 logs_storage = "logs"
 tmp_storage = "tmp"
 data_storage = "data"
@@ -93,6 +94,7 @@
 mlcube_inspect_timeout = None
 
 configurable_parameters = [
+    "storage",
     "server",
     "certificate",
     "auth_class",

cli/medperf/decorators.py

@@ -1,3 +1,4 @@
+import os
 import sys
 import typer
 import logging
@@ -67,7 +68,13 @@ def configurable(func: Callable) -> Callable:
     #       by changing configurable_parameters
     @merge_args(func)
     def wrapper(
+        ctx: typer.Context,
         *args,
+        storage: str = typer.Option(
+            config.storage,
+            "--storage",
+            help="Location to store medperf-related artifacts",
+        ),
         server: str = typer.Option(
             config.server, "--server", help="URL of a hosted MedPerf API instance"
         ),
@@ -153,6 +160,8 @@ def wrapper(
         ),
         **kwargs,
     ):
-        return func(*args, **kwargs)
+        # Transform the context as necessary
+        ctx.params["storage"] = os.path.expanduser(storage)
+        return func(ctx, *args, **kwargs)
 
     return wrapper

cli/medperf/utils.py

@@ -91,7 +91,7 @@ def storage_path(subpath: str):
 
 def base_storage_path(subpath: str):
     """Helper function that converts a path to base storage-related path"""
-    return os.path.join(config.storage, subpath)
+    return os.path.join(config.base_storage, subpath)
 
 
 def get_file_hash(path: str) -> str: