#!/usr/bin/env python3.5

import psutil
import mysql.connector
import argparse
import json
import datetime

def get_cpu_info(verbose):
 cpu_info={}
 if verbose >0:
  print("[cpu] start collect cpu info ...")
 data=psutil.cpu_times_percent(3)
 cpu_info['user']=data[0]
 cpu_info['system']=data[2]
 cpu_info['idle']=data[3]
 cpu_info['iowait']=data[4]
 cpu_info['hardirq']=data[5]
 cpu_info['softirq']=data[6]
 cpu_info['cpu_cores']=psutil.cpu_count()
 if verbose >0:
  print("{0}".format(json.dumps(cpu_info,ensure_ascii=False,indent=4)))
  print("[cpu] collection compeleted ...")
 return cpu_info

def get_mem_info(verbose):
 mem_info={}
 if verbose >0:
  print("[mem] start collect mem info ...")
 data=psutil.virtual_memory()
 mem_info['total']=data[0]/1024/1024/1024
 mem_info['avariable']=data[1]/1024/1024/1024
 if verbose>0:
  print("{0}".format(json.dumps(mem_info,ensure_ascii=False,indent=4)))
  print("[mem] collection compeletd ...")
 return mem_info

def get_disk_info(verbose):
 disk_info={}
 if verbose >0:
  print("[disk] start collect disk info ...")
 partitions=psutil.disk_partitions()
 partitions=[(partition[1],partition[2])for partition in partitions if partition[2]!='iso9660']
 disk_info={}
 for partition in partitions:
  disk_info[partition[0]]={}
  disk_info[partition[0]]['fstype']=partition[1]
 for mount_point in disk_info.keys():
  data=psutil.disk_usage(mount_point)
  disk_info[mount_point]['total']=data[0]/1024/1024/1024
  disk_info[mount_point]['used_percent']=data[3]
 if verbose >0:
  print("{0}".format(json.dumps(disk_info,ensure_ascii=False,indent=4)))
  print("[disk] collection compeleted ....")
 return disk_info

def get_mysql_info(cnx_args,status_list):
 config={
  'user':cnx_args.user,
  'password':cnx_args.password,
  'host':cnx_args.host,
  'port':cnx_args.port}
 cnx=None
 cursor=None
 mysql_info={}
 try:
  cnx=mysql.connector.connect(**config)
  cursor=cnx.cursor(prepared=True)
  for index in range(len(status_list)):
   status_list[index].get_status(cursor)
   status=status_list[index]
   mysql_info[status.name]=status.value
  mysql_info['port']=config['port']
 except mysql.connector.Error as err:
  print(err)
 finally:
  if cursor != None:
   cursor.close()
  if cnx != None:
   cnx.close()
 return mysql_info

class Status(object):
 def __init__(self,name):
  self.name=name
  self._value=None


 def get_status(self,cursor):
  stmt="show global status like '{0}';".format(self.name)
  cursor.execute(stmt)
  value=cursor.fetchone()[1].decode('utf8')
  self._value=int(value)


 @property
 def value(self):
  if self._value==None:
   raise Exception("cant get value befor execute the get_status function")
  else:
   return self._value

IntStatus=Status


class diskResource(object):
 def __init__(self,mount_point,status):
  self.mount_point=mount_point
  self.status=status

 def __str__(self):
  result='''    <div class="stage-list">
     <div class="stage-title"><span>{0}</span></div>
     <div class="detail">
      <p class="detail-list">
       <span class="detail-title">区分格式</span>
       <span class="detail-describe">{1}</span>
      </p>
      <p class="detail-list">
       <span class="detail-title">总空间大小</span>
       <span class="detail-describe">{2:8.2f}G</span>
      </p>
      <p class="detail-list">
       <span class="detail-title">空闲空间(%)</span>
       <span class="detail-describe">{3:8.2f}</span>
      </p>
      <p class="detail-list">
       
      </p>
     </div>
    </div>\n'''.format(self.mount_point,self.status['fstype'],self.status['total'],self.status['used_percent'])
  return result

