Add better support for pyenv virtualenvs

modules/python/init.zsh

@@ -35,16 +35,21 @@ 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 zstyle -T ':prezto:module:python' skip-virtualenvwrapper-init; then
-if (( $? && $+commands[virtualenvwrapper.sh] )); 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
 
+  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.
 if (( $+commands[pip] )); then