python获取esxi的磁盘使用率信息

#!/usr/bin/python3

#coding:utf-8

#Author: ziming

"""

只用于模拟开发功能测试

"""

from pyVmomi import vim

from pyVim.connect import SmartConnect, Disconnect, SmartConnectNoSSL

import sys

import atexit

import argparse

from optparse import OptionParser

import json

class Exsi(object):

    # 获取vcenter的相关参数

    def __init__(self, host = 'x.x.x.x', port=443, user = 'zabbix', password='zabbix'):

        self._host = host

        self._port = port

        self._user = user

        self._password = password

    def get_obj(self,content, vimtype, name=None):

        '''

        列表返回,name 可以指定匹配的对象

        '''

        container = content.viewManager.CreateContainerView(content.rootFolder, vimtype, True)

        obj = [ view for view in container.view]

        return obj

    def get_conn(self):

        # 连接vcenter

        si = SmartConnectNoSSL(

                host=self._host,

                user=self._user,

                pwd=self._password,

                port=self._port)

        # disconnect this thing

        atexit.register(Disconnect, si)

        content = si.RetrieveContent()

        return content

    # 获取exsi的信息ip，磁盘使用情况等

    def get_esxi_host(self):

        esxi_host = {}

        content = self.get_conn()

        esxi_obj = self.get_obj(content, [vim.HostSystem])

        for esxi in esxi_obj:

            esxi_host[esxi.name] = {'datastore':{}}

            # for ds in esxi.datastore:

            #     esxi_host[esxi.name]['datastore'][ds.name] = {}

            #     esxi_host[esxi.name]['datastore'][ds.name]['总容量(G)'] = int((ds.summary.capacity)/1024/1024/1024)

            #     esxi_host[esxi.name]['datastore'][ds.name]['空闲容量(G)'] = int((ds.summary.freeSpace)/1024/1024/1024)

            #     esxi_host[esxi.name]['datastore'][ds.name]['类型'] = (ds.summary.type)

            total_capacity = 0

            free_capacity = 0

            for ds in esxi.datastore:

                # esxi_host[esxi.name]['datastore'][ds.name] = {}

                esxi_host[esxi.name] = {}

                total_capacity += int((ds.summary.capacity)/1024/1024/1024)

                free_capacity += int((ds.summary.freeSpace)/1024/1024/1024)

            esxi_host[esxi.name]['sotrage_used_percent'] = 100*(total_capacity - free_capacity)/total_capacity

        # print(esxi_host)

        return esxi_host

    # 获取exsi的host即ip列表

    def get_esxi_iplist(self):

        esxi_host = self.get_esxi_host()

        data = list()

        for exsi_ip in esxi_host:

            data.append({"{#EXSI_IP}": exsi_ip})

        return json.dumps({'data': data}, sort_keys=True, indent=7, separators=(",",":"))

    # 获取指定esxi的磁盘信息

    def get_esxi_storageinfo(self, exsi_ip):

        esxi_host = self.get_esxi_host()

        return esxi_host[exsi_ip]['sotrage_used_percent']

def main():

    try:

        usage = "usage: %prog [options]\ngGet exsi Stat"

        parser = OptionParser(usage)

        # 接收参数

        parser.add_option("-l", "--list",

                          action="store_true", dest="is_list", default=False,  help="if list all exsi ip")

        parser.add_option("--ip", "--ipaddr",

                          action="store", dest="ipaddr", type="string",

                          default="192.168.254.56", help="execute 'exsi info' to see more infomation")

        (options, args) = parser.parse_args()

        if 1 >= len(sys.argv):

            parser.print_help()

            return

        #exsi_ins = Exsi(options.ipaddr)

        exsi_ins = Exsi()

        if options.is_list == True:

            print(exsi_ins.get_esxi_iplist())

            return

        # print(redis.ins.get_esxi_storageinfo(optins.ip, exsi_ip = ip))

        print(exsi_ins.get_esxi_storageinfo(options.ipaddr))

    except Exception as expt:

        import traceback

        tb = traceback.format_exc()

        print(tb)

if __name__ == '__main__':

    main()