#!/usr/bin/python3 # @Date: 2020/8/20 14:16 # @Author: zhangcheng # @email: 3359957053@qq.com # -*- coding: utf-8 -*- import pymysql import time,datetimeimport math #zabbix数据库信息： zdbhost = "192.168.200.41" zdbuser = "root" zdbpass = "zabbix" zdbport = 3306 zdbname = "zabbix" #需要查询的key列表 keys = { \'trends_uint\':[ # \'vfs.fs.size[/,used]\', \'vm.memory.size[available]\', \'vm.memory.size[total]\', ], \'trends\':[ \'system.cpu.load[percpu,avg15]\', \'system.cpu.util[,idle]\', \'system.cpu.num\', ], } class ReportForm: def __init__(self): \'\'\'打开数据库连接\'\'\' self.conn = pymysql.connect(host=zdbhost,user=zdbuser,passwd=zdbpass,port=zdbport,db=zdbname,charset="utf8") self.cursor = self.conn.cursor() #生成zabbix哪个分组报表 self.groupname = "智能产品事业部" #获取IP信息： self.IpInfoList = self.__getHostList() def __getHostList(self): \'\'\'根据zabbix组名获取该组所有IP\'\'\' #查询组ID: sql = \'\'\'select groupid from hstgrp where name = "{0}"\'\'\'.format(self.groupname) self.cursor.execute(sql) groupid = self.cursor.fetchone()[0] #根据groupid查询该分组下面的所有主机ID（hostid）： sql = \'\'\'select hostid from hosts_groups where groupid = {0}\'\'\'.format(groupid) self.cursor.execute(sql) hostlist = [i for i in list("%s" %j for j in self.cursor.fetchall())] # [\'10550\', \'10551\', \'10552\', \'10553\', \'10555\', \'10556\', \'10557\', \'10558\', \'10559\', \'10560\', \'10561\', \'10562\',\'10563\', \'10564\'] #生成IP信息字典：结构为{\'112.111.222.55\':{\'hostid\':\'10086L\'},} IpInfoList = {} for hostid in hostlist: sql = \'\'\'select host from hosts where status = 0 and hostid = {0}\'\'\'.format(hostid) ret = self.cursor.execute(sql) if ret: IpInfoList["".join("%s" %i for i in list(self.cursor.fetchone()))] = {\'hostid\':hostid} return IpInfoList def __getItemid(self,hostid,itemname): \'\'\'获取itemid\'\'\' sql = \'\'\'select itemid from items where hostid = {0} and key_ = "{1}" \'\'\'.format(hostid, itemname) print(sql) if self.cursor.execute(sql): itemid = "".join("%s" %i for i in list(self.cursor.fetchone())) else: print(hostid,itemname) itemid = None return itemid def getTrendsValue(self,itemid, start_time, stop_time): \'\'\'查询trends_uint表的值,type的值为min,max,avg三种\'\'\' resultlist = {} for type in [\'min\',\'max\',\'avg\']: sql = \'\'\'select {0}(value_{1}) as result from trends where itemid = {2} and clock >= {3} and clock <= {4}\'\'\'.format(type, type, itemid, start_time, stop_time) self.cursor.execute(sql) result = self.cursor.fetchone()[0] if result == None: result = 0 resultlist[type] = result return resultlist def getTrends_uintValue(self,itemid, start_time, stop_time): \'\'\'查询trends_uint表的值,type的值为min,max,avg三种\'\'\' resultlist = {} for type in [\'min\',\'max\',\'avg\']: sql = \'\'\'select {0}(value_{1}) as result from trends_uint where itemid = {2} and clock >= {3} and clock <= {4}\'\'\'.format(type, type, itemid, start_time, stop_time) self.cursor.execute(sql) result = self.cursor.fetchone()[0] if result: resultlist[type] = result else: resultlist[type] = 0 return resultlist def getLastMonthData(self,hostid,table,itemname): \'\'\'根据hostid,itemname获取该监控项的值\'\'\' #获取上个月的第一天和最后一天 ts_first = int(time.mktime(datetime.date(datetime.date.today().year,datetime.date.today().month,1).timetuple())) # lst_last = datetime.date(datetime.date.today().year,datetime.date.today().month,1)-datetime.timedelta(1) lst_last = datetime.date(datetime.date.today().year,datetime.date.today().month+1,1)-datetime.timedelta(1) ts_last = int(time.mktime(lst_last.timetuple())) itemid = self.__getItemid(hostid, itemname) function = getattr(self,\'get{0}Value\'.format(table.capitalize())) return function(itemid, ts_first, ts_last) def getInfo(self): #循环读取IP列表信息 for ip,resultdict in zabbix.IpInfoList.items(): print("正在查询 IP:%-15s hostid:%5d 的信息" %(ip,int(resultdict[\'hostid\']))) #循环读取keys，逐个key统计数据： for table, keylists in keys.items(): for key in keylists: print("\t正在统计 key_:\'{0}\'".format(key)) data = zabbix.getLastMonthData(resultdict[\'hostid\'],table,key) zabbix.IpInfoList[ip][key] = data def writeToXls(self): \'\'\'生成xls文件\'\'\' try: import xlsxwriter date = time.strftime("%Y-%m",time.localtime()) #创建文件 workbook = xlsxwriter.Workbook(self.groupname + "-" + date + \'.xls\') #创建工作薄 worksheet = workbook.add_worksheet() #写入标题（第一行） i = 0 for value in ["归属部门","主机","CPU核数","CPU平均空闲值","CPU最小空闲值","CPU15分钟负载","总内存","平均可用内存","最小可用内存"]: worksheet.write(0,i, value) i = i + 1 #写入内容： j = 1 for ip,value in self.IpInfoList.items(): worksheet.write(j,0, self.groupname) worksheet.write(j,1, ip) worksheet.write(j,2, \'{}\'.format(value[\'system.cpu.num\'][\'avg\'])) worksheet.write(j,3, \'{:.2f}%\'.format(value[\'system.cpu.util[,idle]\'][\'avg\'])) worksheet.write(j,4, \'{:.2f}%\'.format(value[\'system.cpu.util[,idle]\'][\'min\'])) worksheet.write(j,5, \'{:.2f}\'.format(value[\'system.cpu.load[percpu,avg15]\'][\'avg\'])) worksheet.write(j,6, \'{}GB\'.format(math.ceil(value[\'vm.memory.size[total]\'][\'avg\'] / 1024 / 1024 / 1024))) worksheet.write(j,7, \'{}GB\'.format(math.ceil(value[\'vm.memory.size[available]\'][\'avg\'] / 1024 / 1024 / 1024))) worksheet.write(j,8, \'{}GB\'.format(math.ceil(value[\'vm.memory.size[available]\'][\'min\'] / 1024 / 1024 / 1024))) j = j + 1 workbook.close() except Exception as e: print(e) def __del__(self): \'\'\'关闭数据库连接\'\'\' self.cursor.close() self.conn.close() if __name__ == "__main__": zabbix = ReportForm() zabbix.getInfo() zabbix.writeToXls()
效果图如下，由于CPU核数是新增的，故在本月是没有数据的：