cd hook for zsh to activate a venv in the cwd

venv-activate.zsh

function create-venv() {
    name=$(basename $PWD)
    venv_path="$HOME/.venvs/$name"
    if [ -d "$venv_path" ]; then
        echo "$venv_path already exists"
        return
    fi
    python3 -m venv "$venv_path"
    source "$venv_path/bin/activate"
    pip install --upgrade pip
    if [ -e "requirements.txt" ]; then
        pip install -r requirements.txt
    fi
}

function activate-closest-venv() {
    declare -f deactivate > /dev/null && deactivate
    check="$PWD"
    while [ "$check" != "/" ]; do
        base=$(basename $check)
        [ -f "$HOME/.venvs/$base/bin/activate" ] && source "$HOME/.venvs/$base/bin/activate" && return
        check=$(realpath "$check/..")
    done
}

add-zsh-hook chpwd activate-closest-venv

cd $PWD