Support YAML (default), JSON, and Pickle formats, depending on the client's choice

[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 427a586..8b78b22 100755 (executable)
--- a/files/usr/sbin/sipb-xen-listvms
+++ b/files/usr/sbin/sipb-xen-listvms
@@ -35,8 +35,12 @@ def get_dom(domid):
 
 if __name__ == '__main__':
     vms = live_vms()
 
 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:
         import cPickle
         cPickle.dump(vms, sys.stdout, cPickle.HIGHEST_PROTOCOL)
     else:

-        print vms
+        import yaml
+        print yaml.safe_dump(vms, default_flow_style=False)