In invirt-remote:
[invirt/packages/invirt-remote.git] / 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     
46     return name, data
47
48 if __name__ == '__main__':
49     vms = live_vms()
50     if '--json' in sys.argv[1:]:
51         import cjson
52         print cjson.encode(vms)
53     elif '--pickle' in sys.argv[1:]:
54         import cPickle
55         cPickle.dump(vms, sys.stdout, cPickle.HIGHEST_PROTOCOL)
56     else:
57         import yaml
58         print yaml.dump(vms, Dumper=yaml.CSafeDumper, default_flow_style=False)