From 1cc32be5f632ff933b3e3f1c7bd9e612a274c095 Mon Sep 17 00:00:00 2001
From: ajatkj <ajatkj@yahoo.co.in>
Date: Wed, 20 Sep 2023 20:21:27 +0530
Subject: feat(poetry-env): create plugin (#11069)
---
plugins/poetry-env/README.md | 10 ++++++++++
plugins/poetry-env/poetry-env.plugin.zsh | 27 +++++++++++++++++++++++++++
2 files changed, 37 insertions(+)
create mode 100644 plugins/poetry-env/README.md
create mode 100644 plugins/poetry-env/poetry-env.plugin.zsh
diff --git a/plugins/poetry-env/README.md b/plugins/poetry-env/README.md
new file mode 100644
index 000000000..a7d16563e
--- /dev/null
+++ b/plugins/poetry-env/README.md
@@ -0,0 +1,10 @@
+# Poetry Environment Plugin
+
+This plugin automatically changes poetry environment when you cd into or out of the project directory.
+Note: Script looks for pyproject.toml file to determine poetry if its a poetry environment
+
+To use it, add `poetry-env` to the plugins array in your zshrc file:
+
+```zsh
+plugins=(... poetry-env)
+```
diff --git a/plugins/poetry-env/poetry-env.plugin.zsh b/plugins/poetry-env/poetry-env.plugin.zsh
new file mode 100644
index 000000000..86e5fad4e
--- /dev/null
+++ b/plugins/poetry-env/poetry-env.plugin.zsh
@@ -0,0 +1,27 @@
+# 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"; then
+ export poetry_active=1
+ export poetry_dir="$PWD"
+ source "$(poetry env info --path)/bin/activate"
+ fi
+ fi
+ fi
+}
+autoload -U add-zsh-hook
+add-zsh-hook chpwd _togglePoetryShell
+_togglePoetryShell
--
cgit v1.2.3-70-g09d2