Get total vms from Vmware and send to InfluxDB

get_total_vms.py

from vpoller.client import VPollerClient
import json
from influxdb.influxdb08 import InfluxDBClient
client = VPollerClient(endpoint='tcp://localhost:10123')

def get_total_vms():
    msg = {'method': 'vm.discover', 'hostname': '1.1.1.1'}
    r = json.loads(client.run(msg))
    data = [
        {
            'name': 'vmware',
            'columns': ['name', 'location', 'value'],
            'points': [
                    ['total_vms', 'Central Data Center', len(r['result'])]
            ]
        }
    ]
    return data

def send_to_influx(data):
    '''Send data to InfluxDB'''
    client = InfluxDBClient(
        'localhost', 8086, 'monitor', 'monitor', 'vmware')
    client.write_points(data)

send_to_influx(get_total_vms())