bin: add some python3 variants
This will allow python3 wrappers to get python3-specific behavior.
We can then update the old ones to issue warnings to see if anyone
is still using them.
BUG=chromium:1052692
TEST=CQ passes
Change-Id: Idab721ba132f89f1b5a3f70b539bd98362bdc0b2
diff --git a/bin/create_venv3 b/bin/create_venv3
new file mode 100755
index 0000000..8a53419
--- /dev/null
+++ b/bin/create_venv3
@@ -0,0 +1,15 @@
+#!/bin/bash
+# Copyright 2020 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.
+#
+# Create or update a virtualenv.
+#
+# $ create_venv path/to/requirements.txt
+#
+# This creates a versioned virtualenv with the given requirements
+# installed. The path to the created virtualenv is printed to stdout.
+set -eu
+basedir=$(readlink -f -- "$(dirname -- "${BASH_SOURCE[0]}" )")
+venvdir=$(cd "${basedir}"; cd ../venv; pwd)
+PYTHONPATH="${venvdir}" python3 -m cros_venv.scripts.create_venv "$@"
diff --git a/bin/python_venv3 b/bin/python_venv3
new file mode 100755
index 0000000..dd31699
--- /dev/null
+++ b/bin/python_venv3
@@ -0,0 +1,73 @@
+#!/bin/bash
+# Copyright 2020 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=(
+ ".."
+)
+
+readonly bin_dir="$(readlink -e -- "$(dirname -- "$0")")"
+if [[ ! -d "${bin_dir}" ]]; then
+ echo "python_venv: Cannot locate 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_venv3")
+ 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
+python_venv: create_venv script could not be located
+python_venv: Possible causes: python_venv configured incorrectly or
+python_venv: incomplete repo checkout
+EOF
+ exit 1
+fi
+
+readonly extra_imports_dir=$(realpath ../venv)
+if [[ ! -d "${extra_imports_dir}" ]]; then
+ cat <<EOF >&2
+python_venv: ${bin_dir}/../venv does not exist
+python_venv: 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" "$@"
diff --git a/bin/run_tests b/bin/run_tests
index 54b5eae..1891dcd 100755
--- a/bin/run_tests
+++ b/bin/run_tests
@@ -12,4 +12,5 @@
find . -name "*.pyc" -print0 | xargs -0 rm -f
-exec "${bin_dir}/python_venv" -m pytest "$@"
+"${bin_dir}/python_venv" -m pytest "$@"
+"${bin_dir}/python_venv3" -m pytest "$@"
