diff --git a/lib/virtualenv.zsh b/lib/virtualenv.zsh new file mode 100644 index 000000000..87049a532 --- /dev/null +++ b/lib/virtualenv.zsh @@ -0,0 +1,5 @@ +# get the name of the virtualenv we are on +function virtualenv_prompt_info() { + active_virtualenv=$(basename $VIRTUAL_ENV 2> /dev/null) || return + [[ -n active_virtualenv ]] && echo "$ZSH_THEME_VIRTUALENV_PROMPT_PREFIX$active_virtualenv$ZSH_THEME_VIRTUALENV_PROMPT_SUFFIX" +} diff --git a/plugins/virtualenv/virtualenv.plugin.zsh b/plugins/virtualenv/virtualenv.plugin.zsh new file mode 100644 index 000000000..5fc197ddc --- /dev/null +++ b/plugins/virtualenv/virtualenv.plugin.zsh @@ -0,0 +1,3 @@ +# Prevent virtualenv from prepending (virtualenv name) to your prompt +# Use virtualenv_prompt_info to add the env to your prompt. +export VIRTUAL_ENV_DISABLE_PROMPT=1