class diskResources(object):
 def __init__(self,status):
  self.disks=[]
  for mount_point in status.keys():
   self.disks.append(diskResource(mount_point,status[mount_point]))

 def __str__(self):
  result='''  <div class="list-item">
   <div class="category">
    <span>磁盘</span>
   </div>
   <div class="second-stage">\n'''
  for index in range(len(self.disks)):
   result=result+self.disks[index].__str__()
  result=result+'''   </div>
  </div>\n'''
  return result

class cpuResources(object):
 def __init__(self,status):
  self.status=status
 def __str__(self):
  result='''  <div class="list-item">
   <div class="category">
    <span>CPU</span>
   </div>
   <div class="second-stage">
    <div class="stage-list">
     <div class="stage-title"><span>global</span></div>
     <div class="detail">
      <p class="detail-list">
       <span class="detail-title">用户空间使用（%）</span>
       <span class="detail-describe">{0}</span>
      </p>
      <p class="detail-list">
       <span class="detail-title">内核空间使用（%）</span>
       <span class="detail-describe">{1}</span>
      </p>
      <p class="detail-list">
       <span class="detail-title">空闲（%）</span>
       <span class="detail-describe">{2}</span>
      </p>
      <p class="detail-list">
       <span class="detail-title">硬中断（%）</span>
       <span class="detail-describe">{3}</span>
      </p>
      <p class="detail-list">
       <span class="detail-title">软中断（%）</span>
       <span class="detail-describe">{4}</span>
      </p>
      <p class="detail-list">
       <span class="detail-title">io等待（%）</span>
       <span class="detail-describe">{5}</span>
      </p>
      <p class="detail-list">

      </p>
     </div>
    </div>
   </div>
  </div>\n'''.format(self.status['user'],self.status['system'],self.status['idle'],self.status['hardirq'],self.status['softirq'],self.status['iowait'])
  return result

class memResources(object):
 def __init__(self,status):
  self.status=status

 def __str__(self):
  result='''  <div class="list-item">
   <div class="category">
    <span>MEM</span>
   </div>
   <div class="second-stage">
    <div class="stage-list">
     <div class="stage-title"><span>global</span></div>
     <div class="detail">
      <p class="detail-list">
       <span class="detail-title">总大小</span>
       <span class="detail-describe">{0:8.2f}G</span>
      </p>
      <p class="detail-list">
       <span class="detail-title">空闲大小</span>
       <span class="detail-describe">{1:8.2f}G</span>
      </p>
      
      <p class="detail-list">
       
      </p>
     </div>
    </div>
   </div>
  </div>'''.format(self.status['total'],self.status['avariable'])
  return result


class mysqlResources(object):
 def __init__(self,status):
  self.status=status
 def __str__(self):
  result='''  <div class="list-item">
   <div class="category">
    <span>MYSQL</span>
   </div>
   <div class="second-stage">
    <div class="stage-list">
     <div class="stage-title"><span>{0}</span></div>
     <div class="detail">
      <p class="detail-list">
       <span class="detail-title">innodb_log_wait</span>
       <span class="detail-describe">{1}</span>
      </p>
      <p class="detail-list">
       <span class="detail-title">binlog_cache_use</span>
       <span class="detail-describe">{2}</span>
      </p>
      <p class="detail-list">
       <span class="detail-title">create_temp_disk_table</span>
       <span class="detail-describe">{3}</span>
      </p>
            <p class="detail-list">
              <span class="detail-title">Slow_querys</span>
              <span class="detail-describe">{4}</span>
            </p>

      <p class="detail-list">
       
      </p>
     </div>
    </div>
   </div>
  </div>'''.format(self.status['port'],self.status['Innodb_log_waits'],self.status['Binlog_cache_use'],
       self.status['Created_tmp_disk_tables'],self.status['Slow_queries'])

  return result

