Print virtualenv directory if active

diff --git a/states/zsh/files/zshrc b/states/zsh/files/zshrc
index 77dec342c97500439993cd3b700629962bed507b..f02b7e36e59b5ccb84bd3769ec8ee20dac2d75da 100644 (file)
--- a/states/zsh/files/zshrc
+++ b/states/zsh/files/zshrc
@@ -32,16 +32,25 @@ bindkey -v
 NEWLINE=$'\n'
 setopt prompt_subst
 
+
 p="%(?;%F{green};%F{red})[%h]%f "
 p+="%F{red}%B%n%b%f@%F{red}%B%m%b%f"
 p+=":%F{blue}%~%f"
 p+="${NEWLINE}%(!.#.$) "
 
+virtual_env() {
+       if [ -n "${VIRTUAL_ENV+1}" ]; then
+               home_rel_path=$(echo ${VIRTUAL_ENV} | sed "s;^${HOME};~;")
+               echo -n "(${home_rel_path}) "
+       fi
+}
+
 source ~/.git-prompt.sh
 precmd () {
-       __git_ps1 "" "${p}" "${NEWLINE}[%s]${NEWLINE}"
+       __git_ps1 "" "$(virtual_env)${p}" "${NEWLINE}[%s]${NEWLINE}"
 }
 
+
 if ! pgrep -x -u "${USER}" gpg-agent >/dev/null 2>&1; then
   gpg-connect-agent /bye >/dev/null 2>&1
 fi