luca
/
dotfiles
1
0
Fork 0
Code

add functionality for recursive venvs, fix minor bug with always toggling venv on CD

This commit is contained in:
Luca Bilke 2022-08-04 18:12:05 +02:00
parent 05b1410933
commit 06a2986e74
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
.local/share/zsh/plugins/autopyenv/autopyenv.plugin.zsh
View File

@ -2,13 +2,13 @@ PYENV_DIR="${PYENVS_HOME:-${XDG_DATA_HOME:-$HOME/.local/share}/virtualenv}"
function chpwd_activate(){
[[ "$(pwd)" == "/" ]] && return 0
for pydir in $(ls $PYENV_DIR); do
if [[ "$(pwd|sed -e s@/@_@g|cut -c2-)" =~ "^$pydir$" ]]; then
if [[ "$(pwd|sed -e s@/@_@g|cut -c2-)" =~ "^$pydir$" ]] || [[ "r-$(pwd|sed -e s@/@_@g|cut -c2-)" =~ "^$pydir" ]]; then
if [ "x$VIRTUAL_ENV" != "x$PYENV_DIR/$pydir" ]; then
export VIRTUAL_ENV_DISABLE_PROMPT=1
export PS1="$PS1"
export VIRTUAL_ENV_DISABLE_PROMPT=1
source "$PYENV_DIR/$pydir/bin/activate"
return
fi
return
fi
done
if [ "x$VIRTUAL_ENV" != "x" ]; then
@ -18,14 +18,17 @@ function chpwd_activate(){
fi
}
function venv-here(){
function venv(){
[[ "$(pwd)" == "/" ]] && echo "Cannot create venv at root" && return 1
[[ "$(pwd)" != "/" ]] && envdir=$(pwd|sed -e s@/@_@g|cut -c2-)
[[ "$1" == "-r" ]] && envdir="r-$envdir" # create a venv that will be activated for all subdirectories as well
if [ ! -e "$PYENV_DIR/${envdir}" ]; then
echo "Creating python venv for ${envdir}.."
mkdir "$PYENV_DIR" -p
python3 -m venv "$PYENV_DIR/${envdir}"
echo "Activating virtual env ${envdir}"
[[ ! ${PS1:0:3} == "  " ]] && export PS1="$PS1"
export VIRTUAL_ENV_DISABLE_PROMPT=1
source "$PYENV_DIR/${envdir}/bin/activate"
else
echo "A venv for this path already exists"