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 is not None and 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     # CD or autoinstall, if we're using either
  36     vbds = xsc.ls('', '/local/domain/%d/device/vbd' % domid)
  37     for vbd in vbds:
  38         dev = '/local/domain/%d/device/vbd/%s' % (domid, vbd)
  39         dev_type = xsc.read('', '%s/device-type' % dev)
  40         dev_params = xsc.read('', '%s/params' % xsc.read('', '%s/backend' % dev))
  41         if dev_type == 'cdrom':
  42             data['cdrom'] = dev_params
  43         elif dev_type == 'disk' and dev_params == '/dev/xenvg/s_install_hda':
  44             data['autoinstall'] = True
  45     if 'auto=true' in xsc.read('', '%s/image/cmdline' % data['vm']):
  46         data['autoinstall'] = True
  47
  48     return name, data
  49
  50 if __name__ == '__main__':
  51     vms = live_vms()
  52     if '--json' in sys.argv[1:]:
  53         import cjson
  54         print cjson.encode(vms)
  55     elif '--pickle' in sys.argv[1:]:
  56         import cPickle
  57         cPickle.dump(vms, sys.stdout, cPickle.HIGHEST_PROTOCOL)
  58     else:
  59         import yaml
  60         print yaml.dump(vms, Dumper=yaml.CSafeDumper, default_flow_style=False)