90 lines
2.8 KiB
Bash
90 lines
2.8 KiB
Bash
|
# vim:sts=2:sw=2:ft=zsh
|
||
|
|
# Taken from: https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/plugins/python/python.plugin.zsh
|
||
|
|
#
|
||
|
|
# python command
|
||
|
|
alias py='python'
|
||
|
|
|
||
|
|
# Find python file
|
||
|
|
alias pyfind='find . -name "*.py"'
|
||
|
|
|
||
|
|
# Remove python compiled byte-code and mypy/pytest cache in either the current
|
||
|
|
# directory or in a list of specified directories (including sub directories).
|
||
|
|
function pyclean() {
|
||
|
|
ZSH_PYCLEAN_PLACES=${*:-'.'}
|
||
|
|
find ${ZSH_PYCLEAN_PLACES} -type f -name "*.py[co]" -delete
|
||
|
|
find ${ZSH_PYCLEAN_PLACES} -type d -name "__pycache__" -delete
|
||
|
|
find ${ZSH_PYCLEAN_PLACES} -depth -type d -name ".mypy_cache" -exec rm -r "{}" +
|
||
|
|
find ${ZSH_PYCLEAN_PLACES} -depth -type d -name ".pytest_cache" -exec rm -r "{}" +
|
||
|
|
}
|
||
|
|
|
||
|
|
# Add the user installed site-packages paths to PYTHONPATH, only if the
|
||
|
|
# directory exists. Also preserve the current PYTHONPATH value.
|
||
|
|
# Feel free to autorun this when .zshrc loads.
|
||
|
|
function pyuserpaths() {
|
||
|
|
local targets=("python2" "python3") # bins
|
||
|
|
|
||
|
|
# Get existing interpreters.
|
||
|
|
local interps=()
|
||
|
|
for target in $targets; do
|
||
|
|
[ `command -v $target` ] && interps+=($target)
|
||
|
|
done
|
||
|
|
|
||
|
|
# Check for a non-standard install directory.
|
||
|
|
local user_base="${HOME}/.local"
|
||
|
|
[ $PYTHONUSERBASE ] && user_base=$PYTHONUSERBASE
|
||
|
|
|
||
|
|
# Add version specific paths, if:
|
||
|
|
# it exists in the filesystem;
|
||
|
|
# it isn't in PYTHONPATH already.
|
||
|
|
for interp in $interps; do
|
||
|
|
# Get minor release version.
|
||
|
|
local ver=`$interp -V 2>&1`
|
||
|
|
ver=`echo ${ver:7} | cut -d '.' -f 1,2` # The patch version is variable length, truncate it.
|
||
|
|
|
||
|
|
local site_pkgs="${user_base}/lib/python${ver}/site-packages"
|
||
|
|
[[ -d $site_pkgs && ! $PYTHONPATH =~ $site_pkgs ]] && export PYTHONPATH=${site_pkgs}:$PYTHONPATH
|
||
|
|
done
|
||
|
|
}
|
||
|
|
|
||
|
|
# Grep among .py files
|
||
|
|
alias pygrep='grep -nr --include="*.py"'
|
||
|
|
|
||
|
|
# Run proper IPython regarding current virtualenv (if any)
|
||
|
|
alias ipython="python -c 'import IPython; IPython.terminal.ipapp.launch_new_instance()'"
|
||
|
|
|
||
|
|
# Share local directory as a HTTP server
|
||
|
|
alias pyserver="python -m http.server"
|
||
|
|
|
||
|
|
|
||
|
|
## venv utilities
|
||
|
|
|
||
|
|
# Activate a the python virtual environment specified.
|
||
|
|
# If none specified, use 'venv'.
|
||
|
|
function vrun() {
|
||
|
|
local name="${1:-venv}"
|
||
|
|
local venvpath="${name:P}"
|
||
|
|
|
||
|
|
if [[ ! -d "$venvpath" ]]; then
|
||
|
|
echo >&2 "Error: no such venv in current directory: $name"
|
||
|
|
return 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [[ ! -f "${venvpath}/bin/activate" ]]; then
|
||
|
|
echo >&2 "Error: '${name}' is not a proper virtual environment"
|
||
|
|
return 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
. "${venvpath}/bin/activate" || return $?
|
||
|
|
echo "Activated virtual environment ${name}"
|
||
|
|
}
|
||
|
|
|
||
|
|
# Create a new virtual environment, with default name 'venv'.
|
||
|
|
function mkv() {
|
||
|
|
local name="${1:-venv}"
|
||
|
|
local venvpath="${name:P}"
|
||
|
|
|
||
|
|
python3 -m venv "${name}" || return
|
||
|
|
echo >&2 "Created venv in '${venvpath}'"
|
||
|
|
vrun "${name}"
|
||
|
|
}
|