diff --git a/plugins/virtualenv/virtualenv.plugin.zsh b/plugins/virtualenv/virtualenv.plugin.zsh index e8458389f..934f54aa3 100644 --- a/plugins/virtualenv/virtualenv.plugin.zsh +++ b/plugins/virtualenv/virtualenv.plugin.zsh @@ -1,8 +1,20 @@ -function virtualenv_prompt_info(){ +function virtualenv_check() { local virtualenv_path="$VIRTUAL_ENV" if [[ -n $virtualenv_path ]]; then local virtualenv_name=`basename $virtualenv_path` - printf "%s[%s] " "%{${fg[yellow]}%}" $virtualenv_name + printf "%s" $virtualenv_name + fi +} + +function virtualenv_prompt_info() { + local prefix="$ZSH_THEME_VIRTUALENV_PROMPT_PREFIX" + local suffix="$ZSH_THEME_VIRTUALENV_PROMPT_SUFFIX" + if [[ -n $(virtualenv_check) ]]; then + if [[ -n $prefix ]]; then + printf "%s%s%s" $prefix $(virtualenv_check) $suffix + else + printf "%s[%s] " "%{${fg[yellow]}%}" $(virtualenv_check) + fi fi } diff --git a/themes/blinks.zsh-theme b/themes/blinks.zsh-theme index ddb32f7c3..1bf69bf0f 100644 --- a/themes/blinks.zsh-theme +++ b/themes/blinks.zsh-theme @@ -23,8 +23,18 @@ ZSH_THEME_GIT_PROMPT_SUFFIX="%{%f%k%b%K{${bkg}}%B%F{green}%}]" ZSH_THEME_GIT_PROMPT_DIRTY=" %{%F{red}%}*%{%f%k%b%}" ZSH_THEME_GIT_PROMPT_CLEAN="" +ZSH_THEME_VIRTUALENV_PROMPT_PREFIX=" (%{%B%F{blue}%}" +ZSH_THEME_VIRTUALENV_PROMPT_SUFFIX="%{%f%k%b%K{${bkg}}%B%F{green}%})" + +function _virtualenv() { + # check if the virtualenv plugin is activated + if [[ ${plugins[(r)virtualenv]} == virtualenv ]] then; + echo "%s" $(virtualenv_prompt_info) + fi +} + PROMPT='%{%f%k%b%} -%{%K{${bkg}}%B%F{green}%}%n%{%B%F{blue}%}@%{%B%F{cyan}%}%m%{%B%F{green}%} %{%b%F{yellow}%K{${bkg}}%}%~%{%B%F{green}%}$(git_prompt_info)%E%{%f%k%b%} +%{%K{${bkg}}%B%F{green}%}%n%{%B%F{blue}%}@%{%B%F{cyan}%}%m%{%B%F{green}%} %{%b%F{yellow}%K{${bkg}}%}%~%{%B%F{green}%}$(_virtualenv)$(git_prompt_info)%E%{%f%k%b%} %{%K{${bkg}}%}$(_prompt_char)%{%K{${bkg}}%} %#%{%f%k%b%} ' RPROMPT='!%{%B%F{cyan}%}%!%{%f%k%b%}'