zabbix api批量添加主机

renran421

　　#!/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://zabbix.efunfun.test/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": "efunfun5tgb" #修改密码
　　},
　　"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, hostgroupName, templateName):
　　if self.host_get(hostip):
　　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": hostip,
　　"interfaces": [
　　{
　　"type": 1,
　　"main": 1,
　　"useip": 1,
　　"ip": "127.0.0.1",
　　"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  api ',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=3,metavar=('192.168.2.1', 'test01,test02', '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])
　　if args.disablehost:
　　zabbix.host_disable(args.disablehost)
　　if args.deletehost:
　　zabbix.host_delete(args.deletehost[0])