通过itchat调用zabbixAPI实现微信确认zabbix告警

环境：python3.6.1 ,pip 9.0.1

1，通过itchat,实现微信的扫码登录，和关键定识别

#coding=utf-8
import urllib.request, urllib.error, urllib.parse
import sys
import json
import argparse
from login import *
from function import *
########################################################################

import itchat
import time
import re
import requests
import threading
import shutil
import os

'''
reload(sys)
sys.setdefaultencoding('utf-8')
'''
self_name='PH'

 
@itchat.msg_register(itchat.content.TEXT)
def cmd(revmsg):
    msg = revmsg['Text']
    
    if msg == "注销登录":
        quit()
        print("已注销")
    
#    elif msg == "检索事件":
#         getevents()

    elif str.isdigit("".join(re.findall(r'查询事件(\d+)',msg))):
         print(msg)
         eventid = "".join(re.findall(r'查询事件(\d+)',msg))
         print(eventid)
         itchat.send('识别到事件id，开始查询事件'+eventid,toUserName = revmsg['FromUserName'])
         getevent(eventid,revmsg['FromUserName'])      
    
    elif str.isdigit("".join(re.findall(r'确认事件(\d+)',msg))):
        eventid = "".join(re.findall(r'确认事件(\d+)',msg))
        #messages = "".join(re.findall(r'确认消息<(.+)>',msg))
        #有确认消息，不关闭问题
        if "".join(re.findall(r'(确认消息)<.+>',msg)) == "确认消息" and "".join(re.findall(r'【关闭】',msg)) == "":
            messages = "".join(re.findall(r'确认消息<(.+)>',msg))
            itchat.send("识别到事件id，开始确认事件"+eventid+"，确认消息为:"+messages,toUserName = revmsg['FromUserName'])
            ackevent(eventid,revmsg['FromUserName'],messages)
        #无确认消息，关闭问题    
        elif "".join(re.findall(r'(确认消息)<.+>',msg)) == "" and "".join(re.findall(r'【关闭】',msg)) == "【关闭】":
            itchat.send("识别到事件id，开始确认事件"+eventid+"，无确认消息。关闭问题。",toUserName = revmsg['FromUserName'])
            ackevent(eventid,revmsg['FromUserName'],action=1)
        #有确认消息，关闭问题
        elif "".join(re.findall(r'(确认消息)<.+>',msg)) == "确认消息" and "".join(re.findall(r'【关闭】',msg)) == "【关闭】":
            messages = "".join(re.findall(r'确认消息<(.+)>',msg))
            itchat.send("识别到事件id，开始确认事件"+eventid+"，确认消息为:"+messages+"。关闭问题。",toUserName = revmsg['FromUserName'])
            ackevent(eventid,revmsg['FromUserName'],messages,1)
        #无确认消息，不关闭问题。
        else:
            print("识别到事件id:"+eventid)
            itchat.send("识别到事件id，开始确认事件:"+eventid,toUserName = revmsg['FromUserName'])
            ackevent(eventid,revmsg['FromUserName'])      
    
    elif msg == "查询告警":
         gettrigger(revmsg['FromUserName'])
         print("查询问题触发器")
         
  #  elif msg == "告警":
  #       gaojing()
  #       print("执行告警")

    else:
        welcome = "您好，欢迎使用zabbix微信告警系统！你可以回复关键字\n(查询告警、确认事件、查询事件等)实现功能\n如：确认事件12345,确认消息<已解决>【关闭】（确认消息以“<>”为分隔符）\n如：查询事件1234\n更多功能正在研发中，敬请期待！"
        itchat.send(welcome,toUserName = revmsg['FromUserName'])
#######################################
#######################################
def quit():
    itchat.logout()
#######################################
def gaojing(fromuser):
    users = itchat.search_friends(name=u'Feiger')
    userName = users[0]['UserName']
    #登陆zabbix获取auth
    auth = authenticate(url, username, password)
    #状态0是启用监控，1是禁用监控
    status=1
    #定义操作ip
    hostip='192.168.15.139'
    #通过hostip获取zabbix 
    hostids=ipgetHostsid(hostip,url,auth)
    hostid=hostids[0]['hostid']
    alerts=actionidgetalert(url,auth)
    xiaoxi=str(alerts[0])
    itchat.send(xiaoxi,toUserName = userName)

