zoukankan      html  css  js  c++  java
  • 获取Zabbix 中资源的使用率

    import pymysql as MySQLdb
    import time
    import datetime
    import xlsxwriter
    
    # zabbix数据库信息:
    zdbhost = 'xxx.xxx.xxx.xxx'
    zdbuser = 'MySQL-username'
    zdbpass = MySQL-PassWord'
    zdbport = 3306
    zdbname = 'zabbix-DBName'
    
    # 生成文件名称:
    xlsfilename = 'Group_Production_Server.xlsx'
    
    # 需要查询的key列表 [名称,表名,key值,取值,格式化,数据整除处理]
    keys = [
        ['CPU空闲率(%)', 'trends', 'system.cpu.util[,idle]', 'avg', '%.2f', 1],
        ['内存使用率(G)', 'trends_uint', 'vm.memory.size[total]', 'avg', '%.2f', 1],
        ['磁盘使用率(%)', 'trends', 'system.cpu.util[,iowait]', 'avg', '%.2f', 1],
    ]
    
    
    class ReportForm(object):
    
        def __init__(self):
            """打开数据库连接"""
            self.conn = MySQLdb.connect(host=zdbhost, user=zdbuser, passwd=zdbpass, port=zdbport, db=zdbname)
            self.cursor = self.conn.cursor()
            # 生成zabbix哪个分组报表
            self.groupname = 'Mysql'
            # 获取IP信息:
            self.IpInfoList = self.__getHostList()
    
        def __getHostList(self):
            """根据zabbix组名获取该组所有IP"""
            # 查询组ID:
            sql = '''select groupid from groups where name = '%s' ''' % self.groupname
            self.cursor.execute(sql)
            groupid = self.cursor.fetchone()[0]
            # 根据groupid查询该分组下面的所有主机ID(hostid):
            sql = '''select hostid from hosts_groups where groupid = '%s' ''' % groupid
            self.cursor.execute(sql)
            hostlist = self.cursor.fetchall()
            # 生成IP信息字典:结构为{'119.146.207.19':{'hostid':10086L,},}
            IpInfoList = {}
            for i in hostlist:
                hostid = i[0]
                sql = '''SELECT ip from interface WHERE hostid = '%s' ''' % hostid
                ret = self.cursor.execute(sql)
                if ret:
                    IpInfoList[self.cursor.fetchone()[0]] = {'hostid': hostid}
            return IpInfoList
    
        def __getItemid(self, hostid, itemname):
            """获取itemid"""
    
            sql = '''select itemid from items where hostid = '%s' and key_ = '%s' ''' % (hostid, itemname)
            self.cursor.execute(sql)
            itemid = self.cursor.fetchone()[0]
            # print('hostid --> ', hostid)
            # print('itemid --> ', itemid)
            return itemid
    
        def getTrendsValue(self, type, itemid, start_time, stop_time):
            """查询trends_uint表的值,type的值为min,max,avg三种"""
    
            sql = '''select %s(value_%s) as result from trends where itemid = '%s' and clock >= '%s' and clock <= '%s' ''' % (
                type, type, itemid, start_time, stop_time)
            # print('trends --> ', sql)
            self.cursor.execute(sql)
            result = self.cursor.fetchone()[0]
            if result == None:
                result = 0
            # print('trends - result --> ', result)
            return result
    
        def getTrends_uintValue(self, type, itemid, start_time, stop_time):
            """查询trends_uint表的值,type的值为min,max,avg三种"""
    
            sql = '''select %s(value_%s) as result from trends_uint where itemid = '%s' and clock >= '%s' and clock <= '%s' ''' % (
                type, type, itemid, start_time, stop_time)
            # print('trends_uint --> ', sql)
            self.cursor.execute(sql)
            result = self.cursor.fetchone()[0] / 1073741824
            if result:
                result = int(result)
            else:
                result = 0
            # print('trends_uint - result--> ', result)
            return result
    
        def getLastMonthData(self, type, hostid, table, itemname):
            """根据hostid,itemname获取该监控项的值"""
    
            # 获取上个月的第20天和最后1天
            ts_first = int(
                time.mktime(datetime.date(datetime.date.today().year, datetime.date.today().month - 1, 20).timetuple()))
            lst_last = datetime.date(datetime.date.today().year, datetime.date.today().month, 1) - datetime.timedelta(1)
            ts_last = int(time.mktime(lst_last.timetuple()))
            itemid = self.__getItemid(hostid, itemname)
            function = getattr(self, 'get%sValue' % table.capitalize())
            return function(type, itemid, ts_first, ts_last)
    
        def getInfo(self):
    
            # 循环读取IP列表信息
            for ip, resultdict in zabbix.IpInfoList.items():
                print("正在查询 IP:%-15s hostid:%5d 的信息!" % (ip, resultdict['hostid']))
            # 循环读取keys,逐个key统计数据:
                for value in keys:
                    print("	正在统计 key_:%s" % value[2])
                    if not value[2] in zabbix.IpInfoList[ip]:
                        zabbix.IpInfoList[ip][value[2]] = {}
                    data = zabbix.getLastMonthData(value[3], resultdict['hostid'], value[1], value[2])
                    zabbix.IpInfoList[ip][value[2]][value[3]] = data
    
        def writeToXls2(self):
            """生成xls文件"""
    
            # 创建文件
            workbook = xlsxwriter.Workbook(xlsfilename)
            # 创建工作薄
            worksheet = workbook.add_worksheet()
            # 写入第一列:
            worksheet.write(0, 0, "主机")
            i = 1
            for ip in self.IpInfoList:
                worksheet.write(i, 0, ip)
                i = i + 1
            # 写入其他列:
            for i in range(1, 5):
                i = 1
                for value in keys:
                    worksheet.write(0, i, value[0])
                    # 写入该列内容:
                    j = 1
                    for ip, result in self.IpInfoList.items():
                        if value[4]:
                            worksheet.write(j, i, value[4] % result[value[2]][value[3]])
                        else:
                            worksheet.write(j, i, result[value[2]][value[3]] / value[5])
                        j = j + 1
                    i = i + 1
            workbook.close()
    
        def __del__(self):
            """关闭数据库连接"""
    
            self.cursor.close()
            self.conn.close()
    
    
    if __name__ == "__main__":
        zabbix = ReportForm()
        zabbix.getInfo()
        zabbix.writeToXls2()
    

    参考: https://www.jb51.net/article/167771.htm

  • 相关阅读:
    闭包Closures
    jsp中的四种对象作用域
    JFrame小练习1
    swift函数的用法,及其嵌套实例
    图的储存
    洛谷P3366 【模板】最小生成树(Kruskal && Prim)
    洛谷P3371 【模板】单源最短路径(弱化版)(SPFA解法)
    POJ 1577 Falling Leaves
    POJ 3784 Running Median
    洛谷P3374 【模板】树状数组 1
  • 原文地址:https://www.cnblogs.com/winstom/p/11549812.html
Copyright © 2011-2022 走看看