From 1050a0a290324f3b96256d3b1fefe4a84ab402d0 Mon Sep 17 00:00:00 2001
From: Kaleb Elwert <belak@coded.io>
Date: Tue, 18 Apr 2017 03:14:22 -0700
Subject: [PATCH] Add better support for pyenv virtualenvs
---
modules/python/init.zsh | 11 ++++++++---
1 file changed, 8 insertions(+), 3 deletions(-)
diff --git a/modules/python/init.zsh b/modules/python/init.zsh
index 8b63251a..e40bca6b 100644
--- a/modules/python/init.zsh
+++ b/modules/python/init.zsh
@@ -35,15 +35,20 @@ if (( ! $+commands[python] && ! $+commands[pyenv] )); then
fi
# Load virtualenvwrapper into the shell session, unless requested not to
-zstyle -t ':prezto:module:python' skip-virtualenvwrapper-init
-if (( $? && $+commands[virtualenvwrapper.sh] )); then
+if zstyle -T ':prezto:module:python' skip-virtualenvwrapper-init; then
# Set the directory where virtual environments are stored.
export WORKON_HOME="${WORKON_HOME:-$HOME/.virtualenvs}"
# Disable the virtualenv prompt.
VIRTUAL_ENV_DISABLE_PROMPT=1
- source "$commands[virtualenvwrapper.sh]"
+ if (( $+commands[pyenv-virtualenvwrapper] )); then
+ pyenv virtualenvwrapper
+ elif (( $+commands[pyenv-virtualenv-init] )); then
+ eval "$(pyenv virtualenv-init -)"
+ elif (( $+commands[virtualenvwrapper.sh] )); then
+ source "$commands[virtualenvwrapper.sh]"
+ fi
fi
# Load PIP completion.