from itertools import chain
from subprocess import CalledProcessError, PIPE, Popen
-from sys import argv
-
-###
-
-import compiler
-
-class Unsafe_Source_Error(Exception):
- def __init__(self,error,descr = None,node = None):
- self.error = error
- self.descr = descr
- self.node = node
- self.lineno = getattr(node,"lineno",None)
-
- def __repr__(self):
- return "Line %d. %s: %s" % (self.lineno, self.error, self.descr)
- __str__ = __repr__
-
-class SafeEval(object):
-
- def visit(self, node,**kw):
- cls = node.__class__
- meth = getattr(self,'visit'+cls.__name__,self.default)
- return meth(node, **kw)
-
- def default(self, node, **kw):
- for child in node.getChildNodes():
- return self.visit(child, **kw)
-
- visitExpression = default
-
- def visitConst(self, node, **kw):
- return node.value
-
- def visitDict(self,node,**kw):
- return dict([(self.visit(k),self.visit(v)) for k,v in node.items])
-
- def visitTuple(self,node, **kw):
- return tuple(self.visit(i) for i in node.nodes)
-
- def visitList(self,node, **kw):
- return [self.visit(i) for i in node.nodes]
-
-class SafeEvalWithErrors(SafeEval):
-
- def default(self, node, **kw):
- raise Unsafe_Source_Error("Unsupported source construct",
- node.__class__,node)
-
- def visitName(self,node, **kw):
- if node.name == 'None': return None
- raise Unsafe_Source_Error("Strings must be quoted",
- node.name, node)
-
- # Add more specific errors if desired
-
-def safe_eval(source, fail_on_error = True):
- if source.strip() == '': return None
- walker = fail_on_error and SafeEvalWithErrors() or SafeEval()
- try:
- ast = compiler.parse(source,"eval")
- except SyntaxError, err:
- raise
- try:
- return walker.visit(ast)
- except Unsafe_Source_Error, err:
- raise
+from sys import argv, stdout
+from yaml import safe_dump, safe_load
###
def main(argv):
# Query each of the server for their VMs.
- # run('kinit -k host/sipb-vm-58.mit.edu'.split())
# TODO get `servers` from a real list of all the VM hosts (instead of
# hardcoding the list here)
servers = [ 'black-mesa.mit.edu', 'sx-blade-2.mit.edu' ]
# XXX
- results = [ safe_eval(run(['remctl', server, 'remote', 'web', 'listvms'] + argv[1:]))
+ results = [ safe_load(run(['remctl', server, 'remote', 'web', 'listvms']))
for server in servers ]
results = filter( lambda x: x is not None, results )
# Merge the results and print.
merged = {}
for result in results: merged.update(result)
- print merged
- print '.'
+ print safe_dump(merged, default_flow_style=False)
if __name__ == '__main__':
main(argv)