From a31c2124e436edf496248783294776db9dec043d Mon Sep 17 00:00:00 2001
From: ykswang <ykswang@outlook.com>
Date: Wed, 9 Dec 2015 23:26:59 +0800
Subject: [PATCH] support to show python virtualenv

pygmalion.zsh-theme support to show python virtualenv name
---
 themes/pygmalion.zsh-theme | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/themes/pygmalion.zsh-theme b/themes/pygmalion.zsh-theme
index 5f5fe7f9..33f72da4 100644
--- a/themes/pygmalion.zsh-theme
+++ b/themes/pygmalion.zsh-theme
@@ -23,10 +23,19 @@ prompt_pygmalion_precmd(){
 
   local nl=""
 
-  if [[ $prompt_length -gt 40 ]]; then
-    nl=$'\n%{\r%}';
+  if [ $prompt_length -gt 40 ] 
+  then
+    nl=$'\n%{\r%}'
   fi
-  PROMPT="$base_prompt$gitinfo$nl$post_prompt"
+  
+  if [ ${#VIRTUAL_ENV} -gt 3 ]
+  then
+    venv=$' %{$fg[blue]%}[${VIRTUAL_ENV:t}]%{$reset_color%} '
+  else
+    venv=$''
+  fi
+
+  PROMPT="$base_prompt$gitinfo$nl$venv$post_prompt"
 }
 
 prompt_setup_pygmalion