Skip to content

Commit 1c4f2bb

Browse files
committed
Added a re-vendor script in Python
1 parent e8ca48a commit 1c4f2bb

File tree

1 file changed

+34
-0
lines changed

1 file changed

+34
-0
lines changed

pip/vendor/re-vendor.py

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
import os
2+
import sys
3+
import pip
4+
import glob
5+
import shutil
6+
7+
here = os.path.abspath(os.path.dirname(__file__))
8+
9+
def usage():
10+
print("Usage: re-vendor.py [clean|vendor]")
11+
sys.exit(1)
12+
13+
def clean():
14+
for fn in os.listdir(here):
15+
dirname = os.path.join(here, fn)
16+
if os.path.isdir(dirname):
17+
shutil.rmtree(dirname)
18+
# six is a single file, not a package
19+
os.unlink(os.path.join(here, 'six.py'))
20+
21+
def vendor():
22+
pip.main(['install', '-t', here, '-r', 'vendor.txt'])
23+
for dirname in glob.glob('*.egg-info'):
24+
shutil.rmtree(dirname)
25+
26+
if __name__ == '__main__':
27+
if len(sys.argv) != 2:
28+
usage()
29+
if sys.argv[1] == 'clean':
30+
clean()
31+
elif sys.argv[1] == 'vendor':
32+
vendor()
33+
else:
34+
usage()

0 commit comments

Comments
 (0)