host/usr/sbin/invirt-listvms

   1 #!/usr/bin/python
   2
   3 import sys
   4 import time
   5 from xen.lowlevel import xs
   6
   7 xsc = xs.xs()
   8
   9 def live_vms():
  10     domids = set(xsc.ls('', '/local/domain'))
  11     domids.discard('0')
  12
  13     vms = dict()
  14
  15     for domid in domids:
  16         try:
  17             name, data = get_dom(int(domid))
  18         except (xs.Error, TypeError):
  19             continue # went down since we started
  20         if name.startswith('d_'):
  21             name = name[2:]
  22             vms[name] = data
  23     return vms
  24
  25 def get_dom(domid):
  26     name = xsc.read('', '/local/domain/%d/name' % domid)
  27     data = dict()
  28     data['domid'] = domid
  29     # presence of a graphical console
  30     data['console'] = xsc.read('', '/local/domain/%d/device/vfb/0/state' % domid)
  31     # uptime
  32     data['vm'] = xsc.read('', '/local/domain/%d/vm' % domid)
  33     data['start_time'] = float(xsc.read('', '%s/start_time' % data['vm']))
  34     data['uptime'] = time.time()-data['start_time']
  35
  36     return name, data
  37
  38 if __name__ == '__main__':
  39     vms = live_vms()
  40     if '--json' in sys.argv[1:]:
  41         import cjson
  42         print cjson.encode(vms)
  43     elif '--pickle' in sys.argv[1:]:
  44         import cPickle
  45         cPickle.dump(vms, sys.stdout, cPickle.HIGHEST_PROTOCOL)
  46     else:
  47         import yaml
  48         print yaml.dump(vms, Dumper=yaml.CSafeDumper, default_flow_style=False)