#!/usr/bin/python

import sys
import time
sys.path.insert(0, '/usr/lib/xen-3.1-1/lib/python')
from xen.lowlevel.xs import xs

trans = xs()

def live_vms():
    domids = set(trans.ls('', '/local/domain'))
    domids.discard('0')

    vms = dict()

    for domid in domids:
        name, data = get_dom(int(domid))
        if name.startswith('d_'):
            name = name[2:]
            vms[name] = data
    return vms

def get_dom(domid):
    name = trans.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)
    # uptime
    data['vm'] = trans.read('', '/local/domain/%d/vm' % domid)
    data['start_time'] = float(trans.read('', '%s/start_time' % data['vm']))
    data['uptime'] = time.time()-data['start_time']
    
    return name, data

if __name__ == '__main__':
    vms = live_vms()
    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.safe_dump(vms, default_flow_style=False)
