From 172abf341062c40e827a6ca42cf7ecf749205b1d Mon Sep 17 00:00:00 2001
From: Daylin Morgan <daylinmorgan@gmail.com>
Date: Thu, 1 Sep 2022 04:31:07 -0500
Subject: [PATCH] add brute-force python env activator function

---
 lib/conda.zsh | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/lib/conda.zsh b/lib/conda.zsh
index bf965e6..bd5a663 100755
--- a/lib/conda.zsh
+++ b/lib/conda.zsh
@@ -21,3 +21,31 @@ fi
 # <<< conda initialize <<<
 
 ! is-tty && [ -d "$HOME/mambaforge/envs/dev" ] && conda activate dev
+
+snake() {
+	if [[ $1 == "no" ]]; then
+		if [[ $VIRTUAL_ENV != "" ]]; then
+			echo "deactivate python virtualenv: $VIRTUAL_ENV"
+			deactivate
+		elif [[ $CONDA_DEFAULT_ENV != "" ]]; then
+			echo "deactivating conda env: $CONDA_DEFAULT_ENV"
+			conda deactivate
+		else
+			echo 'no environment to leave'
+		fi
+	else
+		if [[ -d $(pwd)/env ]]; then
+      printf 'activating project-specific env: %s\n' "${PWD##*/}/env"
+			conda activate ./env
+		elif [[ -d $(pwd)/venv ]]; then
+			echo "activating python virtualenv"
+			source ./venv/bin/activate
+		elif [[ -d $(pwd)/.venv ]]; then
+			echo "activating python virtualenv"
+			source ./.venv/bin/activate
+		else
+			echo "is there an environment to activate?"
+      echo "I was expecting either a conda(env) or virtualenv(venv,.venv)"
+		fi
+	fi
+}