respect PYSTORE_PATH if set

PyQuantSharp · Sep 27, 2020 · e0af12a · e0af12a
1 parent 79fc038
commit e0af12a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/README.rst b/README.rst
@@ -83,7 +83,8 @@ Using PyStore
     import pystore
     import quandl
 
-    # Set storage path (optional, default is `~/pystore`)
+    # Set storage path (optional)
+    # Defaults to `~/pystore` or `PYSTORE_PATH` environment variable (if set)
     pystore.set_path("~/pystore")
 
     # List stores

diff --git a/pystore/config.py b/pystore/config.py
@@ -18,9 +18,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from os.path import expanduser as _expanduser
+import os as os
+from .utils import Path
 
-DEFAULT_PATH = _expanduser("~/pystore")
+DEFAULT_PATH = os.environ.get("PYSTORE_PATH", Path.home() / "pystore")
 DEFAULT_PARTITION_SIZE = 99e+6  # ~99MB
 PARTITION_SIZE = 99e+6  # ~99MB