From f8689401def3f114ebb113acdac9d422f0aa914e Mon Sep 17 00:00:00 2001
From: Sorin Ionescu <sorin@krustyroo.local>
Date: Mon, 3 Sep 2012 10:53:19 -0400
Subject: [PATCH] Set the virtualenvwrapper $WORKON_HOME variable

---
 modules/python/README.md | 10 +++++-----
 modules/python/init.zsh  |  6 +++++-
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/modules/python/README.md b/modules/python/README.md
index 2363f87..5747147 100644
--- a/modules/python/README.md
+++ b/modules/python/README.md
@@ -6,7 +6,7 @@ Enables local Python and local Python package installation.
 Local Python Installation
 -------------------------
 
-[pythonz][6] builds and installs multiple Python versions locally in the home
+[pythonz][4] builds and installs multiple Python versions locally in the home
 directory.
 
 This module prepends the pythonz directory to the path variable to enable the
@@ -41,8 +41,9 @@ convenient shell functions to create, switch, and manage them.
 
 ### Usage
 
-Install virtualenvwrapper and set [`$WORKON_HOME`][4] to the path where virtual
-environments will be stored.
+Install virtualenvwrapper.
+
+Virtual environments are stored in *~/.virtualenvs*.
 
 ### Theming
 
@@ -87,7 +88,6 @@ Authors
 [1]: http://www.python.org/dev/peps/pep-0370/
 [2]: http://www.doughellmann.com/projects/virtualenvwrapper/
 [3]: http://pypi.python.org/pypi/virtualenv
-[4]: http://www.doughellmann.com/docs/virtualenvwrapper/#introduction
+[4]: http://saghul.github.com/pythonz/
 [5]: https://github.com/sorin-ionescu/oh-my-zsh/issues
-[6]: http://saghul.github.com/pythonz/
 
diff --git a/modules/python/init.zsh b/modules/python/init.zsh
index 13f8dee..927dc9d 100644
--- a/modules/python/init.zsh
+++ b/modules/python/init.zsh
@@ -23,9 +23,13 @@ else
 fi
 
 # Load virtualenvwrapper into the shell session.
-if [[ -n "$WORKON_HOME" ]] && (( $+commands[virtualenvwrapper_lazy.sh] )); then
+if (( $+commands[virtualenvwrapper_lazy.sh] )); then
+  # Set the directory where virtual environments are stored.
+  export WORKON_HOME=$HOME/.virtualenvs
+
   # Disable the virtualenv prompt.
   VIRTUAL_ENV_DISABLE_PROMPT=1
+
   source "$commands[virtualenvwrapper_lazy.sh]"
 fi