blob: 3a687642529ad26d92c5f393f1e247e956971535 [file] [log] [blame] [edit]
#!/bin/bash
# Copyright 2017 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
#
# Starts a python interpreter in virtualenv.
#
# This script will set up a virtualenv when it has not been created yet and
# executes the Python interpreter.
#
# The canonical version of this script is in infra_virtualenv repository.
# See infra_virtualenv/README.md about how to adopt virtualenv to your project.
set -eu
# Change this constant to the path(s) to infra_virtualenv directory when you
# copy this script to other repos.
# A path can be a relative path from this script, or an absolute path. If this
# array contains multiple paths, they are searched in the listed order.
readonly -a infra_virtualenv_paths=(
"../../../../../infra_virtualenv"
"/opt/infra_virtualenv"
)
readonly bin_dir="$(readlink -e -- "$(dirname -- "$0")")"
if [[ ! -d "${bin_dir}" ]]; then
echo "ERROR: Can not locate the location of python_env!" >&2
exit 1
fi
realpath() {
pushd "${bin_dir}" > /dev/null 2>&1
readlink -e -- "$1"
popd > /dev/null 2>&1
}
find_create_venv() {
local p
for p in "${infra_virtualenv_paths[@]}"; do
local create_venv=$(realpath "${p}/bin/create_venv")
if [[ -f "${create_venv}" ]]; then
echo "${create_venv}"
break
fi
done
}
readonly create_venv=$(find_create_venv)
if [[ ! -f "${create_venv}" ]]; then
cat <<EOF >&2
ERROR: create_venv script could not be located.
You need to update a constant inside python_venv, or your checkout might be
incomplete.
EOF
exit 1
fi
readonly extra_imports_dir=$(realpath ../venv)
if [[ ! -d "${extra_imports_dir}" ]]; then
cat <<EOF >&2
ERROR: ${bin_dir}/../venv does not exist
See infra_virtualenv/README.md for details.
EOF
exit 1
fi
readonly venv_dir=$("${create_venv}" "${extra_imports_dir}/requirements.txt")
if [[ ! -d "${venv_dir}" ]]; then
echo "ERROR: Failed to set up a virtualenv." >&2
exit 1
fi
export PYTHONPATH="${extra_imports_dir}"
exec "${venv_dir}/bin/python" "$@"