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
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)