mirror of
https://github.com/ohmyzsh/ohmyzsh.git
synced 2024-12-27 15:30:15 +08:00
28 lines
821 B
Bash
28 lines
821 B
Bash
# Automatic poetry environment activation/deactivation
|
|
_togglePoetryShell() {
|
|
# deactivate environment if pyproject.toml doesn't exist and not in a subdir
|
|
if [[ ! -f "$PWD/pyproject.toml" ]] ; then
|
|
if [[ "$poetry_active" == 1 ]]; then
|
|
if [[ "$PWD" != "$poetry_dir"* ]]; then
|
|
export poetry_active=0
|
|
deactivate
|
|
return
|
|
fi
|
|
fi
|
|
fi
|
|
|
|
# activate the environment if pyproject.toml exists
|
|
if [[ "$poetry_active" != 1 ]]; then
|
|
if [[ -f "$PWD/pyproject.toml" ]]; then
|
|
if grep -q 'tool.poetry' "$PWD/pyproject.toml" && venv_dir=$(poetry env info --path); then
|
|
export poetry_active=1
|
|
export poetry_dir="$PWD"
|
|
source "${venv_dir}/bin/activate"
|
|
fi
|
|
fi
|
|
fi
|
|
}
|
|
autoload -U add-zsh-hook
|
|
add-zsh-hook chpwd _togglePoetryShell
|
|
_togglePoetryShell
|