mirror of
https://github.com/pyenv/pyenv-virtualenv.git
synced 2026-09-10 07:16:18 -04:00
37ab83f1b0
* With 160 envs, cuts time from ~14s to ~2.5s * Change the output to be more like `pyenv-versions' * Shellcheck fixes * virtualenv: Only show non-envs for completion since env engines don't support piggy-backing envs * Remove dead/unused code * Fix IFS handling `IFS=: <command>' doesn't apply that IFS to expansions in <command>, it has to be set separately earlier * Adjust and refactor tests --------- Co-authored-by: Ivan Pozdeev <vano@mail.mipt.ru>
116 lines
3.6 KiB
Bash
Executable file
116 lines
3.6 KiB
Bash
Executable file
#!/usr/bin/env bash
|
|
#
|
|
# Summary: Display real_prefix for a Python virtualenv version
|
|
# Usage: pyenv virtualenv-prefix [<virtualenv>]
|
|
#
|
|
|
|
set -e
|
|
[ -n "$PYENV_DEBUG" ] && set -x
|
|
if [ -L "${BASH_SOURCE}" ]; then
|
|
READLINK=$(type -p greadlink readlink | head -1)
|
|
if [ -z "$READLINK" ]; then
|
|
echo "pyenv: cannot find readlink - are you missing GNU coreutils?" >&2
|
|
exit 1
|
|
fi
|
|
resolve_link() {
|
|
$READLINK -f "$1"
|
|
}
|
|
script_path=$(resolve_link ${BASH_SOURCE})
|
|
else
|
|
script_path=${BASH_SOURCE}
|
|
fi
|
|
|
|
. ${script_path%/*}/../libexec/pyenv-virtualenv-realpath
|
|
|
|
if [ -z "$PYENV_ROOT" ]; then
|
|
PYENV_ROOT="${HOME}/.pyenv"
|
|
fi
|
|
|
|
OLDIFS="$IFS"
|
|
IFS=:
|
|
if [ -n "$1" ]; then
|
|
# $@ is not affected by IFS
|
|
versions=("$@")
|
|
PYENV_VERSION="${versions[*]}"
|
|
export PYENV_VERSION
|
|
else
|
|
versions=($(pyenv-version-name))
|
|
fi
|
|
IFS="$OLDIFS"
|
|
|
|
append_virtualenv_prefix() {
|
|
if [ -d "${VIRTUALENV_PREFIX_PATH}" ]; then
|
|
VIRTUALENV_PREFIX_PATHS=("${VIRTUALENV_PREFIX_PATHS[@]}" "${VIRTUALENV_PREFIX_PATH:-${PYENV_PREFIX_PATH}}")
|
|
else
|
|
echo "pyenv-virtualenv: version \`${version}' is not a virtualenv" 1>&2
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
VIRTUALENV_PREFIX_PATHS=()
|
|
for version in "${versions[@]}"; do
|
|
if [ "$version" = "system" ]; then
|
|
echo "pyenv-virtualenv: version \`${version}' is not a virtualenv" 1>&2
|
|
exit 1
|
|
fi
|
|
|
|
# In the vast majority of cases, there's a direct match and
|
|
# not spawning `pyenv-prefix' saves about half the invocation time
|
|
# with a signle argument which accumulates when called repeatedly
|
|
# (e.g. from `pyenv-virtualenvs').
|
|
# `pyenv-prefix' also does not have hooks to worry about.
|
|
# XXX: refactor the test into a shared module?
|
|
PYENV_PREFIX_PATH="${PYENV_ROOT}/versions/${version}"
|
|
if [[ ! -d "$PYENV_PREFIX_PATH" ]]; then
|
|
PYENV_PREFIX_PATH="$(pyenv-prefix "${version}")"
|
|
fi
|
|
|
|
if [ -x "${PYENV_PREFIX_PATH}/bin/python" ]; then
|
|
if [ -f "${PYENV_PREFIX_PATH}/bin/activate" ]; then
|
|
if [ -f "${PYENV_PREFIX_PATH}/bin/conda" ]; then
|
|
# conda
|
|
VIRTUALENV_PREFIX_PATH="${PYENV_PREFIX_PATH}"
|
|
else
|
|
if [ -f "${PYENV_ROOT}/versions/${version}/pyvenv.cfg" ]; then
|
|
# venv
|
|
virtualenv_binpath="$(cut -b 1-1024 "${PYENV_ROOT}/versions/${version}/pyvenv.cfg" | sed -n '/^ *home *= */s///p' || true)"
|
|
VIRTUALENV_PREFIX_PATH="${virtualenv_binpath%/bin}"
|
|
else
|
|
# virtualenv
|
|
if [ -d "${PYENV_ROOT}/versions/${version}/Lib" ]; then
|
|
# jython
|
|
virtualenv_libpath="${PYENV_ROOT}/versions/${version}/Lib"
|
|
else
|
|
if [ -d "${PYENV_ROOT}/versions/${version}/lib-python" ]; then
|
|
# pypy
|
|
virtualenv_libpath="${PYENV_ROOT}/versions/${version}/lib-python"
|
|
else
|
|
virtualenv_libpath="${PYENV_ROOT}/versions/${version}/lib"
|
|
fi
|
|
fi
|
|
virtualenv_orig_prefix="$(find "${virtualenv_libpath}/" -maxdepth 2 -type f -and -name "orig-prefix.txt" 2>/dev/null | head -1)"
|
|
if [ -f "${virtualenv_orig_prefix}" ]; then
|
|
VIRTUALENV_PREFIX_PATH="$(cat "${virtualenv_orig_prefix}" 2>/dev/null || true)"
|
|
fi
|
|
fi
|
|
fi
|
|
append_virtualenv_prefix
|
|
elif [ -d "${PYENV_PREFIX_PATH}/conda-meta" ]; then
|
|
# conda
|
|
VIRTUALENV_PREFIX_PATH="$(realpath "${PYENV_PREFIX_PATH}"/../..)"
|
|
append_virtualenv_prefix
|
|
else
|
|
echo "pyenv-virtualenv: version \`${version}' is not a virtualenv" 1>&2
|
|
exit 1
|
|
fi
|
|
else
|
|
echo "pyenv-virtualenv: \`python' not found in version \`${version}'" 1>&2
|
|
exit 1
|
|
fi
|
|
done
|
|
|
|
OLDIFS="$IFS"
|
|
IFS=:
|
|
echo "${VIRTUALENV_PREFIX_PATHS[*]}"
|
|
IFS="$OLDIFS"
|