From cdc527aece013fff3fcbd6ffed33fc6565c79186 Mon Sep 17 00:00:00 2001 From: Anastasios Zouzias Date: Mon, 31 Mar 2025 10:15:41 +0000 Subject: [PATCH] (config) allow user to customize output location --- demo.ipynb | 10 ++++++++++ ssgetpy/config.py | 11 ++++++----- 2 files changed, 16 insertions(+), 5 deletions(-) diff --git a/demo.ipynb b/demo.ipynb index 471c341..804ab88 100644 --- a/demo.ipynb +++ b/demo.ipynb @@ -56,6 +56,16 @@ "import ssgetpy" ] }, + { + "cell_type": "markdown", + "metadata": {}, + "source": [ + "# Custom output directory via `SSGETPY_OUTPUT_LOCATION` environmental variable\n", + "!export SSGETPY_OUTPUT_LOCATION = /my/custom/folder/\n", + "\n", + "# The above environmental variable will overwrite the default output location, see `config.py`" + ] + }, { "cell_type": "markdown", "metadata": {}, diff --git a/ssgetpy/config.py b/ssgetpy/config.py index c363bd1..18cd83c 100644 --- a/ssgetpy/config.py +++ b/ssgetpy/config.py @@ -4,16 +4,17 @@ logger = logging.getLogger(__name__) -SS_DIR = None +SS_DIR = os.environ.get(["SSGETPY_OUTPUT_LOCATION"], None) SS_DB = "index.db" SS_TABLE = "MATRICES" SS_ROOT_URL = "https://sparse.tamu.edu" SS_INDEX_URL = "/".join((SS_ROOT_URL, "files", "ssstats.csv")) -if sys.platform == "win32": - SS_DIR = os.path.join(os.environ["APPDATA"], "ssgetpy") -else: - SS_DIR = os.path.join(os.environ["HOME"], ".ssgetpy") +if SS_DIR is None: + if sys.platform == "win32": + SS_DIR = os.path.join(os.environ["APPDATA"], "ssgetpy") + else: + SS_DIR = os.path.join(os.environ["HOME"], ".ssgetpy") SS_DB = os.path.join(SS_DIR, SS_DB)