from pysphere import VIServer
import dns.query
import dns.zone
import dns.rdatatype
import re


username = 'root'
password = 'poot'
domain = 'poot.poot'
nameserver = '192.168.10.16'
wikipage = '/var/moinmoin/citylink/data/pages/EsxHosts/revisions/00000001'


zone_data= dns.zone.from_xfr(dns.query.xfr(nameserver, domain))

servers = []
for (name, rdataset) in zone_data.iterate_rdatasets(dns.rdatatype.A):
    if re.search("vm", name.to_text()):
        servers.append(name.to_text() + "." + domain)

esx = {}

for host in servers:
    esx[host] = {}
    server = VIServer()
    server.connect(host, "root", "poot")
    vmlist = server.get_registered_vms()
    esx[host]['version'] = server.get_api_version()
    esx[host]['vms'] = []
    for vm in vmlist:
        esx[host]['vms'].append(vm.split('/')[0].split(' ')[1])
    server.disconnect()

esxPage = open(wikipage,'w')

for host in esx.keys():
    esxPage.write('== %s ==\n\n' % host)
    esxPage.write('=== Version: %s ===\n\n' % esx[host]['version'])
    esxPage.write('=== VMs: ===\n\n')
    for client in esx[host]['vms']:
        esxPage.write(' * %s\n\n' % client)

esxPage.close()