This commit is contained in:
Matt Sweeney 2013-05-01 03:18:17 -07:00
commit 191d737e37
2 changed files with 25 additions and 3 deletions

View file

@ -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
}

View file

@ -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%}'