1 | #!/usr/bin/python |
---|
2 | |
---|
3 | """ |
---|
4 | Collates the results of listvms from multiple VM servers. Part of the xvm |
---|
5 | suite. |
---|
6 | """ |
---|
7 | |
---|
8 | from subprocess import PIPE, Popen |
---|
9 | try: |
---|
10 | from subprocess import CalledProcessError |
---|
11 | except ImportError: |
---|
12 | # Python 2.4 doesn't implement CalledProcessError |
---|
13 | class CalledProcessError(Exception): |
---|
14 | """This exception is raised when a process run by check_call() returns |
---|
15 | a non-zero exit status. The exit status will be stored in the |
---|
16 | returncode attribute.""" |
---|
17 | def __init__(self, returncode, cmd): |
---|
18 | self.returncode = returncode |
---|
19 | self.cmd = cmd |
---|
20 | def __str__(self): |
---|
21 | return "Command '%s' returned non-zero exit status %d" % (self.cmd, self.returncode) |
---|
22 | import sys |
---|
23 | import yaml |
---|
24 | |
---|
25 | ### |
---|
26 | |
---|
27 | def main(argv): |
---|
28 | # Query each of the server for their VMs. |
---|
29 | # TODO get `servers` from a real list of all the VM hosts (instead of |
---|
30 | # hardcoding the list here) |
---|
31 | servers = ['black-mesa.mit.edu', 'sx-blade-2.mit.edu'] |
---|
32 | # XXX |
---|
33 | pipes = [Popen(['remctl', server, 'remote', 'web', 'listvms'], stdout=PIPE) |
---|
34 | for server in servers] |
---|
35 | outputs = [p.communicate()[0] for p in pipes] |
---|
36 | for p in pipes: |
---|
37 | if p.returncode != 0: |
---|
38 | raise CalledProcessError(p.returncode, 'remctl') |
---|
39 | results = [yaml.load(o, yaml.CSafeLoader) for o in outputs] |
---|
40 | results = filter(lambda x: x is not None, results) |
---|
41 | |
---|
42 | # Merge the results and print. |
---|
43 | merged = {} |
---|
44 | for result in results: |
---|
45 | merged.update(result) |
---|
46 | |
---|
47 | print yaml.dump(merged, Dumper=yaml.CSafeDumper, default_flow_style=False) |
---|
48 | |
---|
49 | if __name__ == '__main__': |
---|
50 | main(sys.argv) |
---|
51 | |
---|
52 | # vim:et:sw=2:ts=4 |
---|