mirror of
https://github.com/ohmyzsh/ohmyzsh.git
synced 2026-01-23 02:35:38 +01:00
3a6f538109
Since version 2.0.0, poetry stopped requiring '[tool.poetry]' section in pyproject.toml and new environments are created without it, but the zsh plugin still required it. With this change, the plugin will check for `pyproject.toml` and `poetry.lock` files to reliably check if it's a poetry project. Tested with both poetry versions 1.8.5, 2.0.1 and 2.1.2.
27 lines
1,023 B
Bash
27 lines
1,023 B
Bash
_togglePoetryShell() {
|
|
# Determine if currently in a Poetry-managed directory
|
|
local in_poetry_dir=0
|
|
if [[ -f "$PWD/pyproject.toml" && -f "$PWD/poetry.lock" ]]; then
|
|
in_poetry_dir=1
|
|
fi
|
|
|
|
# Deactivate the current environment if moving out of a Poetry directory or into a different Poetry directory
|
|
if [[ $poetry_active -eq 1 ]] && { [[ $in_poetry_dir -eq 0 ]] || [[ "$PWD" != "$poetry_dir"* ]]; }; then
|
|
export poetry_active=0
|
|
unset poetry_dir
|
|
(( $+functions[deactivate] )) && deactivate
|
|
fi
|
|
|
|
# Activate the environment if in a Poetry directory and no environment is currently active
|
|
if [[ $in_poetry_dir -eq 1 ]] && [[ $poetry_active -ne 1 ]]; then
|
|
venv_dir=$(poetry env info --path 2>/dev/null)
|
|
if [[ -n "$venv_dir" ]]; then
|
|
export poetry_active=1
|
|
export poetry_dir="$PWD"
|
|
source "${venv_dir}/bin/activate"
|
|
fi
|
|
fi
|
|
}
|
|
autoload -U add-zsh-hook
|
|
add-zsh-hook chpwd _togglePoetryShell
|
|
_togglePoetryShell # Initial call to check the current directory at shell startup
|