def gettrigger(fromuser): #获取当前触发器
    auth = authenticate(url, username, password)
    trigerIDs = gettrigetID(url,auth)    #获取触发器列表
    for trigerIDresault in trigerIDs:    #遍历触发器
        eventlist = triggergetevents(trigerIDresault['triggerid'],url,auth)  #获取事件列表
        #for event in eventlist:        #遍历事件
        #   if event['acknowledged'] == '0' and event['value'] == '0':
        #       print(event)
        #       print(event['eventid'])
        #   else:
        #       continue  
    xiaoxi=str(trigerIDs)
    itchat.send(xiaoxi,toUserName = fromuser)

def ackevent(eventid,fromuser,message="已确认(微信默认消息)",action=0):  #确认事件
    auth = authenticate(url, username, password)
    #执行事件确认操作,并返回已确认事件的对象
    eventobj=eventackknowledge(eventid,url,auth,message,action)  #调取确认事件的方法
    if 'eventids' in eventobj: #eventobj这事件ID的字典
        itchat.send(str(eventobj['eventids'])+"事件确认成功",toUserName = fromuser)
    else:
        itchat.send("确认事件失败："+eventobj,toUserName = fromuser)

def getevent(eventid,fromuser):
    auth = authenticate(url, username, password)
    event=eventget(eventid,url,auth)  #查询事件
    itchat.send(str(event),toUserName = fromuser)

def getevents():   #通过时间获取事件
    auth = authenticate(url, username, password)
    users = itchat.search_friends(name=u'Feiger')
    userName = users[0]['UserName']
    events = timegetevents("1523030400","1523116800",url,auth)
    print(events)
    #xiaoxi=str(events[0])
    #itchat.send(xiaoxi,toUserName = userName)
    
#######################################
if __name__ == '__main__':
    itchat.auto_login(enableCmdQR=2)
    itchat.run()

2，zabbix登录逻辑

#!/usr/bin/env python
#coding:utf-8
import urllib.request, urllib.error, urllib.parse
import json
#定义URL账户密码
url = 'http://192.168.165.179:8027/api_jsonrpc.php'
username = 'admin'
password = 'zabbix'
#定义通过HTTP方式访问API地址的函数，后面每次请求API的各个方法都会调用这个函数
def requestJson(url,values):        
    data = json.dumps(values)
    data=bytes(data,'utf8')
    req = urllib.request.Request(url, data, {'Content-Type': 'application/json-rpc'})
    response = urllib.request.urlopen(req, data)
    output = json.loads(response.read())
    print(output)
    try:
        message = output['result']
    except:
        message = output['error']['data']
        print(message)

    return message

#API接口认证的函数，登录成功会返回一个Token
def authenticate(url, username, password):
    values = {'jsonrpc': '2.0',
              'method': 'user.login',
              'params': {
                  'user': username,
                  'password': password
              },
              'id': '0'
              }
    idvalue = requestJson(url,values)
    return idvalue

3，zabbixAPI，此段代码也是网上找到的，自己看了Zabbix的官方文档添加了些许功能。理论上来说可以通过微信触发关键字的方式实现一切对zabbix的操作。

#coding=utf-8 
import sys
import argparse
import urllib.request, urllib.error, urllib.parse
import json
from login import *
#定义检索问题状态下所有触发器的ID，名称和严重性，并按降序对其进行严重性排序。
def gettrigetID(url,auth):
    values = { 
        "jsonrpc" : "2.0" ,
        "method" : "trigger.get" ,
        "params" : { 
            "output" : [ 
                "triggerid" ,
                "description" ,
                "priority" 
                      ] ,
            "filter" : { 
            "value" : 1 
             } ,
        "sortfield" : "priority" ,
        "sortorder" : "DESC" 
    } ,
    "auth" : auth ,
    "id" : 1 
    }
    output = requestJson(url,values)
    return output
