Repository navigation
Expand file tree
/
Copy pathinstall-local.py
More file actions
103 lines (78 loc) · 2.53 KB
/
Copy pathinstall-local.py
File metadata and controls
103 lines (78 loc) · 2.53 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#!/usr/bin/env python
from __future__ import absolute_import
from __future__ import unicode_literals
import contextlib
import distutils.spawn
import io
import os.path
import shutil
import subprocess
import sys
import tarfile
if str is bytes:
from urllib import urlopen
else:
from urllib.request import urlopen
TGZ = (
'https://pypi.python.org/packages/d4/0c/'
'9840c08189e030873387a73b90ada981885010dd9aea134d6de30cd24cb8/'
'virtualenv-15.1.0.tar.gz'
)
PKG_PATH = '/tmp/.virtualenv-pkg'
def clean(dirname):
if os.path.exists(dirname):
shutil.rmtree(dirname)
@contextlib.contextmanager
def clean_path():
try:
yield
finally:
clean(PKG_PATH)
def virtualenv(path):
clean(PKG_PATH)
clean(path)
print('Downloading ' + TGZ)
tar_contents = io.BytesIO(urlopen(TGZ).read())
with contextlib.closing(tarfile.open(fileobj=tar_contents)) as tarfile_obj:
# Chop off the first path segment to avoid having the version in
# the path
for member in tarfile_obj.getmembers():
_, _, member.name = member.name.partition('/')
if member.name:
tarfile_obj.extract(member, PKG_PATH)
print('Done.')
with clean_path():
return subprocess.call((
sys.executable, os.path.join(PKG_PATH, 'virtualenv.py'), path,
))
def main():
venv_path = os.path.join(os.environ['HOME'], '.pre-commit-venv')
bin_dir = os.path.join(os.environ['HOME'], 'bin')
script_src = os.path.join(venv_path, 'bin', 'pre-commit')
script_dest = os.path.join(bin_dir, 'pre-commit')
if sys.argv[1:] == ['uninstall']:
clean(PKG_PATH)
clean(venv_path)
if os.path.lexists(script_dest):
os.remove(script_dest)
print('Cleaned ~/.pre-commit-venv ~/bin/pre-commit')
return 0
virtualenv(venv_path)
subprocess.check_call((
os.path.join(venv_path, 'bin', 'pip'), 'install', 'pre-commit',
))
print('*' * 79)
print('Installing pre-commit to {0}'.format(script_dest))
print('*' * 79)
if not os.path.exists(bin_dir):
os.mkdir(bin_dir)
# os.symlink is not idempotent
if os.path.exists(script_dest):
os.remove(script_dest)
os.symlink(script_src, script_dest)
if not distutils.spawn.find_executable('pre-commit'):
print('It looks like {0} is not on your path'.format(bin_dir))
print('You may want to add it.')
print('Often this does the trick: source ~/.profile')
if __name__ == '__main__':
exit(main())