Using WORKON_HOME instead of envroot.

This improves compatibility with virtualenvwrapper.

2 files changed, 11 insertions, 6 deletions

diff --git a/README.md b/README.md
index 3ddcf66..fc74104 100644
--- a/README.md
+++ b/README.md
@@ -19,7 +19,9 @@ will be printed instead.
 
 ## Where venvs are created
 When `workon` is called, venvs are created or loaded from the
-`$HOME/.venvs` folder.
+`$HOME/.venvs` folder by default.
+To change this behavior, set the `WORKON_HOME` environment
+variable before sourcing `venvworkon.sh`.
 
 ## Post-load script
 If the file `$HOME/.venvs/<venv>/postload.sh` exists, it is
diff --git a/venvworkon.sh b/venvworkon.sh
index 1da01bd..adca644 100755
--- a/venvworkon.sh
+++ b/venvworkon.sh
@@ -1,20 +1,23 @@
 workon() {
-    envroot="$HOME/.venv"
+    if [ -z "$WORKON_HOME" ]
+    then
+        WORKON_HOME="$HOME/.venv"
+    fi
 
-    if [ ! -e "$envroot" ]
+    if [ ! -e "$WORKON_HOME" ]
     then
-        mkdir -p "$envroot"
+        mkdir -p "$WORKON_HOME"
     fi
 
     if [ "$#" -lt "1" ]
     then
-        ls "$envroot"
+        ls "$WORKON_HOME"
         return
     fi
 
     envname=$1
 
-    envdir="$envroot/$envname"
+    envdir="$WORKON_HOME/$envname"
 
     if [ ! -e "$envdir" ]
     then