|
|
|
#!/usr/bin/env bash
|
|
|
|
# vim:ts=4:sts=4:sw=4:et
|
|
|
|
#
|
|
|
|
# Author: Hari Sekhon
|
|
|
|
# Date: 2021-05-07 16:28:58 +0100 (Fri, 07 May 2021)
|
|
|
|
#
|
|
|
|
# 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="$(dirname "${BASH_SOURCE[0]}")"
|
|
|
|
|
|
|
|
# shellcheck disable=SC1090
|
|
|
|
. "$srcdir/lib/utils.sh"
|
|
|
|
|
|
|
|
# shellcheck disable=SC2034,SC2154
|
|
|
|
usage_description="
|
|
|
|
Finds all Python modules installed in a script and installs them if not already present
|
|
|
|
|
|
|
|
Uses adjacent scripts:
|
|
|
|
|
|
|
|
python_translate_import_to_pip.sh
|
|
|
|
python_pip_install_if_absent.sh
|
|
|
|
|
|
|
|
|
|
|
|
Will break on custom local modules since they won't be found on PyPI, use --exclude <regex> to avoid that. Regex is in ERE format
|
|
|
|
|
|
|
|
Eg.
|
|
|
|
${0##*/} *.py --exclude harisekhon # excludes my personal local library which is not on PyPI
|
|
|
|
|
|
|
|
|
|
|
|
If you supply .pyc or .pyo filenames, it will infer to .py instead. This is useful if calling from a Makefile looking for .pyc or .pyo dynamic targets
|
|
|
|
"
|
|
|
|
|
|
|
|
# used by usage() in lib/utils.sh
|
|
|
|
# shellcheck disable=SC2034
|
|
|
|
usage_args="<script1.py> [<script2.py> <script3.py>]"
|
|
|
|
|
|
|
|
help_usage "$@"
|
|
|
|
|
|
|
|
min_args 1 "$@"
|
|
|
|
|
|
|
|
scripts=()
|
|
|
|
exclude_regex=""
|
|
|
|
|
|
|
|
until [ $# -lt 1 ]; do
|
|
|
|
case "$1" in
|
|
|
|
-e|--exclude) exclude_regex="${2:-}"
|
|
|
|
shift || :
|
|
|
|
;;
|
|
|
|
-*) usage
|
|
|
|
;;
|
|
|
|
*) if [[ "$1" =~ \.py[co] ]]; then
|
|
|
|
filename="$1"
|
|
|
|
filename="${filename%.pyc}"
|
|
|
|
filename="${filename%.pyo}"
|
|
|
|
filename="$filename.py"
|
|
|
|
scripts+=("$filename")
|
|
|
|
else
|
|
|
|
scripts+=("$1")
|
|
|
|
fi
|
|
|
|
;;
|
|
|
|
esac
|
|
|
|
shift || :
|
|
|
|
done
|
|
|
|
|
|
|
|
if [ ${#scripts[@]} -eq 0 ]; then
|
|
|
|
usage
|
|
|
|
fi
|
|
|
|
|
|
|
|
pip_modules_import="$(
|
|
|
|
grep -Eh '^[[:space:]]*import[[:space:]]' "${scripts[@]}" |
|
|
|
|
awk '{print $2}' |
|
|
|
|
tr -d '\r' |
|
|
|
|
sort -u
|
|
|
|
)"
|
|
|
|
pip_modules_from_import="$(
|
|
|
|
grep -Eho '^[[:space:]]*from[[:space:]][^[[:space:]_]+[[:space:]]+import[[:space:]]+[^[:space:]]+$' "${scripts[@]}" |
|
|
|
|
sed 's/^[[:space:]]*from[[:space:]]*//; s/[[:space:]][[:space:]]*import[[:space:]].*$//' |
|
|
|
|
sort -u
|
|
|
|
)"
|
|
|
|
|
|
|
|
pip_modules="$(
|
|
|
|
sort -u <<-EOF |
|
|
|
|
$pip_modules_import
|
|
|
|
$pip_modules_from_import
|
|
|
|
EOF
|
|
|
|
if [ -n "$exclude_regex" ]; then
|
|
|
|
grep -Ev "$exclude_regex" || :
|
|
|
|
else
|
|
|
|
cat
|
|
|
|
fi
|
|
|
|
)"
|
|
|
|
|
|
|
|
pip_modules="$("$srcdir/python_translate_import_to_module.sh" <<< "$pip_modules")"
|
|
|
|
|
|
|
|
"$srcdir/python_pip_install_if_absent.sh" <<< "$pip_modules"
|