files/usr/sbin/sipb-xen-remote-listvms

   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 yaml import safe_dump, safe_load
  12
  13 ###
  14
  15 def main(argv):
  16     # Query each of the server for their VMs.
  17     # TODO get `servers` from a real list of all the VM hosts (instead of
  18     # hardcoding the list here)
  19     servers = ['black-mesa.mit.edu', 'sx-blade-2.mit.edu']
  20     # XXX
  21     pipes = [Popen(['remctl', server, 'remote', 'web', 'listvms'], stdout=PIPE)
  22              for server in servers]
  23     outputs = [p.communicate()[0] for p in pipes]
  24     for p in pipes:
  25         if p.returncode != 0:
  26             raise CalledProcessError(p.returncode, cmd)
  27     results = [safe_load(o) for o in outputs]
  28     results = filter(lambda x: x is not None, results)
  29
  30     # Merge the results and print.
  31     merged = {}
  32     for result in results:
  33         merged.update(result)
  34     print safe_dump(merged, default_flow_style=False)
  35
  36 if __name__ == '__main__':
  37     main(argv)
  38
  39 # vim:et:sw=2:ts=4