sentry/scripts/ensure-venv.sh

#!/bin/bash

# optionally opt out of virtualenv creation
# WARNING: this will be removed (most likely renamed) soon!
if [[ "$SENTRY_NO_VIRTUALENV_CREATION" == "1" ]]; then
    exit 0
fi

red="$(tput setaf 1)"
yellow="$(tput setaf 3)"
bold="$(tput bold)"
reset="$(tput sgr0)"

venv_name=".venv"

die() {
    cat <<EOF
$@
EOF
    exit 1
}

if [[ -n "$VIRTUAL_ENV" ]]; then
    minor=$(python -c "import sys; print(sys.version_info[1])")
    # If .venv is less than Python 3.6 fail
    [[ "$minor" -lt 6 ]] &&
        die "Remove $VIRTUAL_ENV and try again since the Python version installed should be at least 3.6."
    # If .venv is created with Python greater than 3.6 you might encounter problems and we want to ask you to downgrade
    # unless you explicitely set an environment variable
    if [[ "$minor" -gt 6 ]]; then
        if [[ -n "$SENTRY_PYTHON_VERSION" ]]; then
            cat <<EOF
${yellow}${bold}
You have explicitly set a non-recommended Python version (${SENTRY_PYTHON_VERSION}). You're on your own.
${reset}
EOF
        else
            cat <<EOF
${red}${bold}
ERROR! You are running a virtualenv with a Python version different than 3.6
We recommend you start with a fresh virtualenv or to set the variable SENTRY_PYTHON_VERSION
to the Python version you want to use (e.g. 3.7).
${reset}
EOF
            exit 1
        fi
    fi
else
    if [[ ! -f "${venv_name}/bin/activate" ]]; then
        die "You don't seem to have a virtualenv. Please create one by running: python3.6 -m venv ${venv_name}"
    fi
    die "You have a virtualenv, but it doesn't seem to be activated. Please run: source ${venv_name}/bin/activate"
fi

# Somehow it does not succeed unless I exit with 0
exit 0