#触发器检索事件,返回一系列以事件为元素的列表。
def triggergetevents(triggerID,url,auth):
    values = {
    "jsonrpc": "2.0",
    "method": "event.get",
    "params": {
        "output": ["eventid","objectid"],
        "select_acknowledges":"extend",
        #"output": "extend",
        "objectids": triggerID,
        "sortfield": ["clock", "eventid"],
        "sortorder": "DESC"
    },
    "auth": auth,
    "id": 1
    }
    output = requestJson(url,values)
    return output
#按时间检索事件
def timegetevents(time_from,time_till,url,auth):
    values = {
    "jsonrpc": "2.0",
    "method": "event.get",
    "params": {
        "output": "extend",
        "time_from": time_from,
        "time_till": time_till,
        "sortfield": ["clock", "eventid"],
        "sortorder": "desc"
    },
    "auth": auth,
    "id": 1
    }
    output = requestJson(url,values)
    return output
#通过eventID查询事件
def eventget(eventid,url,auth):
    values = {
            "jsonrpc": "2.0",
            "method": "event.get",
            "params": {
                    "output": ["eventid","acknowledged","objectid"],
                    "select_acknowledges":["eventid","message","action","alias"],
                    "eventids": eventid,
                        },
            "auth": auth,
            "id": 1
            } 
    output = requestJson(url,values)
    return output
#定义确认事件方法

def eventackknowledge(eventid,url,auth,message="已确认(微信默认确认消息)",action=0):
   values = { 
            'jsonrpc' : '2.0',
            'method' : 'event.acknowledge',
            'params' : { 
                      'eventids' : eventid,
                      'message' : message,
                      'action' : action
                      },
            'auth' : auth,
            'id' : 1 
           }
   print("调用了eventacknowledge方法,正在确认事件：%s"%(eventid))
   output = requestJson(url,values)
   if 'eventids' in output:
        print("确认事件成功")
   else:
        print("确认事件失败")
   return output

#通过动作ID检索警报
def actionidgetalert(url,auth):
    values = {
    "jsonrpc": "2.0",
    "method": "alert.get",
    "params": {
        "output": "extend",
        "actionids": '7'
    },
    "auth":auth,
    "id": 1
    }
    output = requestJson(url,values)
    return output
#定义更新action函数
def mediatypeupdate(mediatypeid,status,auth):
    values = {'jsonrpc': '2.0',
              'method': 'mediatype.update',
              'params': {
                  "mediatypeid": mediatypeid,
                  "status": status
              },
              'auth': auth,
              'id': '1'
              }
    output = requestJson(url,values)
#定义读取状态函数
def triggerget(auth):
    values = {'jsonrpc': '2.0',
           "method":"trigger.get",
               "params": {
                        "output": [
                        "triggerid",
                        "description",
                        "priority"
                        ],
              "filter": {
                         "value": 1
                         },
              "expandData":"hostname",
              "sortfield": "priority",
              "sortorder": "DESC"
            },
              'auth': auth,
              'id': '2'
              }
    output = requestJson(url,values)
    return output

#定义通过ip获取主机id的函数
def ipgetHostsid(ip,url,auth):
    values = {'jsonrpc': '2.0',
              'method': 'host.get',
              'params': {
                  'output': [ "host" ], 
                  'filter': {
                      'ip': ip
                  },
              },
              'auth': auth,
              'id': '3'
              }
    output = requestJson(url,values)
    return output
    
#定义通过主机id获取开启关闭监控函数
def idupdatehost(status,hostid,url,auth):
    values = {'jsonrpc': '2.0',
              'method': 'host.update',
              'params': {
                  "hostid": hostid, 
                  "status": status
              },
              'auth': auth,
              'id': '4'
              }
    output = requestJson(url,values)
    return output
#定义通过项目hostid获取itemid函数
def getHostsitemsid(hostid,itemsname,url,auth):
    values = {'jsonrpc': '2.0',
              'method': "item.get",
              "params": {
                    "output": ["itemids"],
                    "hostids": hostid,
            "filter":{
                    "key_": itemsname,
                },
                },
            
              'auth': auth,
              'id': '5'
              }
    output = requestJson(url,values)
    if len(output)==0:
        return output
    else:
        return output[0]['itemid']


