mirror of
https://github.com/ohmyzsh/ohmyzsh.git
synced 2024-12-21 03:41:57 +00:00
b9d5a39098
Using lazy loading for virtualenvwrapper gives a mariginal speed improvement and doesn't stop workon_cd from working. It has the undesired effect of forcing you to call certain virtualenv commands twice before they work (only once per shell instantiation).
40 lines
1.6 KiB
Bash
40 lines
1.6 KiB
Bash
wrapsource=`which virtualenvwrapper_lazy.sh`
|
|
|
|
if [[ -f "$wrapsource" ]]; then
|
|
source $wrapsource
|
|
|
|
if [[ ! $DISABLE_VENV_CD -eq 1 ]]; then
|
|
# Automatically activate Git projects' virtual environments based on the
|
|
# directory name of the project. Virtual environment name can be overridden
|
|
# by placing a .venv file in the project root with a virtualenv name in it
|
|
function workon_cwd {
|
|
# Check that this is a Git repo
|
|
PROJECT_ROOT=`git rev-parse --show-toplevel 2> /dev/null`
|
|
if (( $? == 0 )); then
|
|
# Check for virtualenv name override
|
|
ENV_NAME=`basename "$PROJECT_ROOT"`
|
|
if [[ -f "$PROJECT_ROOT/.venv" ]]; then
|
|
ENV_NAME=`cat "$PROJECT_ROOT/.venv"`
|
|
fi
|
|
# Activate the environment only if it is not already active
|
|
if [[ "$VIRTUAL_ENV" != "$WORKON_HOME/$ENV_NAME" ]]; then
|
|
if [[ -e "$WORKON_HOME/$ENV_NAME/bin/activate" ]]; then
|
|
workon "$ENV_NAME" && export CD_VIRTUAL_ENV="$ENV_NAME"
|
|
fi
|
|
fi
|
|
elif [ $CD_VIRTUAL_ENV ]; then
|
|
# We've just left the repo, deactivate the environment
|
|
# Note: this only happens if the virtualenv was activated automatically
|
|
deactivate && unset CD_VIRTUAL_ENV
|
|
fi
|
|
unset PROJECT_ROOT
|
|
}
|
|
|
|
# New cd function that does the virtualenv magic
|
|
function cd {
|
|
builtin cd "$@" && workon_cwd
|
|
}
|
|
fi
|
|
else
|
|
print "zsh virtualenvwrapper plugin: Cannot find virtualenvwrapper_lazy.sh. Please install with \`pip install virtualenvwrapper\`."
|
|
fi
|