DevOps-Bash-tools/python_compile.sh

#!/usr/bin/env bash
#  vim:ts=4:sts=4:sw=4:et
#
#  Author: Hari Sekhon
#  Date: 2015-05-25 01:38:24 +0100 (Mon, 25 May 2015)
#
#  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 improve or steer this or other code I publish
#
#  https://www.linkedin.com/in/harisekhon
#

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

# shellcheck source=lib/utils.sh
. "$srcdir/lib/utils.sh"

python="${PYTHON:-python}"

filelist="$(find "${1:-.}" -maxdepth 2 -type f -iname '*.py' -o -iname '*.jy' | sort)"

if [ -z "$filelist" ]; then
    echo "no Python / Jython files found to compile"
    echo
    echo "usage: ${0##*/} <python_file_or_directory>"
    echo
    return 0 &>/dev/null || :
    exit 0
fi

section "Compiling Python / Jython files"

if ! type -P "$python" &>/dev/null; then
    echo "$python not found in \$PATH, skipping compile"
    exit 0
fi

start_time="$(start_timer)"

# opts:
#
# -O  - optimize
# -3  - warn on Python 3 incompatibilies that 2to3 cannot easily fix
# -t  - warn on inconsistent use of tabs

opts=""

if python -V 2>&1 | grep -q 'Python 2'; then
    opts="$opts -3"
fi

if ! is_travis && ! type -P pypy &>/dev/null; then
    opts="$opts -t"
fi

if [ -n "${NOCOMPILE:-}" ]; then
    echo "\$NOCOMPILE environment variable set, skipping python compile"
elif [ -n "${QUICK:-}" ]; then
    echo "\$QUICK environment variable set, skipping python compile"
else
    if [ -n "${FAST:-}" ]; then
        # want opt expansion
        # shellcheck disable=SC2086
        python $opts -m compileall "${1:-.}" || :
    else
        for x in $filelist; do
            type isExcluded &>/dev/null && isExcluded "$x" && continue
            echo "compiling $x"
            # want opt expansion
            # shellcheck disable=SC2086
            "$python" -O $opts -m py_compile "$x"
        done
    fi
fi

time_taken "$start_time"
section2 "Finished compiling Python / Jython files"
echo