zabbix监控管理维护脚本

goodmm

　　zabbix监控的机器越来越多，新增机器很频繁，废弃的，每天都在手动同样的事情，写了个脚本自动维护，已经加入到MySQL初始化安装、机器下线、高可用等流程中，自动处理：
　　脚本如下：
#!/usr/bin/python
#coding:utf-8
import json
import urllib2
from urllib2 import URLError
import sys,argparse
class zabbix_api:
def __init__(self):
    self.url = 'http://172.17.61.52:8080/api_jsonrpc.php' #修改URL
    self.header = {"Content-Type":"application/json"}         
def user_login(self):
    data = json.dumps({
                       "jsonrpc": "2.0",
                       "method": "user.login",
                       "params": {
                                  "user": "Admin", #修改用户名
                                  "password": "wy-dba" #修改密码
                                  },
                       "id": 0
                       })
    request = urllib2.Request(self.url, data)
    for key in self.header:
        request.add_header(key, self.header[key])
    try:
        result = urllib2.urlopen(request)
    except URLError as e:
        print "\033[041m 用户认证失败，请检查 !\033[0m", e.code
    else:
        response = json.loads(result.read())
        result.close()
        #print response['result']
        self.authID = response['result']
        return self.authID
def host_get(self,hostName=''):
    data=json.dumps({
            "jsonrpc": "2.0",
            "method": "host.get",
            "params": {
                      "output": "extend",
                      "filter":{"host":hostName}
                      },
            "auth": self.user_login(),
            "id": 1
            })
    request = urllib2.Request(self.url,data)
    for key in self.header:
        request.add_header(key, self.header[key])
    try:
        result = urllib2.urlopen(request)
    except URLError as e:
        if hasattr(e, 'reason'):
            print 'We failed to reach a server.'
            print 'Reason: ', e.reason
        elif hasattr(e, 'code'):
            print 'The server could not fulfill the request.'
            print 'Error code: ', e.code
    else:
        response = json.loads(result.read())
        #print response
        result.close()
        print "主机数量: \033[31m%s\033[0m"%(len(response['result']))
        for host in response['result']:      
            status={"0":"OK","1":"Disabled"}
available={"0":"Unknown","1":"available","2":"Unavailable"}
#print host
if len(hostName)==0:
                print "HostID : %s\t HostName : %s\t Status :\033[32m%s\033[0m \t Available :\033[31m%s\033[0m"%(host['hostid'],host['name'],status[host['status']],available[host['available']])
else:
                print "HostID : %s\t HostName : %s\t Status :\033[32m%s\033[0m \t Available :\033[31m%s\033[0m"%(host['hostid'],host['name'],status[host['status']],available[host['available']])
return host['hostid']
def hostgroup_get(self, hostgroupName=''):
    data = json.dumps({
                       "jsonrpc":"2.0",
                       "method":"hostgroup.get",
                       "params":{
                                 "output": "extend",
                                 "filter": {
                                            "name": hostgroupName
                                            }
                                 },
                       "auth":self.user_login(),
                       "id":1,
                       })
    request = urllib2.Request(self.url,data)
    for key in self.header:
        request.add_header(key, self.header[key])
    try:
        result = urllib2.urlopen(request)
    except URLError as e:
        print "Error as ", e
    else:
        #print result.read()
        response = json.loads(result.read())
        result.close()
        #print response()
        for group in response['result']:
        if  len(hostgroupName)==0:
          print "hostgroup:  \033[31m%s\033[0m \tgroupid : %s" %(group['name'],group['groupid'])
else:
          print "hostgroup:  \033[31m%s\033[0m\tgroupid : %s" %(group['name'],group['groupid'])
           self.hostgroupID = group['groupid']
           return group['groupid']

def template_get(self,templateName=''):
    data = json.dumps({
                       "jsonrpc":"2.0",
                       "method": "template.get",
                       "params": {
                                  "output": "extend",
                                  "filter": {
                                             "name":templateName                                                        
                                             }
                                  },
                       "auth":self.user_login(),
                       "id":1,
                       })
    request = urllib2.Request(self.url, data)
    for key in self.header:
        request.add_header(key, self.header[key])
    try:
        result = urllib2.urlopen(request)
    except URLError as e:
        print "Error as ", e
    else:
        response = json.loads(result.read())
        result.close()
        #print response
        for template in response['result']:               
            if len(templateName)==0:
                print "template : \033[31m%s\033[0m\t  id : %s" % (template['name'], template['templateid'])
            else:
                self.templateID = response['result'][0]['templateid']
print "Template Name :  \033[31m%s\033[0m "%templateName
                return response['result'][0]['templateid']
def hostgroup_create(self,hostgroupName):
    if self.hostgroup_get(hostgroupName):
        print "hostgroup  \033[42m%s\033[0m is exist !"%hostgroupName
        sys.exit(1)
    data = json.dumps({
                      "jsonrpc": "2.0",
                      "method": "hostgroup.create",
                      "params": {
                      "name": hostgroupName
                      },
                      "auth": self.user_login(),
                      "id": 1
                      })
    request=urllib2.Request(self.url,data)
    for key in self.header:
        request.add_header(key, self.header[key])
    try:
        result = urllib2.urlopen(request)
    except URLError as e:
        print "Error as ", e
    else:
        response = json.loads(result.read())
        result.close()
        print "\033[042m 添加主机组:%s\033[0m  hostgroupID : %s"%(hostgroupName,response['result']['groupids'])

