#!/usr/bin/python
#coding:utf-8
 
import MySQLdb
import time,datetime
 
 
#zabbix數據庫信息：
zdbhost = '172.16.8.200'
zdbuser = 'zabbix'
zdbpass = 'zabbix'
zdbport = 3306
zdbname = 'zabbix'
 
#生成文件名稱：
xlsfilename = 'zabbix.xls'
 
#需要查詢的key列表 [名稱，表名，key值，取值，格式化，數據整除處理]
keys = [
  ['CPU核心數','trends_uint','system.cpu.num','avg','',1],
  ['CPU平均空閑值','trends','system.cpu.util[,idle]','avg','%.2f',1],
  ['CPU最小空閑值','trends','system.cpu.util[,idle]','min','%.2f',1],
  ['CPU5分鐘負載','trends','system.cpu.load[percpu,avg5]','avg','%.2f',1],
  ['物理內存大小(單位G)','trends_uint','vm.memory.size[total]','avg','',1048576000],
  ['可用平均內存(單位G)','trends_uint','vm.memory.size[available]','avg','',1048576000],
  ['可用最小內存(單位G)','trends_uint','vm.memory.size[available]','min','',1048576000],
  ['swap總大小(單位G)','trends_uint','system.swap.size[,total]','avg','',1048576000],
  ['swap平均剩余(單位G)','trends_uint','system.swap.size[,free]','avg','',1048576000],
  ['根分區總大小(單位G)','trends_uint','vfs.fs.size[/,total]','avg','',1073741824],
  ['根分區平均剩余(單位G)','trends_uint','vfs.fs.size[/,free]','avg','',1073741824],
  ['進入最大流量（單位Kbps）','trends_uint','net.if.in[eth0]','max','',1000],
  ['進入平均流量（單位Kbps）','trends_uint','net.if.in[eth0]','avg','',1000],
  ['出去最大流量（單位Kbps）','trends_uint','net.if.out[eth0]','max','',1000],
  ['出去平均流量（單位Kbps）','trends_uint','net.if.out[eth0]','avg','',1000],
]
 
 
class ReportForm:
 
  def __init__(self):
    '''打開數據庫連接'''
    self.conn = MySQLdb.connect(host=zdbhost,user=zdbuser,passwd=zdbpass,port=zdbport,db=zdbname)
    self.cursor = self.conn.cursor(cursorclass=MySQLdb.cursors.DictCursor)
 
    #生成zabbix哪個分組報表
    self.groupname = 'zabbix server'
 
    #獲取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()['groupid']
 
    #根據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['hostid']
      sql = '''select host from hosts where status = 0 and hostid = %s''' % hostid
      ret = self.cursor.execute(sql)
      if ret:
        IpInfoList[self.cursor.fetchone()['host']] = {'hostid':hostid}
    return IpInfoList
 
  def __getItemid(self,hostid,itemname):
    '''獲取itemid'''
    sql = '''select itemid from items where hostid = %s and key_ = '%s' ''' % (hostid, itemname)
    if self.cursor.execute(sql):
      itemid = self.cursor.fetchone()['itemid']
    else:
      itemid = None
    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)
    self.cursor.execute(sql)
    result = self.cursor.fetchone()['result']
    if result == None:
      result = 0
    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)
    self.cursor.execute(sql)
    result = self.cursor.fetchone()['result']
    if result:
      result = int(result)
    else:
      result = 0
    return result
 
 
  def getLastMonthData(self,type,hostid,table,itemname):
    '''根據hostid,itemname獲取該監控項的值'''
    #獲取上個月的第一天和最后一天
    ts_first = int(time.mktime(datetime.date(datetime.date.today().year,datetime.date.today().month-1,1).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 "\t正在統計 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文件'''
    try:
      import xlsxwriter
 
      #創建文件
      workbook = xlsxwriter.Workbook(xlsfilename)
 
      #創建工作薄
      worksheet = workbook.add_worksheet()
 
      #寫入第一列：
      worksheet.write(0,0,"主機".decode('utf-8'))
      i = 1
      for ip in self.IpInfoList:
        worksheet.write(i,0,ip)
        i = i + 1
 
      #寫入其他列：
      i = 1
      for value in keys:
        worksheet.write(0,i,value[0].decode('utf-8'))
 
        #寫入該列內容：
        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
    except Exception,e:
      print e
 
 
 
  def __del__(self):
    '''關閉數據庫連接'''
    self.cursor.close()
    self.conn.close()
 
if __name__ == "__main__":
  zabbix = ReportForm()
  zabbix.getInfo()
  zabbix.writeToXls2()