Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
pip.plugin.zsh 2.52 KB
Newer Older
Brandon W Maister's avatar
pip: successfully cache all the packages    
Brandon W Maister committed
1
2
3
4
5
6
7
8
9
10
11 
# 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".
Goose's avatar
pip: move plugin cache to XDG folder (#9299)    
Goose committed
12
13
14
15
16 
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
Fabricio Biazzotto's avatar
pip: fix indexes URL (#6986)    
Fabricio Biazzotto committed
17 
ZSH_PIP_INDEXES=(https://pypi.org/simple/)
Brandon W Maister's avatar
pip: successfully cache all the packages    
Brandon W Maister committed
18
19
20
21
22
23 

zsh-pip-clear-cache() {
  rm $ZSH_PIP_CACHE_FILE
  unset piplist
}
Brandon W Maister's avatar
Make the pip cache work with djangopypi2    
Brandon W Maister committed
24 
zsh-pip-clean-packages() {
aph's avatar
fix pip plugin    
aph committed
25 
    sed -n '/<a href/ s/.*>\([^<]\{1,\}\).*/\1/p'
Brandon W Maister's avatar
Make the pip cache work with djangopypi2    
Brandon W Maister committed
26
27 
}
Brandon W Maister's avatar
pip: successfully cache all the packages    
Brandon W Maister committed
28 
zsh-pip-cache-packages() {
aph's avatar
fix pip plugin    
aph committed
29
30 
  if [[ ! -d ${ZSH_PIP_CACHE_FILE:h} ]]; then
      mkdir -p ${ZSH_PIP_CACHE_FILE:h}
Brandon W Maister's avatar
pip: successfully cache all the packages    
Brandon W Maister committed
31
32
33
34
35 
  fi

  if [[ ! -f $ZSH_PIP_CACHE_FILE ]]; then
      echo -n "(...caching package index...)"
      tmp_cache=/tmp/zsh_tmp_cache
Ryan Greenblatt's avatar
pip: fix no such file or directory error (#7916)    
Ryan Greenblatt committed
36 
      touch $tmp_cache
Brandon W Maister's avatar
pip: successfully cache all the packages    
Brandon W Maister committed
37
38 
      for index in $ZSH_PIP_INDEXES ; do
          # well... I've already got two problems
Fabricio Biazzotto's avatar
pip: fix indexes URL (#6986)    
Fabricio Biazzotto committed
39 
          curl -L $index 2>/dev/null | \
Brandon W Maister's avatar
Make the pip cache work with djangopypi2    
Brandon W Maister committed
40 
              zsh-pip-clean-packages \
Brandon W Maister's avatar
pip: successfully cache all the packages    
Brandon W Maister committed
41
42
43
44
45
46 
               >> $tmp_cache
      done
      sort $tmp_cache | uniq | tr '\n' ' ' > $ZSH_PIP_CACHE_FILE
      rm $tmp_cache
  fi
}
Brandon W Maister's avatar
Make the pip cache work with djangopypi2    
Brandon W Maister committed
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83 

# 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
}
Kevin Ndung'u's avatar
Disable zsh file globbing when running pip (#5348)    
Kevin Ndung'u committed
84
85
86 

alias pip="noglob pip" # allows square brackets for pip command invocation