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
|
import argparse
import os
import shutil
import sys
from molotov import __version__
_DEFAULTS = {"target_dir": "."}
_PREFIX = "> "
_HERE = os.path.dirname(__file__)
class ValidationError(Exception):
pass
def _input(msg):
return input(msg) # pragma: no cover
def _prompt(text, default, validator=None):
while True:
try:
res = _input(_PREFIX + "{} [{}]: ".format(text, default))
if not res and default:
res = default
if validator:
res = validator(res)
return res
except ValidationError as e:
print(e)
def _yes(x):
if x.upper() not in ("Y", "YES", "N", "NO"):
raise ValidationError("Please enter either 'y' or 'n'.")
return x.upper() in ("Y", "YES")
def _parser():
parser = argparse.ArgumentParser(description="Quickstart")
parser.add_argument(
"--version",
action="store_true",
default=False,
help="Displays version and exits.",
)
return parser
def _copy_file(name, target_dir):
print("…copying {!r} in {!r}".format(name, target_dir))
target = os.path.join(target_dir, name)
if os.path.exists(target):
print("%r already exists. Cowardly stopping here" % target)
sys.exit(1)
shutil.copyfile(os.path.join(_HERE, name), target)
def main():
parser = _parser()
args = parser.parse_args()
if args.version:
print(__version__)
sys.exit(0)
# XXX
print("**** Molotov Quickstart ****")
print("")
print("Answer to a few questions to get started...")
target_dir = _prompt("Target directory", ".")
create_makefile = _prompt("Create Makefile", "y", validator=_yes)
print("Generating Molotov test...")
if not os.path.exists(target_dir):
os.makedirs(target_dir)
if create_makefile:
_copy_file("Makefile", target_dir)
_copy_file("loadtest.py", target_dir)
_copy_file("molotov.json", target_dir)
print("")
print("All done. Happy Breaking!")
print("Go in %r" % target_dir)
if create_makefile:
print("Run 'make build' to get started...")
|
