1 | #!/usr/bin/env python2.5 |
---|
2 | |
---|
3 | """ |
---|
4 | Collates the results of listvms from multiple VM servers. Part of the xvm |
---|
5 | suite. |
---|
6 | """ |
---|
7 | |
---|
8 | from itertools import chain |
---|
9 | from subprocess import CalledProcessError, PIPE, Popen |
---|
10 | from sys import argv, stdout |
---|
11 | from cPickle import dump, loads |
---|
12 | |
---|
13 | ### |
---|
14 | |
---|
15 | #import compiler |
---|
16 | # |
---|
17 | #class Unsafe_Source_Error(Exception): |
---|
18 | # def __init__(self,error,descr = None,node = None): |
---|
19 | # self.error = error |
---|
20 | # self.descr = descr |
---|
21 | # self.node = node |
---|
22 | # self.lineno = getattr(node,"lineno",None) |
---|
23 | # |
---|
24 | # def __repr__(self): |
---|
25 | # return "Line %d. %s: %s" % (self.lineno, self.error, self.descr) |
---|
26 | # __str__ = __repr__ |
---|
27 | # |
---|
28 | #class SafeEval(object): |
---|
29 | # |
---|
30 | # def visit(self, node,**kw): |
---|
31 | # cls = node.__class__ |
---|
32 | # meth = getattr(self,'visit'+cls.__name__,self.default) |
---|
33 | # return meth(node, **kw) |
---|
34 | # |
---|
35 | # def default(self, node, **kw): |
---|
36 | # for child in node.getChildNodes(): |
---|
37 | # return self.visit(child, **kw) |
---|
38 | # |
---|
39 | # visitExpression = default |
---|
40 | # |
---|
41 | # def visitConst(self, node, **kw): |
---|
42 | # return node.value |
---|
43 | # |
---|
44 | # def visitDict(self,node,**kw): |
---|
45 | # return dict([(self.visit(k),self.visit(v)) for k,v in node.items]) |
---|
46 | # |
---|
47 | # def visitTuple(self,node, **kw): |
---|
48 | # return tuple(self.visit(i) for i in node.nodes) |
---|
49 | # |
---|
50 | # def visitList(self,node, **kw): |
---|
51 | # return [self.visit(i) for i in node.nodes] |
---|
52 | # |
---|
53 | #class SafeEvalWithErrors(SafeEval): |
---|
54 | # |
---|
55 | # def default(self, node, **kw): |
---|
56 | # raise Unsafe_Source_Error("Unsupported source construct", |
---|
57 | # node.__class__,node) |
---|
58 | # |
---|
59 | # def visitName(self,node, **kw): |
---|
60 | # if node.name == 'None': return None |
---|
61 | # raise Unsafe_Source_Error("Strings must be quoted", |
---|
62 | # node.name, node) |
---|
63 | # |
---|
64 | # # Add more specific errors if desired |
---|
65 | # |
---|
66 | #def safe_eval(source, fail_on_error = True): |
---|
67 | # if source.strip() == '': return None |
---|
68 | # walker = fail_on_error and SafeEvalWithErrors() or SafeEval() |
---|
69 | # try: |
---|
70 | # ast = compiler.parse(source,"eval") |
---|
71 | # except SyntaxError, err: |
---|
72 | # raise |
---|
73 | # try: |
---|
74 | # return walker.visit(ast) |
---|
75 | # except Unsafe_Source_Error, err: |
---|
76 | # raise |
---|
77 | |
---|
78 | ### |
---|
79 | |
---|
80 | def run(cmd): |
---|
81 | """ |
---|
82 | Run the given command (a list of program and argument strings) and return the |
---|
83 | stdout as a string, raising a CalledProcessError if the program exited with a |
---|
84 | non-zero status. |
---|
85 | """ |
---|
86 | p = Popen(cmd, stdout=PIPE) |
---|
87 | stdout = p.communicate()[0] |
---|
88 | if p.returncode != 0: raise CalledProcessError(p.returncode, cmd) |
---|
89 | return stdout |
---|
90 | |
---|
91 | def main(argv): |
---|
92 | # Query each of the server for their VMs. |
---|
93 | # run('kinit -k host/sipb-vm-58.mit.edu'.split()) |
---|
94 | # TODO get `servers` from a real list of all the VM hosts (instead of |
---|
95 | # hardcoding the list here) |
---|
96 | servers = [ 'black-mesa.mit.edu', 'sx-blade-2.mit.edu' ] |
---|
97 | # XXX |
---|
98 | results = [ loads(run(['remctl', server, 'remote', 'web', 'listvms', '--pickle'])) |
---|
99 | for server in servers ] |
---|
100 | results = filter( lambda x: x is not None, results ) |
---|
101 | |
---|
102 | # Merge the results and print. |
---|
103 | merged = {} |
---|
104 | for result in results: merged.update(result) |
---|
105 | if argv[1:] == ['--pickle']: |
---|
106 | dump(merged, stdout) |
---|
107 | else: |
---|
108 | print merged |
---|
109 | print '.' |
---|
110 | |
---|
111 | if __name__ == '__main__': |
---|
112 | main(argv) |
---|
113 | |
---|
114 | # vim:et:sw=2:ts=2 |
---|