#定义通过项目id获取监控项目最近值信息的函数
def getHostsitemsvalue(itemid,url,auth):
    values = {'jsonrpc': '2.0',
              'method': "history.get",
              "params": {
                    "output": "extend",
                    "history":3,
                    "itemids":itemid,
                    "sortfield": "clock",
                    "sortorder": "DESC",
                    "limit":1,
                },
            
              'auth': auth,
              'id': '6'
              }
    output = requestJson(url,values)
    if len(output)==0:
        return output
    else:
        return output[0]["value"]
        
#定义更新读取状态action函数
def mediatypeget(mediatypeid,auth):
    values = {'jsonrpc': '2.0',
              'method': 'mediatype.get',
              'params': {
                "output": "extend",

              "filter": {
                        "mediatypeid":mediatypeid,
                         },
              },

              'auth': auth,
              'id': '7'
              }
    output = requestJson(url,values)
    if len(output)==0:
        return output
    else:
        return output[0]['status']

        
#定义maintenance维修模式host函数
def maintenancecreate(maintenancename,active_since,active_till,hostid,auth):
    values = {'jsonrpc': '2.0',
              'method': 'maintenance.create',
              'params': {
              "name": maintenancename,
              "active_since": active_since,
              "active_till": active_till,
              "hostids": [
                    hostid
                ],
                "timeperiods": [
                            {
                "timeperiod_type": 0,
                "every": 1,
                "dayofweek": 64,
                "start_time": 64800,
                "period": 3600
                            }
                                ]
              },
              'auth': auth,
              'id': '8'
              }
    output = requestJson(url,values)
    
#定义通过模糊获取关闭主机信息函数
def disabledhostget(url,auth):
    values = {'jsonrpc': '2.0',
              'method': 'host.get',
            "params": {
                "output": ["host"],
                'selectInterfaces': [ "ip" ],
                "filter": {
                    "status":1
        }
    },
              'auth': auth,
              'id': '9'
              }
    output = requestJson(url,values)
    return output

#定义maintenance维修模式group函数
def maintenancecreategroup(maintenancename,active_since,active_till,groupid,auth):
    values = {'jsonrpc': '2.0',
              'method': 'maintenance.create',
              'params': {
              "name": maintenancename,
              "active_since": active_since,
              "active_till": active_till,
              "groupids": [
                    groupid
                ],
                "timeperiods": [
                            {
                "timeperiod_type": 0,
                "every": 1,
                "dayofweek": 64,
                "start_time": 64800,
                "period": 3600
                            }
                                ]
              },
              'auth': auth,
              'id': '10'
              }
    output = requestJson(url,values)

#定义通过host groups named 获取groupid
def groupnameGroupid(groupname,auth):
    values = {'jsonrpc': '2.0',
              'method': 'hostgroup.get',
              "params": {
                    "output": "extend",
                    "filter": {
                        "name": [
                            groupname
                        ]
                    }
                },
              'auth': auth,
              'id': '11'
              }
    output = requestJson(url,values)
    return output

#定义模糊查询维护主机
def maintenanceget(url,auth):
    values = {'jsonrpc': '2.0',
              'method': 'maintenance.get',
              "params": {
                    "output": "extend",
                },
              'auth': auth,
              'id': '12'
              }
    output = requestJson(url,values)
    return output

#定义批量恢复处于维护主机
def maintenancedelete(maintenanceid,url,auth):
    values = {'jsonrpc': '2.0',
              'method': 'maintenance.delete',
              "params": [
                    maintenanceid
                ],
              'auth': auth,
              'id': '13'
              }
    output = requestJson(url,values)
    return output

#定义通过hostid获取graphid的函数
def getgraphid(hostid,graphname,url,auth):
        values = {'jsonrpc': '2.0',
                          'method': 'graph.get',
                          'params': {
                                  "output": "name",
                                  "hostids": hostid,
                                  "sortfield": "name",
                          'filter': {
                                        "name": graphname
                                  },

                          },
                          'auth': auth,
                          'id': '14'
                          }
        output = requestJson(url,values)
        return output

09:57:03