scripts/misc/venv

94 lines
1.9 KiB
Plaintext
Raw Normal View History

2020-04-02 08:38:36 +00:00
#!/bin/zsh
2020-04-07 09:16:29 +00:00
source ${$(realpath "$0"):h:h}/lib/common.zsh
zparseopts -D -E \
c=cdvenv -cdvenv=cdvenv \
h=help -help=help \
n=newvenv -new=newvenv \
t=tmsu -tmsu=tmsu \
p=pyenv -pyenv=pyenv \
f=find -find=find
if [[ $help ]]; then
2020-04-02 08:38:36 +00:00
<<-HELP
Usage: ${0:t} [-c] [DIR]
${0:t} -t
If DIR contains bin/activate, start a shell with that virtualenv.
Otherwise recursively look up virtualenvs in DIR (by looking for dirs ending
in "venv") and select one with fzf.
2020-04-07 09:16:29 +00:00
OPTIONS:
2020-04-02 08:38:36 +00:00
-c, --cdvenv Change to the venv directory instead of going back to the
2020-04-07 09:16:29 +00:00
current working directory (not for pyenv virtualenvs)
-t, --tmsu list only venvs named "venv" in a directory with tmsu tag lang=python
-p, --pyenv Use pyenv virtualenvs
-f, --find Search for virtualenvs by looking for directories ending with 'venv'
If neither -p or -f are given, the default behaviour is to use both
2020-04-02 08:38:36 +00:00
HELP
exit
fi
ORIGDIR=$PWD
venvsh() {
cd $1
. bin/activate
if [[ -z $cdvenv ]]; then
cd $ORIGDIR
fi
$SHELL
}
2020-04-07 09:16:29 +00:00
pyenvsh() {
export PYENV_VIRTUALENV_DISABLE_PROMPT=1
eval "$(pyenv init -)"
eval "$(pyenv virtualenv-init -)"
pyenv activate "$1"
$SHELL
}
2020-04-02 08:38:36 +00:00
if [[ -n $tmsu ]]; then
venv=$(
for i in $(tmsu files --directory lang=python); do
[[ -d $i/venv ]] && echo $i/venv
done | fzf)
if [[ -n $venv ]]; then
exec venvsh $venv
else
exit 1
fi
fi
2020-04-07 09:16:29 +00:00
if [[ $newvenv ]]; then
2020-04-02 08:38:36 +00:00
python -mvenv ${1:-venv}
exec venvsh ${1:-venv}
fi
if [[ -n $1 && -e $1/bin/activate ]]; then
exec venvsh $1
fi
2020-04-07 09:16:29 +00:00
pyenvs() {
if exists pyenv && pyenv commands | grep -q 'virtualenvs'; then
pyenv virtualenvs --bare --skip-aliases | sed -e 's/^/pyenv!/'
fi
}
if [[ $pyenv && ! $find ]]; then
venv=$(pyenvs | fzf)
elif [[ ! $pyenv && $find ]]; then
venv=$(fd -I -t d 'venv$' $1 | fzf)
else
venv=$(cat <(fd -I -t d 'venv$' $1) <(pyenvs) | fzf)
fi
2020-04-02 08:38:36 +00:00
if [[ -n $venv ]]; then
2020-04-07 09:16:29 +00:00
if [[ $venv =~ pyenv\!* ]]; then
exec pyenvsh ${venv##pyenv!}
else
exec venvsh $venv
fi
2020-04-02 08:38:36 +00:00
fi