func! GetVirtualEnv()
    let virtualenv = system("basename -- \"$(echo -n $VIRTUAL_ENV)\" | tr -d '\n'")
    if !empty(virtualenv)
        return "virtualenv: ".virtualenv
    endif
    return ""
endfunc

set statusline+=%{GetVirtualEnv()}