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
|
#!/usr/bin/python
import os
import sys
from optparse import OptionParser
from checkbox.lib.fifo import FifoReader, FifoWriter
from checkbox.job import Job, FAIL
def main(args):
usage = "Usage: %prog INPUT OUTPUT"
parser = OptionParser(usage=usage)
parser.add_option("-p", "--path",
help="PATH variable to replace in the environment")
(options, args) = parser.parse_args(args)
if len(args) < 2:
parser.error("Missing INPUT and OUTPUT")
if options.path:
os.environ["PATH"] = options.path
reader = FifoReader(args[0])
writer = FifoWriter(args[1])
while True:
try:
message = reader.read_object()
if message == "stop":
break
if message == "ping":
writer.write_object("pong")
continue
if isinstance(message, dict) and "command" in message:
job = Job(message["command"], message.get("environ"),
message.get("timeout"))
result = job.execute()
else:
result = (FAIL, "", 0,)
writer.write_object(result)
except IOError, e:
break
return 0
if __name__ == "__main__":
sys.exit(main(sys.argv[1:]))
|