mirror of
https://github.com/zardus/ctf-tools
synced 2024-12-13 22:42:38 +00:00
34 lines
863 B
Bash
Executable file
34 lines
863 B
Bash
Executable file
#!/bin/bash
|
|
set -e -o pipefail
|
|
|
|
PY_VERSION=2
|
|
if [[ "$0" =~ \.+pip3 ]]; then
|
|
PY_VERSION=3
|
|
fi
|
|
if [[ "$0" =~ \.+pip2 ]]; then
|
|
PY_VERSION=2
|
|
fi
|
|
|
|
CTF_TOOLS_VE="ctftools"
|
|
if [[ "PY_VERSION" -eq 3 ]]; then
|
|
CTF_TOOLS_VE="${CTF_TOOLS_VE}3"
|
|
fi
|
|
|
|
export WORKON_HOME=~/.virtualenvs
|
|
if [[ ! -d "$WORKON_HOME" ]]; then
|
|
mkdir -p "$WORKON_HOME"
|
|
fi
|
|
export VIRTUALENVWRAPPER_PYTHON=/usr/bin/python3
|
|
|
|
source $(which virtualenvwrapper.sh)
|
|
|
|
if ! workon "$CTF_TOOLS_VE" 2>&1 1>/dev/null; then
|
|
echo "#### Creating python#PY_VERSION virtualenv '$CTF_TOOLS_VE' ####" >&2
|
|
# for some reason mkvirtualenv returns non-zero even if it
|
|
# succeeds?
|
|
mkvirtualenv -p $(which "python$PY_VERSION") "$CTF_TOOLS_VE" || true
|
|
echo "#### Switching to '$CTF_TOOLS_VE' virtualenv" >&2
|
|
workon "$CTF_TOOLS_VE" || (deactivate && workon "$CTF_TOOLS_VE")
|
|
fi
|
|
|
|
exec pip "$@"
|