use a slightly cleaner Xen API in listvms

[invirt/packages/invirt-remote.git] / files / usr / sbin / sipb-xen-listvms

diff --git a/files/usr/sbin/sipb-xen-listvms b/files/usr/sbin/sipb-xen-listvms
index 34cff0e..a38c9ae 100755 (executable)
--- a/files/usr/sbin/sipb-xen-listvms
+++ b/files/usr/sbin/sipb-xen-listvms
@@ -3,37 +3,47 @@
 import sys
 import time
 sys.path.insert(0, '/usr/lib/xen-3.1-1/lib/python')
-from xen.lowlevel.xs import xs
-import yaml
+from xen.xend.xenstore import xstransact
 
-trans = xs()
+xst = xstransact.xstransact()
 
 def live_vms():
-    domids = set(trans.ls('', '/local/domain'))
+    domids = set(xst.List('/local/domain'))
     domids.discard('0')
 
     vms = dict()
 
     for domid in domids:
-        name, data = get_dom(int(domid))
+        try:
+            name, data = get_dom(int(domid))
+        except TypeError:
+            continue # went down since we started
         if name.startswith('d_'):
             name = name[2:]
             vms[name] = data
     return vms
 
 def get_dom(domid):
-    name = trans.read('', '/local/domain/%d/name' % domid)
+    name = xst.Read('/local/domain/%d/name' % domid)
     data = dict()
     data['domid'] = domid
     # presence of a graphical console
-    data['console'] = trans.read('', '/local/domain/%d/device/vfb/0/state' % domid)
+    data['console'] = xst.Read('/local/domain/%d/device/vfb/0/state' % domid)
     # uptime
-    data['vm'] = trans.read('', '/local/domain/%d/vm' % domid)
-    data['start_time'] = float(trans.read('', '%s/start_time' % data['vm']))
+    data['vm'] = xst.Read('/local/domain/%d/vm' % domid)
+    data['start_time'] = float(xst.Read('%s/start_time' % data['vm']))
     data['uptime'] = time.time()-data['start_time']
     
     return name, data
 
 if __name__ == '__main__':
     vms = live_vms()
-    print yaml.safe_dump(vms, default_flow_style=False)
+    if '--json' in sys.argv[1:]:
+        import cjson
+        print cjson.encode(vms)
+    elif '--pickle' in sys.argv[1:]:
+        import cPickle
+        cPickle.dump(vms, sys.stdout, cPickle.HIGHEST_PROTOCOL)
+    else:
+        import yaml
+        print yaml.dump(vms, Dumper=yaml.CSafeDumper, default_flow_style=False)