Add support for virtualenv name in prompt.

[zsh.d.git] / S60_virtualenv

diff --git a/S60_virtualenv b/S60_virtualenv
new file mode 100644 (file)
index 0000000..6c5948b
--- /dev/null
+++ b/S60_virtualenv
@@ -0,0 +1,15 @@
+#! /usr/bin/zsh
+
+# Those can be used, for instance to construct the prompt:
+__ZSH_VIRTUAL_ENV_NAME=""
+
+virtualenv_precmd() {
+  if [[ -n "${VIRTUAL_ENV}" ]]; then
+    __ZSH_VIRTUAL_ENV_NAME="$(basename ${VIRTUAL_ENV})"
+  else
+    __ZSH_VIRTUAL_ENV_NAME=""
+  fi
+}
+
+
+precmd_functions+='virtualenv_precmd'