diff --git a/plugins/virtualenv/virtualenv.plugin.zsh b/plugins/virtualenv/virtualenv.plugin.zsh new file mode 100644 index 000000000..f73e7bcc4 --- /dev/null +++ b/plugins/virtualenv/virtualenv.plugin.zsh @@ -0,0 +1,17 @@ +export VIRTUAL_ENV_DISABLE_PROMPT=1 + +ZSH_THEME_VIRTUAL_ENV_PROMPT_PREFIX="(" +ZSH_THEME_VIRTUAL_ENV_PROMPT_SUFFIX=")" + +function virtualenv_prompt_info() { + if [ -n "$VIRTUAL_ENV" ]; then + if [ -f "$VIRTUAL_ENV/__name__" ]; then + local name=`cat $VIRTUAL_ENV/__name__` + elif [ `basename $VIRTUAL_ENV` = "__" ]; then + local name=$(basename $(dirname $VIRTUAL_ENV)) + else + local name=$(basename $VIRTUAL_ENV) + fi + echo "$ZSH_THEME_VIRTUAL_ENV_PROMPT_PREFIX$name$ZSH_THEME_VIRTUAL_ENV_PROMPT_SUFFIX" + fi +}