python: respect PYENV_ROOT if already set

Fixes #1578

modules/python/init.zsh

@@ -7,15 +7,18 @@
 #   Patrick Bos <egpbos@gmail.com>
 #
 
-# Load manually installed pyenv into the shell session.
+# Load manually installed pyenv into the path
-if [[ -s "$HOME/.pyenv/bin/pyenv" ]]; then
+if [[ -n "$PYENV_ROOT" && -s "$PYENV_ROOT/bin/pyenv" ]]; then
+  path=("$PYENV_ROOT/bin" $path)
+elif [[ -s "$HOME/.pyenv/bin/pyenv" ]]; then
   path=("$HOME/.pyenv/bin" $path)
-  export PYENV_ROOT=$(pyenv root)
+fi
-  eval "$(pyenv init - --no-rehash zsh)"
 
-# Load package manager installed pyenv into the shell session.
+# Load pyenv into the current python session
-elif (( $+commands[pyenv] )); then
+if (( $+commands[pyenv] )); then
+  if [[ -z "$PYENV_ROOT" ]]; then
     export PYENV_ROOT=$(pyenv root)
+  fi
   eval "$(pyenv init - --no-rehash zsh)"
 
 # Prepend PEP 370 per user site packages directory, which defaults to