annotate INSTALL.py @ 1:92a402736bcc

[reading material] mxnet
author Jeff Hammel <k0scist@gmail.com>
date Sun, 20 Aug 2017 11:14:59 -0700
parents a70bd2cd508b
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
1 #!/usr/bin/env python
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
2
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
3 """
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
4 installation script for tvii
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
5 TV2
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
6 """
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
7
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
8 import os
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
9 import sys
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
10 import urllib2
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
11 import subprocess
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
12 try:
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
13 from subprocess import check_call as call
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
14 except:
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
15 from subprocess import call
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
16
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
17 REPO='http://k0s.org/hg/tvii'
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
18 DEST='tvii' # name of the virtualenv
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
19 VIRTUALENV='https://raw.github.com/pypa/virtualenv/develop/virtualenv.py'
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
20
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
21 def which(binary, path=os.environ['PATH']):
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
22 dirs = path.split(os.pathsep)
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
23 for dir in dirs:
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
24 if os.path.isfile(os.path.join(dir, fileName)):
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
25 return os.path.join(dir, fileName)
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
26 if os.path.isfile(os.path.join(dir, fileName + ".exe")):
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
27 return os.path.join(dir, fileName + ".exe")
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
28
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
29 def main(args=sys.argv[1:]):
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
30
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
31 # create a virtualenv
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
32 virtualenv = which('virtualenv') or which('virtualenv.py')
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
33 if virtualenv:
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
34 call([virtualenv, DEST])
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
35 else:
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
36 process = subproces.Popen([sys.executable, '-', DEST], stdin=subprocess.PIPE)
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
37 process.communicate(stdin=urllib2.urlopen(VIRTUALENV).read())
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
38
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
39 # create a src directory
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
40 src = os.path.join(DEST, 'src')
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
41 os.mkdir(src)
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
42
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
43 # clone the repository
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
44 call(['hg', 'clone', REPO], cwd=src)
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
45
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
46 # find the virtualenv python
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
47 python = None
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
48 for path in (('bin', 'python'), ('Scripts', 'python.exe')):
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
49 _python = os.path.join(DEST, *path)
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
50 if os.path.exists(_python)
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
51 python = _python
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
52 break
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
53 else:
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
54 raise Exception("Python binary not found in %s" % DEST)
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
55
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
56 # find the clone
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
57 filename = REPO.rstrip('/')
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
58 filename = filename.split('/')[-1]
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
59 clone = os.path.join(src, filename)
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
60 assert os.path.exists(clone), "Clone directory not found in %s" % src
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
61
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
62 # ensure setup.py exists
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
63 assert os.path.exists(os.path.join(clone, 'setup.py')), 'setup.py not found in %s' % clone
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
64
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
65 # install the package in develop mode
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
66 call([python 'setup.py', 'develop'], cwd=clone)
a70bd2cd508b initial commit
Jeff Hammel <k0scist@gmail.com>
parents:
diff changeset
67