如何利用zabbix api 来增加监控

#!/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://localhost/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": "zabbix" #修改密码
},
"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 "33[041m 用户认证失败，请检查 !33[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 "主机数量: 33[31m%s33[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 : %st hostname : %st status :33[32m%s33[0m t available :33[31m%s33[0m"%(host['hostid'],host['name'],status[host['status']],available[host['available']])
else:
print "hostid : %st hostname : %st status :33[32m%s33[0m t available :33[31m%s33[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: 33[31m%s33[0m tgroupid : %s" %(group['name'],group['groupid'])
else:
print "hostgroup: 33[31m%s33[0mtgroupid : %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 : 33[31m%s33[0mt id : %s" % (template['name'], template['templateid'])
else:
self.templateid = response['result'][0]['templateid']
print "template name : 33[31m%s33[0m "%templatename
return response['result'][0]['templateid']
def hostgroup_create(self,hostgroupname):
if self.hostgroup_get(hostgroupname):
print "hostgroup 33[42m%s33[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 "33[042m 添加主机组:%s33[0m hostgroupid : %s"%(hostgroupname,response['result']['groupids'])
def host_create(self, hostip, hostgroupname, templatename):
if self.host_get(hostip):
print "33[041m该主机已经添加!33[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": 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 "添加主机 : 33[42m%s31[0m tid :33[31m%s33[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 "主机 33[041m %s33[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])

直接执行 python zabbix_api.py
usage: zabbix_api.py [options]
zabbix api
optional arguments:
-h, --help show this help message and exit
-h [listhost], --host [listhost]
查询主机
-g [listgroup], --group [listgroup]
查询主机组
-t [listtemp], --template [listtemp]
查询模板信息
-a addgroup, --add-group addgroup
添加主机组
-c 192.168.2.1 test01,test02 template01,template02, --add-host 192.168.2.1 test01,test02 template01,template02
添加主机,多个主机组或模板使用分号
-d 192.168.2.1, --disable 192.168.2.1
禁用主机
-d 192.168.2.1 [192.168.2.1 ...], --delete 192.168.2.1 [192.168.2.1 ...]
删除主机,多个主机之间用分号
-v, --version show program's version number and exit

#!/usr/bin/python
#新增帮助信息，可直接执行脚本
zabbix=zabbix_api()
#获取所有主机列表
zabbix.host_get()
#查询单个主机列表
zabbix.host_get('192.168.2.1')
#获取所有主机组列表
zabbix.hostgroup_get()
#查询单个主机组列表
zabbix.hostgroup_get('test01')
#获取所有模板列表
zabbix.template_get()
#查询单个模板信息
zabbix.template_get('template os linux')
#添加一个主机组
zabbix.hostgroup_create('test01')
#添加一个主机，支持将主机添加进多个组，多个模板，多个组、模板之间用逗号隔开，如果添加的组不存在，新创建组
zabbix.host_create('192.168.2.1', 'test01', 'template os linux')
zabbix.host_create('192.168.2.1', 'linux servers,test01 ', 'template os linux,template app mysql')
#禁用一个主机
zabbix.host_disable('192.168.2.1')
#删除host,支持删除多个，之间用逗号
zabbix.host_delete('192.168.2.1,192.168.2.2')