summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--functions/pip/_pip46
-rw-r--r--plugins/pip.plugin.zsh4
2 files changed, 50 insertions, 0 deletions
diff --git a/functions/pip/_pip b/functions/pip/_pip
new file mode 100644
index 000000000..b58010173
--- /dev/null
+++ b/functions/pip/_pip
@@ -0,0 +1,46 @@
+#compdef pip
+#autoload
+
+# pip zsh completion, based on homebrew completion
+
+_pip_installed() {
+ installed_pkgs=(`pip freeze`)
+}
+
+local -a _1st_arguments
+_1st_arguments=(
+ 'bundle:Create pybundles (archives containing multiple packages)'
+ 'freeze:Output all currently installed packages (exact versions) to stdout'
+ 'help:Show available commands'
+ 'install:Install packages'
+ 'search:Search PyPI'
+ 'uninstall:Uninstall packages'
+ 'unzip:Unzip individual packages'
+ 'zip:Zip individual packages'
+)
+
+local expl
+local -a pkgs installed_pkgs
+
+_arguments \
+ '(--version)--version[Show version number of program and exit]' \
+ '(-v --verbose)'{-v,--verbose}'[Give more output]' \
+ '(-q --quiet)'{-q,--quiet}'[Give less output]' \
+ '(-h --help)'{-h,--help}'[Show help]' \
+ '*:: :->subcmds' && return 0
+
+if (( CURRENT == 1 )); then
+ _describe -t commands "pip subcommand" _1st_arguments
+ return
+fi
+
+case "$words[1]" in
+ list)
+ if [[ "$state" == forms ]]; then
+ _pip_installed
+ _requested installed_pkgs expl 'installed packages' compadd -a installed_pkgs
+ fi ;;
+ uninstall)
+ _pip_installed
+ _wanted installed_pkgs expl 'installed packages' compadd -a installed_pkgs ;;
+esac
diff --git a/plugins/pip.plugin.zsh b/plugins/pip.plugin.zsh
new file mode 100644
index 000000000..69e21f8d7
--- /dev/null
+++ b/plugins/pip.plugin.zsh
@@ -0,0 +1,4 @@
+# add brew completion function to path
+fpath=($ZSH/functions/pip $fpath)
+autoload -U compinit
+compinit -i