Merge pull request #11 from MichaelAquilina/python_virtualenv

Add python virtualenv segment
agnoster · Aug 9, 2016 · f3ff367 · f3ff367
2 parents 43cb371 + e2371ec
commit f3ff367
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/agnoster.zsh-theme b/agnoster.zsh-theme
@@ -120,12 +120,22 @@ prompt_status() {
   [[ -n "$symbols" ]] && prompt_segment $PRIMARY_FG default " $symbols "
 }
 
+# Display current virtual environment
+prompt_virtualenv() {
+  if [[ -n $VIRTUAL_ENV ]]; then
+    color=cyan
+    prompt_segment $color $PRIMARY_FG
+    print -Pn " $(basename $VIRTUAL_ENV) "
+  fi
+}
+
 ## Main prompt
 prompt_agnoster_main() {
   RETVAL=$?
   CURRENT_BG='NONE'
   prompt_status
   prompt_context
+  prompt_virtualenv
   prompt_dir
   prompt_git
   prompt_end