lrx182125 发表于 2017-12-31 20:42:16

zabbix脚本报警

#! /usr/bin/env python3  
import requests
  
import json
  
import logging
  
import sys, os
  

  
print(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
  
# print(sys.path)
  
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
  
print(BASE_DIR)
  

  
class MyLogger:
  def __init__(self, msg):
  self.logger = self.initlog()
  self.msg = msg
  

  def initlog(self):
  logger = logging.getLogger(__name__)
  logger.setLevel(logging.DEBUG)
  fh = logging.FileHandler(BASE_DIR + '/log/' + 'char.log')
  fh.setLevel(logging.DEBUG)
  formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
  fh.setFormatter(formatter)
  logger.addHandler(fh)
  return logger
  

  def savelog(self):
  self.logger.debug(self.msg)
  

  

  

  
class ZabbixSendMail():
  def __init__(self,sys_argv):
  self.argv = sys_argv
  self.url = "https://qyapi.weixin.qq.com/cgi-bin/gettoken"
  self.value = {
  'corpid':"wx5a109e09d5044152",
  'corpsecret' : "0fiTLHoU0SBtNcBm4huzPILDt0QxjhdsVcnNBPVw9lVCR3vGVf7wf2p4HDTiilaQ"
  }
  self.sendmail_url = "https://qyapi.weixin.qq.com/cgi-bin/message/send"
  self.access_resprons = requests.get(self.url, params=self.value)
  # self.token = json.loads(self.access_resprons.text)['access_token']
  self.respons = self.access_resprons.text#获取返回结果
  self.log = MyLogger("action")
  self.log.savelog()
  

  

  def get_token(self):
  ret = json.loads(self.respons)
  return ret["access_token"]
  

  

  def send_mail(self):
  try:
  send_mail_url = self.sendmail_url + "?access_token=%s"% self.get_token()
  #print(send_mail_url)
  msg = "".join(self.argv)
  payload = """{"touser": "%s",
  "msgtype": "text",
  "agentid": 1,
  "text":{"content": "内容:%s"},"safe":0}""" % (self.argv, msg)
  dic_payload = eval(payload)
  # ret = requests.post(send_mail_url, data=json.dumps(payload, ensure_ascii=False))
  

  ret = requests.post(send_mail_url, data=json.dumps(dic_payload, ensure_ascii=False).encode('UTF-8'))
  #发送中文时就需要ensure_ascii与.encode('utf-8')这两个参数配合才可以否则会报错
  log = MyLogger(ret.text)
  log.savelog()
  

  except Exception as e:
  # print(e)
  log = MyLogger(e)
  log.savelog()
  

  

  
if __name__ == "__main__":
  send = ZabbixSendMail(sys.argv)
  send.send_mail()
  
页: [1]
查看完整版本: zabbix脚本报警