X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/invirt-remote.git/blobdiff_plain/ec9f34912b12e2e15f70b796e98a6ad53d4e336a..bb817ab6a7535279e1a6432fb0be2ebfc28116ce:/files/usr/sbin/sipb-xen-listvms

diff --git a/files/usr/sbin/sipb-xen-listvms b/files/usr/sbin/sipb-xen-listvms
index 427a586..4daa953 100755
--- a/files/usr/sbin/sipb-xen-listvms
+++ b/files/usr/sbin/sipb-xen-listvms
@@ -3,40 +3,47 @@
 import sys
 import time
 sys.path.insert(0, '/usr/lib/xen-3.1-1/lib/python')
-from xen.lowlevel.xs import xs
+from xen.lowlevel import xs
 
-trans = xs()
+xsc = xs.xs()
 
 def live_vms():
-    domids = set(trans.ls('', '/local/domain'))
+    domids = set(xsc.ls('', '/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 = xsc.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'] = xsc.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'] = xsc.read('', '/local/domain/%d/vm' % domid)
+    data['start_time'] = float(xsc.read('', '%s/start_time' % data['vm']))
     data['uptime'] = time.time()-data['start_time']
     
     return name, data
 
 if __name__ == '__main__':
     vms = live_vms()
-    if '--pickle' in sys.argv[1:]:
+    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:
-        print vms
+        import yaml
+        print yaml.dump(vms, Dumper=yaml.CSafeDumper, default_flow_style=False)