DevOps-Bash-tools/lib/python.sh

#!/usr/bin/env bash
#  vim:ts=4:sts=4:sw=4:et
#
#  Author: Hari Sekhon
#  Date: 2020-04-03 13:14:22 +0100 (Fri, 03 Apr 2020)
#
#  https://github.com/harisekhon/bash-tools
#
#  License: see accompanying Hari Sekhon LICENSE file
#
#  If you're using my code you're welcome to connect with me on LinkedIn and optionally send me feedback to help steer this or other code I publish
#
#  https://www.linkedin.com/in/harisekhon
#

set -euo pipefail
[ -n "${DEBUG:-}" ] && set -x
srcdir_bash_tools_python="$(dirname "${BASH_SOURCE[0]}")"

# shellcheck disable=SC1090
. "$srcdir_bash_tools_python/ci.sh"

# shellcheck disable=SC1090
. "$srcdir_bash_tools_python/os.sh"

# shellcheck disable=SC1090
#. "$srcdir_bash_tools_python/../.bash.d/python.sh"

if is_semaphore_ci; then
    set -x
fi

if is_CI; then
    echo
    echo "Python and Pip installations:"
    for x in python python2 python3 pip pip2 pip3; do
        find / -type f -name "$x" -exec ls -l {} \; -o \
               -type l -name "$x" -exec ls -l {} \; 2>/dev/null || :
    done
    echo
    echo
fi

# shellcheck disable=SC2034
python="${PYTHON:-python}"

if command -v pip >/dev/null 2>&1; then
    python="$(command -v "$python")"
fi

if [ -n "${PIP:-}" ]; then
    pip="$PIP"
else
    pip="$(command -v pip 2>/dev/null)" ||
    pip="$(command -v pip2 2>/dev/null)" ||
    pip="$(command -v pip3 2>/dev/null)" || :
    if [ -z "$pip" ]; then
        echo "pip not found, falling back to trying just 'pip'" >&2
        pip=pip
    fi
fi

if is_mac &&
   ! command -v "$pip" >/dev/null 2>&1; then
    echo "pip not installed, trying to install manually..."
    brew install openssl || :
    curl -sS https://bootstrap.pypa.io/get-pip.py -o get-pip.py
    brew_prefix="$(brew --prefix)"
    export OPENSSL_INCLUDE="$brew_prefix/opt/openssl/include"
    export OPENSSL_LIB="$brew_prefix/opt/openssl/lib"
    python get-pip.py
fi

pip="$(command -v "$pip")"

set +eo pipefail
# split steps for easier CI debugging in DEBUG mode
python_major_version="$("$python" -V 2>&1)"
python_major_version="$(echo "$python_major_version" | grep -Eom1 '[[:digit:]]+\.[[:digit:]]+')"
pip_python_major_version="$("$pip" -V 2>&1)"
pip_python_major_version="$(echo "$pip_python_major_version" | grep -Eom1 '\(python [[:digit:]]+\.[[:digit:]]+\)' | sed 's/(python[[:space:]]*//; s/)//')"
set -eo pipefail

if [ -n "${python_major_version:-}" ] &&
   [ -n "${pip_python_major_version:-}" ]; then
    if [ "$python_major_version" != "$pip_python_major_version" ]; then
        echo "Python major version '$python_major_version' != Pip Python major version '$pip_python_major_version' !!"
        echo
        echo "Python PyPI modules will not be installed to the correct site-packages and will lead to import failures later on"
        echo
        echo "Fix your \$PATH or \$PYTHON / \$PIP to be aligned to the same installation"
        exit 1
    fi
fi

inside_virtualenv(){
    if [ -n "${VIRTUAL_ENV:-}" ] ||
       #[ -n "${PYENV_ROOT:-}" ] ||
       [ -n "${CODESHIP_VIRTUALENV:-}" ] ||
       [ -n "${CONDA_DEFAULT_ENV:-}" ]; then
        return 0
    elif [ -n "${PYENV_ROOT:-}" ]; then
        if command -v "$python" | grep -q "$PYENV_ROOT" &&
           command -v "$pip" | grep -q "$PYENV_ROOT"; then
            return 0
        fi
    # GitHub Actions Python versions
    elif command -v "$python" | grep -Eqi '/hostedtoolcache/'; then
    #or
    #elif command -v "$pip" | grep -Eqi '/hostedtoolcache/'; then
        return 0
    # CircleCI uses /opt/circleci/.pyenv/shims/python
    # Codeship path when using virtualenv
    elif command -v "$python" | grep -Eqi '/\.pyenv/|/shims/'; then
        return 0
    elif command -v "$pip" | grep -Eqi '/\.pyenv/|/shims/'; then
        return 0
    fi
    return 1
}