Run rehash after easy_install

pyenv · Jan 20, 2014 · e4aeb30 · e4aeb30
1 parent 87b3b6c
commit e4aeb30
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 2 deletions.
diff --git a/etc/pyenv.d/exec/pip.bash b/etc/pyenv.d/exec/pip.bash
@@ -20,7 +20,7 @@ abs_dirname() {
 
 PYENV_PIP_REHASH_ROOT="$(abs_dirname "${BASH_SOURCE[0]}")/../../.."
 
-if [[ "${PYENV_COMMAND##*/}" == "pip" ]]; then
-  PYENV_COMMAND_PATH="${PYENV_PIP_REHASH_ROOT}/libexec/pip"
+if [ -x "${PYENV_PIP_REHASH_ROOT}/libexec/${PYENV_COMMAND##*/}" ]; then
+  PYENV_COMMAND_PATH="${PYENV_PIP_REHASH_ROOT}/libexec/${PYENV_COMMAND##*/}"
   PYENV_BIN_PATH="${PYENV_PIP_REHASH_ROOT}/libexec"
 fi
diff --git a/libexec/easy_install b/libexec/easy_install
@@ -0,0 +1,19 @@
+#!/usr/bin/env bash
+
+set -e
+[ -n "$PYENV_DEBUG" ] && set -x
+
+PYENV_COMMAND_PATH="$(pyenv-which "$(basename "$0")")"
+PYENV_BIN_PATH="${PYENV_COMMAND_PATH%/*}"
+
+export PATH="${PYENV_BIN_PATH}:${PATH}"
+
+STATUS=0
+"$PYENV_COMMAND_PATH" "$@" || STATUS="$?"
+
+# Run `pyenv-rehash` after a successful installation.
+if [ "$STATUS" == "0" ]; then
+  pyenv-rehash
+fi
+
+exit "$STATUS"