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
|
#!/usr/bin/python
usage="""\
This script installs spack in a new prefix.
To use: cd to the prefix and execute this script.
"""
import os
import sys
import getpass
from subprocess import check_call
def escape(s):
"""Returns a TTY escape code if stdout is a tty, otherwise empty string"""
if sys.stdout.isatty():
return "\033[{}m".format(s)
return ''
def bold(n):
return escape("1;{}".format(n))
blue = bold(34)
red = bold(31)
white = bold(39)
reset = escape(0)
def msg(msg):
print "{}==>{} {}{}".format(blue, white, str(msg), reset)
def error(msg):
print "{}==> ERROR: {} {}{}".format(red, white, str(msg), reset)
sys.exit(1)
user = getpass.getuser()
if not user:
error("Couldn't determine username!")
spack_repo = "https://%s@lc.llnl.gov/stash/scm/SCALE/spack.git" % user
msg("Fetching spack from %s" % spack_repo)
prefix = os.getcwd()
if os.path.exists(".git"):
error("There already seems to be a git repository here.")
if os.listdir("."):
error("There is already something in this directory.")
msg("This script will install:")
print " %s/bin/spack" % prefix
print " %s/lib/spack/..." % prefix
check_call(["git", "init", "--shared", "-q"])
check_call(["git", "remote", "add", "origin", spack_repo])
check_call(["git", "fetch", "origin", "master:refs/remotes/origin/master", "-n", "-q"])
check_call(["git", "reset", "--hard", "origin/master", "-q"])
msg("Successfully installed spack in %s" % prefix)
|