def host_create(self, hostip,hostname, hostgroupName, templateName):
     if self.host_get(hostname):
print "\033[041m该主机已经添加!\033[0m"
sys.exit(1)
    group_list=[]
    template_list=[]
    for i in hostgroupName.split(','):
        var = {}
        var['groupid'] = self.hostgroup_get(i)
        group_list.append(var)
    for i in templateName.split(','):
        var={}
        var['templateid']=self.template_get(i)
        template_list.append(var)
    data = json.dumps({
                       "jsonrpc":"2.0",
                       "method":"host.create",
                       "params":{
                                 "host": hostname,
                                 "interfaces": [
                                 {
                                 "type": 1,
                                 "main": 1,
                                 "useip": 1,
                                 "ip": hostip,
                                 "dns": "",
                                 "port": "10050"
                                  }
                                 ],
                               "groups": group_list,
                               "templates": template_list,
                                 },
                       "auth": self.user_login(),
                       "id":1                  
    })
    request = urllib2.Request(self.url, data)
    for key in self.header:
        request.add_header(key, self.header[key])
    try:
        result = urllib2.urlopen(request)
    except URLError as e:
        print "Error as ", e
    else:
        response = json.loads(result.read())
        result.close()
        print "添加主机 : \033[42m%s\031[0m \tid :\033[31m%s\033[0m" % (hostip, response['result']['hostids'])

def host_disable(self,hostip):
data=json.dumps({
"jsonrpc": "2.0",
"method": "host.update",
"params": {
"hostid": self.host_get(hostip),
"status": 1
},
"auth": self.user_login(),
"id": 1
})
request = urllib2.Request(self.url,data)
    for key in self.header:
        request.add_header(key, self.header[key])
    try:
        result = urllib2.urlopen(request)
    except URLError as e:
        print "Error as ", e
    else:
        response = json.loads(result.read())
        result.close()
        print '----主机现在状态------------'
print self.host_get(hostip)
def host_delete(self,hostid):
    hostid_list=[]
    #print type(hostid)
    for i in hostid.split(','):
        var = {}
        var['hostid'] = self.host_get(i)
        hostid_list.append(var)
    data=json.dumps({
"jsonrpc": "2.0",
"method": "host.delete",
"params": hostid_list,
            "auth": self.user_login(),
            "id": 1
            })
    request = urllib2.Request(self.url,data)
    for key in self.header:
        request.add_header(key, self.header[key])
    try:
        result = urllib2.urlopen(request)
    except Exception,e:
        print  e
    else:
        result.close()
       print "主机 \033[041m %s\033[0m  已经删除 !"%hostid

if __name__ == "__main__":
zabbix=zabbix_api()
parser=argparse.ArgumentParser(description='zabbix managmentapi ',usage='%(prog)s [options]')
parser.add_argument('-H','--host',nargs='?',dest='listhost',default='host',help='查询主机')
parser.add_argument('-G','--group',nargs='?',dest='listgroup',default='group',help='查询主机组')
parser.add_argument('-T','--template',nargs='?',dest='listtemp',default='template',help='查询模板信息')
parser.add_argument('-A','--add-group',nargs=1,dest='addgroup',help='添加主机组')
parser.add_argument('-C','--add-host',dest='addhost',nargs=4,metavar=('ip','hostname', '主机组1,主机组2', 'Template01,Template02'),help='添加主机,多个主机组或模板使用分号')
parser.add_argument('-d','--disable',dest='disablehost',nargs=1,metavar=('192.168.2.1'),help='禁用主机')
parser.add_argument('-D','--delete',dest='deletehost',nargs='+',metavar=('192.168.2.1'),help='删除主机,多个主机之间用分号')
parser.add_argument('-v','--version', action='version', version='%(prog)s 1.0')
if len(sys.argv)==1:
print parser.print_help()
else:
args=parser.parse_args()
if args.listhost != 'host' :
if args.listhost:
zabbix.host_get(args.listhost)
else:
zabbix.host_get()
if args.listgroup !='group':
if args.listgroup:
zabbix.hostgroup_get(args.listgroup)
else:
zabbix.hostgroup_get()
if args.listtemp != 'template':
if args.listtemp:
zabbix.template_get(args.listtemp)
else:
zabbix.template_get()
if args.addgroup:
zabbix.hostgroup_create(args.addgroup[0])
if args.addhost:
zabbix.host_create(args.addhost[0], args.addhost[1], args.addhost[2],args.addhost[3])
if args.disablehost:
zabbix.host_disable(args.disablehost)
if args.deletehost:
zabbix.host_delete(args.deletehost[0])　　执行：python 2.7以上，python xx.py -h 先看帮助。