sh zsh插件自动激活本地virtualenv

Posted 2021-05-06

#!/usr/bin/env zsh

dotvenv:current_dir_has_virtualenv_folder(){
    echo "$(pwd)" | grep "$(dirname "$VIRTUAL_ENV")" > /dev/null 2>&1
}

dotvenv:source() {
    local DOTPYVENV_DIR=".venv"
    local DOTPYVENV_FILE=".venvdir"

    if [ -n "$VIRTUAL_ENV"  ]; then
        dotvenv:current_dir_has_virtualenv_folder || deactivate

    elif [ -f "${DOTPYVENV_FILE}" ]; then
        DOTPYVENV_DIR="$(cat "${DOTPYVENV_FILE}")"

        [ -f "${DOTPYVENV_DIR}/bin/activate" ] && {
            source "${DOTPYVENV_DIR}/bin/activate"
        }

    elif [ -f "${DOTPYVENV_DIR}/bin/activate" ]; then
        source "${DOTPYVENV_DIR}/bin/activate"
    fi
}


autoload -U add-zsh-hook
add-zsh-hook chpwd dotvenv:source