Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
pip.plugin.zsh 2.37 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
12
13
14
15
16
17
18
19 
# 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".

ZSH_PIP_CACHE_FILE=~/.pip/zsh-cache
ZSH_PIP_INDEXES=(https://pypi.python.org/simple/)

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
20 
zsh-pip-clean-packages() {
aph's avatar
fix pip plugin    
aph committed
21 
    sed -n '/<a href/ s/.*>\([^<]\{1,\}\).*/\1/p'
Brandon W Maister's avatar
Make the pip cache work with djangopypi2    
Brandon W Maister committed
22
23 
}
Brandon W Maister's avatar
pip: successfully cache all the packages    
Brandon W Maister committed
24 
zsh-pip-cache-packages() {
aph's avatar
fix pip plugin    
aph committed
25
26 
  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
27
28
29
30
31
32
33
34 
  fi

  if [[ ! -f $ZSH_PIP_CACHE_FILE ]]; then
      echo -n "(...caching package index...)"
      tmp_cache=/tmp/zsh_tmp_cache
      for index in $ZSH_PIP_INDEXES ; do
          # well... I've already got two problems
          curl $index 2>/dev/null | \
Brandon W Maister's avatar
Make the pip cache work with djangopypi2    
Brandon W Maister committed
35 
              zsh-pip-clean-packages \
Brandon W Maister's avatar
pip: successfully cache all the packages    
Brandon W Maister committed
36
37
38
39
40
41 
               >> $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
42
43
44
45
46
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 

# 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
79
80
81 

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