summaryrefslogtreecommitdiff
path: root/oh-my-zsh/plugins/pip
diff options
context:
space:
mode:
Diffstat (limited to 'oh-my-zsh/plugins/pip')
-rw-r--r--oh-my-zsh/plugins/pip/README.md36
-rw-r--r--oh-my-zsh/plugins/pip/_pip100
-rw-r--r--oh-my-zsh/plugins/pip/pip.plugin.zsh135
3 files changed, 271 insertions, 0 deletions
diff --git a/oh-my-zsh/plugins/pip/README.md b/oh-my-zsh/plugins/pip/README.md
new file mode 100644
index 0000000..70d40c7
--- /dev/null
+++ b/oh-my-zsh/plugins/pip/README.md
@@ -0,0 +1,36 @@
+# pip plugin
+
+This plugin adds completion for [pip](https://pip.pypa.io/en/latest/),
+the Python package manager.
+
+To use it, add `pip` to the plugins array in your zshrc file:
+
+```zsh
+plugins=(... pip)
+```
+
+## pip cache
+
+The pip plugin caches the names of available pip packages from the PyPI index.
+To trigger the caching process, try to complete `pip install`,
+or you can run `zsh-pip-cache-packages` directly.
+
+To reset the cache, run `zsh-pip-clear-cache` and it will be rebuilt next
+the next time you autocomplete `pip install`.
+
+## Aliases
+
+| Alias | Description |
+| :------- | :-------------------------------------------- |
+| pipi | Install packages |
+| pipig | Install package from GitHub repository |
+| pipigb | Install package from GitHub branch |
+| pipigp | Install package from GitHub pull request |
+| pipu | Upgrade packages |
+| pipun | Uninstall packages |
+| pipgi | Grep through installed packages |
+| piplo | List outdated packages |
+| pipreq | Create requirements file |
+| pipir | Install packages from `requirements.txt` file |
+| pipupall | Update all installed packages |
+| pipunall | Uninstall all installed packages |
diff --git a/oh-my-zsh/plugins/pip/_pip b/oh-my-zsh/plugins/pip/_pip
new file mode 100644
index 0000000..e03be6a
--- /dev/null
+++ b/oh-my-zsh/plugins/pip/_pip
@@ -0,0 +1,100 @@
+#compdef pip pip2 pip-2.7 pip3 pip-3.2 pip-3.3 pip-3.4
+#autoload
+
+# pip zsh completion, based on last stable release (pip8)
+# homebrew completion and backwards compatibility
+
+_pip_all() {
+ # we cache the list of packages (originally from the macports plugin)
+ if (( ! $+piplist )); then
+ zsh-pip-cache-packages
+ piplist=($(cat $ZSH_PIP_CACHE_FILE))
+ fi
+}
+
+_pip_installed() {
+ installed_pkgs=($($service freeze | cut -d '=' -f 1))
+}
+
+local -a _1st_arguments
+_1st_arguments=(
+ 'install:install packages'
+ 'download:download packages'
+ 'uninstall:uninstall packages'
+ 'freeze:output all currently installed packages (exact versions) to stdout'
+ 'list:list installed packages'
+ 'show:show information about installed packages'
+ 'search:search PyPI'
+ 'wheel:build individual wheel archives for your requirements and dependencies'
+ 'hash:compute a hash of a local package archive'
+ 'help:show available commands'
+ 'bundle:create pybundles (archives containing multiple packages)(deprecated)'
+ 'unzip:unzip individual packages(deprecated)'
+ 'zip:zip individual packages(deprecated)'
+)
+
+local expl
+local -a all_pkgs installed_pkgs
+
+_arguments \
+ '(-h --help)'{-h,--help}'[show help]' \
+ '(--isolated)--isolated[run pip in isolated mode, ignores environment variables and user configuration]' \
+ '(-v --verbose)'{-v,--verbose}'[give more output]' \
+ '(-V --version)'{-V,--version}'[show version number of program and exit]' \
+ '(-q --quiet)'{-q,--quiet}'[give less output]' \
+ '(--log)--log[log file location]' \
+ '(--proxy)--proxy[proxy in form user:passwd@proxy.server:port]' \
+ '(--retries)--retries[max number of retries per connection (default 5 times)]' \
+ '(--timeout)--timeout[socket timeout (default 15s)]' \
+ '(--exists-action)--exists-action[default action when a path already exists: (s)witch, (i)gnore, (w)ipe, (b)ackup]' \
+ '(--trusted-host)--trusted-host[mark this host as trusted]' \
+ '(--cert)--cert[path to alternate CA bundle]' \
+ '(--client-cert)--client-cert[path to SSL client certificate]' \
+ '(--cache-dir)--cache-dir[store the cache data in specified directory]' \
+ '(--no-cache-dir)--no-cache-dir[disable de cache]' \
+ '(--disable-pip-version-check)--disable-pip-version-check[do not check periodically for new pip version downloads]' \
+ '(-E --environment)'{-E,--environment}'[virtualenv environment to run pip in (deprecated)]' \
+ '(-s --enable-site-packages)'{-s,--enable-site-packages}'[include site-packages in virtualenv (deprecated)]' \
+ '*:: :->subcmds' && return 0
+
+if (( CURRENT == 1 )); then
+ _describe -t commands "pip subcommand" _1st_arguments
+ return
+fi
+
+case "$words[1]" in
+ search)
+ _arguments \
+ '(--index)--index[base URL of Python Package Index]' ;;
+ freeze)
+ _arguments \
+ '(-l --local)'{-l,--local}'[report only virtualenv packages]' ;;
+ install)
+ _arguments \
+ '(-U --upgrade)'{-U,--upgrade}'[upgrade all packages to the newest available version]' \
+ '(--user)--user[install packages to user home]' \
+ '(-f --find-links)'{-f,--find-links}'[URL for finding packages]' \
+ '(-r --requirement)'{-r,--requirement}'[Requirements file for packages to install]:File:_files' \
+ '(--no-deps --no-dependencies)'{--no-deps,--no-dependencies}'[iIgnore package dependencies]' \
+ '(--no-install)--no-install[only download packages]' \
+ '(--no-download)--no-download[only install downloaded packages]' \
+ '(--install-option)--install-option[extra arguments to be supplied to the setup.py]' \
+ '(--single-version-externally-managed)--single-version-externally-managed[do not download/install dependencies. requires --record or --root]'\
+ '(--root)--root[treat this path as a fake chroot, installing into it. implies --single-version-externally-managed]'\
+ '(--record)--record[file to record all installed files to.]'\
+ '(-r --requirement)'{-r,--requirement}'[requirements file]: :_files'\
+ '(-e --editable)'{-e,--editable}'[path of or url to source to link to instead of installing.]: :_files -/'\
+ '1: :->packages' && return 0
+
+ if [[ "$state" == packages ]]; then
+ _pip_all
+ _wanted piplist expl 'packages' compadd -a piplist
+ _files -g "*.(tar.gz|whl)"
+ fi ;;
+ uninstall)
+ _pip_installed
+ _wanted installed_pkgs expl 'installed packages' compadd -a installed_pkgs ;;
+ show)
+ _pip_installed
+ _wanted installed_pkgs expl 'installed packages' compadd -a installed_pkgs ;;
+esac
diff --git a/oh-my-zsh/plugins/pip/pip.plugin.zsh b/oh-my-zsh/plugins/pip/pip.plugin.zsh
new file mode 100644
index 0000000..bf1aafd
--- /dev/null
+++ b/oh-my-zsh/plugins/pip/pip.plugin.zsh
@@ -0,0 +1,135 @@
+# Usage:
+# Just add pip to your installed plugins.
+
+# If you would like to change the cheeseshops used for autocomplete set
+# ZSH_PIP_INDEXES in your zshrc. If one of your indexes are bogus you won't get
+# any kind of error message, pip will just not autocomplete from them. Double
+# check!
+#
+# If you would like to clear your cache, go ahead and do a
+# "zsh-pip-clear-cache".
+
+if [[ -d "${XDG_CACHE_HOME:-$HOME/.cache}/pip" ]]; then
+ ZSH_PIP_CACHE_FILE="${XDG_CACHE_HOME:-$HOME/.cache}/pip/zsh-cache"
+else
+ ZSH_PIP_CACHE_FILE=~/.pip/zsh-cache
+fi
+ZSH_PIP_INDEXES=(https://pypi.org/simple/)
+
+zsh-pip-clear-cache() {
+ rm $ZSH_PIP_CACHE_FILE
+ unset piplist
+}
+
+zsh-pip-clean-packages() {
+ sed -n '/<a href/ s/.*>\([^<]\{1,\}\).*/\1/p'
+}
+
+zsh-pip-cache-packages() {
+ if [[ ! -d ${ZSH_PIP_CACHE_FILE:h} ]]; then
+ mkdir -p ${ZSH_PIP_CACHE_FILE:h}
+ fi
+
+ if [[ ! -f $ZSH_PIP_CACHE_FILE ]]; then
+ echo -n "(...caching package index...)"
+ tmp_cache=/tmp/zsh_tmp_cache
+ touch $tmp_cache
+ for index in $ZSH_PIP_INDEXES ; do
+ # well... I've already got two problems
+ curl -L $index 2>/dev/null | \
+ zsh-pip-clean-packages \
+ >> $tmp_cache
+ done
+ sort $tmp_cache | uniq | tr '\n' ' ' > $ZSH_PIP_CACHE_FILE
+ rm $tmp_cache
+ fi
+}
+
+# A test function that validates the regex against known forms of the simple
+# index. If you modify the regex to make it work for you, you should add a test
+# case in here and make sure that your changes don't break things for someone
+# else.
+zsh-pip-test-clean-packages() {
+ local expected
+ local actual
+ expected="0x10c-asm
+1009558_nester"
+
+ actual=$(echo -n "<html><head><title>Simple Index</title><meta name=\"api-version\" value=\"2\" /></head><body>
+<a href='0x10c-asm'>0x10c-asm</a><br/>
+<a href='1009558_nester'>1009558_nester</a><br/>
+</body></html>" | zsh-pip-clean-packages)
+
+ if [[ $actual != $expected ]] ; then
+ echo -e "python's simple index is broken:\n$actual\n !=\n$expected"
+ else
+ echo "python's simple index is fine"
+ fi
+
+ actual=$(echo -n '<html>
+ <head>
+ <title>Simple Package Index</title>
+ </head>
+ <body>
+ <a href="0x10c-asm">0x10c-asm</a><br/>
+ <a href="1009558_nester">1009558_nester</a><br/>
+</body></html>' | zsh-pip-clean-packages)
+
+ if [[ $actual != $expected ]] ; then
+ echo -e "the djangopypi2 index is broken:\n$actual\n !=\n$expected"
+ else
+ echo "the djangopypi2 index is fine"
+ fi
+}
+
+if (( $+commands[pip3] && !$+commands[pip] )); then
+ alias pip="noglob pip3"
+else
+ alias pip="noglob pip"
+fi
+
+alias pipi="pip install"
+alias pipu="pip install --upgrade"
+alias pipun="pip uninstall"
+alias pipgi="pip freeze | grep"
+alias piplo="pip list -o"
+
+# Create requirements file
+alias pipreq="pip freeze > requirements.txt"
+
+# Install packages from requirements file
+alias pipir="pip install -r requirements.txt"
+
+# Upgrade all installed packages
+function pipupall {
+ # non-GNU xargs does not support nor need `--no-run-if-empty`
+ local xargs="xargs --no-run-if-empty"
+ xargs --version 2>/dev/null | grep -q GNU || xargs="xargs"
+ pip list --outdated | awk 'NR > 2 { print $1 }' | ${=xargs} pip install --upgrade
+}
+
+# Uninstall all installed packages
+function pipunall {
+ # non-GNU xargs does not support nor need `--no-run-if-empty`
+ local xargs="xargs --no-run-if-empty"
+ xargs --version 2>/dev/null | grep -q GNU || xargs="xargs"
+ pip list --format freeze | cut -d= -f1 | ${=xargs} pip uninstall
+}
+
+# Install from GitHub repository
+function pipig {
+ pip install "git+https://github.com/$1.git"
+}
+compdef _pip pipig
+
+# Install from GitHub branch
+function pipigb {
+ pip install "git+https://github.com/$1.git@$2"
+}
+compdef _pip pipigb
+
+# Install from GitHub pull request
+function pipigp {
+ pip install "git+https://github.com/$1.git@refs/pull/$2/head"
+}
+compdef _pip pipigp