Use CORTEX_CLI_CONFIG_DIR in python client when making temporary files (#1798)

Author: vishalbollu

pkg/cortex/client/cortex/client.py

@@ -33,6 +33,13 @@
 EXPECTED_PYTHON_VERSION = "3.6.9"
 
 
+def cli_config_dir() -> Path:
+    cli_config_dir = os.environ.get("CORTEX_CLI_CONFIG_DIR", "")
+    if cli_config_dir == "":
+        return Path.home() / ".cortex"
+    return Path(cli_config_dir).expanduser().resolve()
+
+
 class Client:
     def __init__(self, env: dict):
         """
@@ -91,7 +98,7 @@ def create_api(
         if api_spec.get("name") is None:
             raise ValueError("`api_spec` must have the `name` key set")
 
-        project_dir = Path.home() / ".cortex" / "deployments" / api_spec["name"]
+        project_dir = cli_config_dir() / "deployments" / api_spec["name"]
 
         if project_dir.exists():
             shutil.rmtree(str(project_dir))
@@ -296,9 +303,7 @@ def patch(self, api_spec: dict, force: bool = False) -> dict:
             force: Override an already in-progress API update.
         """
 
-        cortex_yaml_file = (
-            Path.home() / ".cortex" / "deployments" / f"cortex-{str(uuid.uuid4())}.yaml"
-        )
+        cortex_yaml_file = cli_config_dir() / "deployments" / f"cortex-{str(uuid.uuid4())}.yaml"
         with util.open_temporarily(cortex_yaml_file, "w") as f:
             yaml.dump([api_spec], f)
             args = ["patch", cortex_yaml_file, "--env", self.env_name, "-o", "json"]