[ python3 ] 基於zabbix 自動生成xlsx監控文檔

　　準備作一個每週自動的巡檢報告，數據來源於zabbix，經過python讀取zabbix數據庫獲取數據並製做成excel表格，

原本打算直接郵件發送到指定郵箱，可是都被SMTP過濾掉了，試過126和QQ的，其餘的不知道。

操做系統：Cenos 6.7

python版本： python3.5

如下代碼是親測並生產環境使用的：

#!/usr/local/python/bin/python3.5
import pymysql
import time, datetime
import smtplib
import os
from email.mime.text import MIMEText
from email.header import Header
# zabbix數據庫信息

zdbhost = 'zabbix服務器ip'
zdbuser = 'zabbix數據庫用戶名'
zdbpass = 'zabbix數據庫密碼'
zdbport = 3306
zdbname = 'zabbix'

d = datetime.datetime.now()
day = datetime.date.today()
keys = {
    'trends_uint': [
        'net.if.in[eth0]',
        'net.if.out[eth0]',
        'vfs.fs.size[/,used]',
        'vm.memory.size[available]',
    ],
    'trends': [
        'system.cpu.load[percpu,avg5]',
        'system.cpu.util[,idle]',
    ],
}
class ReportForm:

    def __init__(self):
        self.conn = pymysql.connect(host=zdbhost, user=zdbuser, passwd=zdbpass, port=zdbport, db=zdbname)
        self.cursor = self.conn.cursor()
        self.groupname = 'Linux servers'
        self.IpInfoList = self.__getHostList()
        # return self.IpInfoList

    def __getHostList(self):
        sql = '''select groupid from groups where name = '%s' ''' % self.groupname
        self.cursor.execute(sql)
        groupid = self.cursor.fetchone()[0]
        print(groupid)

        sql = '''select hostid from hosts_groups where groupid = %s''' % groupid
        self.cursor.execute(sql)
        hostlist = self.cursor.fetchall()

        IpInfoList = {}
        for i in hostlist:
            hostid = i[0]
            sql = '''select host from hosts where status = 0 and hostid = %s''' % hostid
            ret = self.cursor.execute(sql)
            if ret:
                IpInfoList[self.cursor.fetchone()[0]] = {'hostid': hostid}
        return IpInfoList

    def __getItemid(self, hostid, itemname):
        sql = '''select itemid from items where hostid = %s and key_ = '%s' ''' % (hostid, itemname)
        if self.cursor.execute(sql):
            itemid = self.cursor.fetchone()[0]
        else:
            itemid = None
        return itemid

    def getTrendsValue(self, itemid, start_time, stop_time):
        resultlist = {}
        for type in ['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)
            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):
        resultlist = {}
        for type in ['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)
            self.cursor.execute(sql)
            result = self.cursor.fetchone()[0]
            if result:
                resultlist[type] = int(result)
            else:
                resultlist[type] = 0
        return resultlist



    def get_week(self, d):
        dayscount = datetime.timedelta(days=d.isoweekday())
        dayto = d - dayscount
        sixdays = datetime.timedelta(days=6)
        dayfrom = dayto - sixdays
        date_from = datetime.datetime(dayfrom.year, dayfrom.month, dayfrom.day, 0, 0, 0)
        date_to = datetime.datetime(dayto.year, dayto.month, dayto.day, 23, 59, 59)
        ts_first = int(time.mktime(datetime.datetime(dayfrom.year, dayfrom.month, dayfrom.day, 0, 0, 0).timetuple()))
        ts_last = int(time.mktime(datetime.datetime(dayto.year, dayto.month, dayto.day, 23, 59, 59).timetuple()))
        return ts_first, ts_last

    def getLastMonthData(self, hostid, table, itemname):
        ts_first = self.get_week(d)[0]
        ts_last = self.get_week(d)[1]
        itemid = self.__getItemid(hostid, itemname)
        # function = getattr(self, 'get %s Value' % table.capitalize())
        function = getattr(self, 'get%sValue' % table.capitalize())
        return function(itemid, ts_first, ts_last)

    def getinfo(self):
        for ip, resultdict in zabbix.IpInfoList.items():
            print("正在查詢 IP:%-15s hostid:%5d 的信息！" % (ip, resultdict['hostid']))
            for table, keylists in keys.items():
                for key in keylists:
                    print("\t正在統計 key_:%s" % key)
                    data = zabbix.getLastMonthData(resultdict['hostid'], table, key)
                    zabbix.IpInfoList[ip][key] = data

    def writeToXls(self):
        dayscount = datetime.timedelta(days=d.isoweekday())
        dayto = d - dayscount
        sixdays = datetime.timedelta(days=6)
        dayfrom = dayto - sixdays
        date_from = datetime.date(dayfrom.year, dayfrom.month, dayfrom.day)
        date_to = datetime.date(dayto.year, dayto.month, dayto.day)
        '''生成xls文件'''
        try:
            import xlsxwriter
            # 建立文件
            workbook = xlsxwriter.Workbook('/usr/monitor/week/%s_%s巡檢報告.xlsx' % (date_from, date_to))
            # 建立工做薄
            worksheet = workbook.add_worksheet()
            # 寫入標題（第一行）
            i = 0
            for value in ["主機", "CPU平均空閒值", "CPU最小空閒值", "可用平均內存(單位M)", "可用最小內存(單位M)", "CPU5分鐘負載", "進入最大流量（單位Kbps）",
                          "進入平均流量（單位Kbps）", "出去最大流量（單位Kbps）", "出去平均流量（單位Kbps）"]:
                worksheet.write(0, i, value)
                i = i + 1
            # 寫入內容：
            j = 1
            for ip, value in self.IpInfoList.items():
                worksheet.write(j, 0, ip)
                worksheet.write(j, 1, '%.2f' % value['system.cpu.util[,idle]']['avg'])
                worksheet.write(j, 2, '%.2f' % value['system.cpu.util[,idle]']['min'])
                worksheet.write(j, 3, '%dM' % int(value['vm.memory.size[available]']['avg'] / 1024 / 1024))
                worksheet.write(j, 4, '%dM' % int(value['vm.memory.size[available]']['min'] / 1024 / 1024))
                worksheet.write(j, 5, '%.2f' % value['system.cpu.load[percpu,avg5]']['avg'])
                worksheet.write(j, 6, value['net.if.in[eth0]']['max'] / 1000)
                worksheet.write(j, 7, value['net.if.in[eth0]']['avg'] / 1000)
                worksheet.write(j, 8, value['net.if.out[eth0]']['max'] / 1000)
                worksheet.write(j, 9, value['net.if.out[eth0]']['avg'] / 1000)
                j = j + 1
            workbook.close()
        except Exception as e:
            print(e)
    def __del__(self):
        '''關閉數據庫鏈接'''
        self.cursor.close()
        self.conn.close()

    def sendmail(self):
        sender = '發送者郵箱'
        receiver = ['接收者1', '接收者2', '接收者3']
        subject = '上週巡檢報告'
        smtpserver = 'smtp.126.com'
        username = '發送者郵箱'
        password = 'smtp認證密碼'

        msg = MIMEText('上週巡檢報告已生成，請檢查。\n報告目錄:/usr/monitor\n因爲郵件沒法發送excel文檔，請使用ftp登陸'
                       '查看。\nftp地址:xxx:端口\nftp用戶名：xxx\n', 'plain', 'utf-8')  # 中文需參數‘utf-8'，單字節字符不須要
        msg['Subject'] = Header(subject, 'utf-8')
        msg['From'] = 'Robot<發送者郵箱>'
        msg['To'] = "接收者名字"
        smtp = smtplib.SMTP()
        smtp.connect('smtp.126.com')
        smtp.login(username, password)
        smtp.sendmail(sender, receiver, msg.as_string())
        print('發送成功！')
        smtp.quit()

if __name__ == "__main__":
    zabbix = ReportForm()
    zabbix.getinfo()
    zabbix.writeToXls()
    zabbix.sendmail()

 

　　說明： 因爲郵件沒法發送附件爲excel，因此我這裏寫了一個發送通知郵件的方法，須要經過ftp登陸並下載。

附上生成execl部分截圖：