890931ad7b2ef201e5f1af9c19104d46b0d7fc10
[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.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)