class hostResources(object):
 def __init__(self,cpu_info,mem_info,disk_info,mysql_info,report_title='MySQL巡检报告'):
  self.cpu=cpuResources(cpu_info)
  self.mem=memResources(mem_info)
  self.disk=diskResources(disk_info)
  self.mysql=mysqlResources(mysql_info)
  self.report_title=report_title
 def __str__(self):
  result='''<!DOCTYPE html>
<html lang="en">
<head>
 <meta charset="UTF-8">
 <title>巡检报告</title>
<style>
*{
 margin: 0;
 padding: 0;
}
 .content{
  width:1000px;
  height: auto;
  margin: 30px auto;
  border-bottom:1px solid #b2b2b2;
 }
 .list-item{
  border:1px solid #b2b2b2;
  border-bottom: none;
  transition: all .35s;
  overflow: hidden;
  display: flex;
 }
 .list-item:empty{
  display: none;
 }
 .top-title{
  line-height: 32px;
  font-size: 16px;
  color: #333;
  text-indent: 10px;
  font-weight: 600;
 }
 .category{
  width:97px;
  height: auto;
  border-right: 1px solid #b2b2b2;
  float: left;
  text-align: center;
  position: relative;
 }
 .stage-title>span,
 .category>span{
  display: block;
  height: 20px;
  width:100%;
  text-align: center;
  line-height: 20px;
  position: absolute;
  top: 50%;
  margin-top: -10px;left: 0;
 }
 .second-stage{
  width:900px;
  float: left;
 }
 .stage-list{
  border-bottom: 1px solid #b2b2b2;
  display: flex;
 }
 .stage-list:last-child{
  border-bottom: 0;
 }
 .stage-title{
  width:99px;
  border-right: 1px solid #b2b2b2;
  position: relative;
 }
 .detail{
  flex: 1;
 }
 .detail-list{
  border-bottom: 1px solid #b2b2b2;
  height: 40px;
  display: flex;
  transition: all .35s;
 }
 .detail-title{
  padding: 10px;
  height: 20px;
  line-height: 20px;
  border-right: 1px solid #b2b2b2;
  width:200px;
 }
 .detail-describe{
  flex: 1;
  padding: 10px;line-height: 20px;
 }
 .detail-list:last-child{
  border-bottom: 0;
 }
 .list-item:hover{
  background-color: #eee;
 }
 .detail-list:hover{
  background-color: #d1d1d1;
 }
</style>
</head>
<body>
 <div class="content">
    <div class="list-item">
      <p class="top-title">report_title</p>
    </div>\n'''

  result=result.replace('report_title',self.report_title)
  result=result+self.cpu.__str__()
  result=result+self.mem.__str__()
  result=result+self.disk.__str__()
  result=result+self.mysql.__str__()
  result=result+''' </div>
</body>
</html>'''
  return result


if __name__=="__main__":
 parser=argparse.ArgumentParser()
 parser.add_argument('--verbose',type=int,default=1,help='verbose for output')
 parser.add_argument('--user',default='chkuser',help='user name for connect to mysql')
 parser.add_argument('--password',default='123456',help='user password for connect to mysql')
 parser.add_argument('--host',default='127.0.0.1',help='mysql host ip')
 parser.add_argument('--port',default=3306,type=int,help='mysql port')
 parser.add_argument('--int-status',default=('Com_select,Com_insert,Com_update,Com_delete,Innodb_log_waits,'
            'Binlog_cache_disk_use,Binlog_cache_use,Created_tmp_disk_tables,'
            'Slow_queries')
      ,help='mysql status its value like int')
 parser.add_argument('--report-title',default='MySQL巡检报告',help='report title')
 parser.add_argument('--output-dir',default='/tmp/',help='default report file output path')
 args=parser.parse_args()
 cpu_info=get_cpu_info(args.verbose)
 mem_info=get_mem_info(args.verbose)
 disk_info=get_disk_info(args.verbose)
 status_list=[ IntStatus(name=item) for item in args.int_status.split(',')]
 mysql_info=get_mysql_info(args,status_list)
 #dr=diskResources(disk_info)
 #cr=cpuResources(cpu_info)
 #mr=memResources(mem_info)
 #msr=mysqlResources(mysql_info)
 hr=hostResources(cpu_info,mem_info,disk_info,mysql_info,args.report_title)
 now=str(datetime.datetime.now()).replace(' ','^')
 if args.output_dir.endswith('/') != True:
  args.output_dir=args.output_dir+'/'
 filename=args.output_dir+'mysql_inspection_{0}.html'.format(now)
 with open(filename,'w') as output:
  output.write(hr.__str__())
 print('[report] the report been saved to {0} ok.... ....'.format(filename))