export VIRTUAL_ENV_DISABLE_PROMPT=1

function prompt {

    local BLACK="\[\033[0;30m\]"
    local BLACKBOLD="\[\033[1;30m\]"
    local RED="\[\033[0;31m\]"
    local REDBOLD="\[\033[1;31m\]"
    local GREEN="\[\033[0;32m\]"
    local GREENBOLD="\[\033[1;32m\]"
    local YELLOW="\[\033[0;33m\]"
    local YELLOWBOLD="\[\033[1;33m\]"
    local BLUE="\[\033[0;34m\]"
    local BLUEBOLD="\[\033[1;34m\]"
    local PURPLE="\[\033[0;35m\]"
    local PURPLEBOLD="\[\033[1;35m\]"
    local CYAN="\[\033[0;36m\]"
    local CYANBOLD="\[\033[1;36m\]"
    local WHITE="\[\033[0;37m\]"
    local WHITEBOLD="\[\033[1;37m\]"
    local ENDCOLOR="\[\e[0m\]"

    function parse_git_branch {
        esc=$(printf '\033')
        git branch --no-color 2> /dev/null | sed -e '/^[^*]/d' -e 's/* \(.*\)/'${esc}'[0;30m on'${esc}'[1;35m \xe2\x8e\x87 \1/'
    }

    function set_venv_name {
        if [[ -n "$VIRTUAL_ENV" ]]; then
            venv="${VIRTUAL_ENV##*/}"
        else
            venv='none'
        fi
        [[ -n "$venv" ]] && echo "$venv"
    }

    local VENV="$CYAN\[\]venv\[\]$BLACK\[\]:\[\]$CYANBOLD\[\]\$(set_venv_name) "

    # export PROMPT_COMMAND="echo"
    export PS1="$BLACKBOLD\342\224\214\342\224\200[$VENV$BLACK\[as\] $REDBOLD\u$BLACK at$BLUEBOLD \H$BLACK in$GREENBOLD \w\$(parse_git_branch)$BLACKBOLD]\n\342\224\224\342\224\200$(printf '\xe2\xac\x9c') $ENDCOLOR"
}
prompt