linux/ohmyzsh
1
0
Fork 0
mirror of https://github.com/ohmyzsh/ohmyzsh.git synced 2024-11-26 21:51:05 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

style(python): fix code style and some hidden bugs

Browse source
This commit is contained in:
Marc Cornellà 2021-12-31 20:45:12 +01:00
parent 9c01dbc30b
commit 78c91ccbf9
No known key found for this signature in database
GPG key ID: 0314585E776A9C1B
1 changed files with 23 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
plugins/python/python.plugin.zsh
View file

@ -7,40 +7,37 @@ alias pyfind='find . -name "*.py"'
# Remove python compiled byte-code and mypy/pytest cache in either the current # Remove python compiled byte-code and mypy/pytest cache in either the current
# directory or in a list of specified directories (including sub directories). # directory or in a list of specified directories (including sub directories).
function pyclean() { function pyclean() {
ZSH_PYCLEAN_PLACES=${*:-'.'} find "${@:-.}" -type f -name "*.py[co]" -delete
find ${ZSH_PYCLEAN_PLACES} -type f -name "*.py[co]" -delete find "${@:-.}" -type d -name "__pycache__" -delete
find ${ZSH_PYCLEAN_PLACES} -type d -name "__pycache__" -delete find "${@:-.}" -depth -type d -name ".mypy_cache" -exec rm -r "{}" +
find ${ZSH_PYCLEAN_PLACES} -depth -type d -name ".mypy_cache" -exec rm -r "{}" + find "${@:-.}" -depth -type d -name ".pytest_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 # Add the user installed site-packages paths to PYTHONPATH, only if the
# directory exists. Also preserve the current PYTHONPATH value. # directory exists. Also preserve the current PYTHONPATH value.
# Feel free to autorun this when .zshrc loads. # Feel free to autorun this when .zshrc loads.
function pyuserpaths() { function pyuserpaths() {
local targets=("python2" "python3") # bins setopt localoptions extendedglob
# Get existing interpreters. # Check for a non-standard install directory.
local interps=() local user_base="${PYTHONUSERBASE:-"${HOME}/.local"}"
for target in $targets; do
[ `command -v $target` ] && interps+=($target)
done
# Check for a non-standard install directory. local python version site_pkgs
local user_base="${HOME}/.local" for python in python2 python3; do
[ $PYTHONUSERBASE ] && user_base=$PYTHONUSERBASE # Check if command exists
(( ${+commands[$python]} )) || continue
# Add version specific paths, if: # Get minor release version.
# it exists in the filesystem; # The patch version is variable length, truncate it.
# it isn't in PYTHONPATH already. version=${(M)${"$($python -V 2>&1)":7}#[^.]##.[^.]##}
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" # Add version specific path, if:
[[ -d $site_pkgs && ! $PYTHONPATH =~ $site_pkgs ]] && export PYTHONPATH=${site_pkgs}:$PYTHONPATH # - it exists in the filesystem
done # - it isn't in $PYTHONPATH already.
site_pkgs="${user_base}/lib/python${version}/site-packages"
[[ -d "$site_pkgs" && ! "$PYTHONPATH" =~ (^|:)"$site_pkgs"(:|$) ]] || continue
export PYTHONPATH="${site_pkgs}${PYTHONPATH+":${PYTHONPATH}"}"
done
} }
# Grep among .py files # Grep among .py files