python zabbix批量添加主机 脚本
转载:https://www.cnblogs.com/momoshouhu/p/8053907.html
1.安装xlrd读取Excel文件
1.1. 下载setuptools-38.2.4.zip,上传至zabbix服务器解压安装,下载地址:https://pypi.python.org/packages/69/56/f0f52281b5175e3d9ca8623dadbc3b684e66350ea9e0006736194b265e99/setuptools-38.2.4.zip#md5=e8e05d4f8162c9341e1089c80f742f64
[root@localhost temp]# ll #上传文件setuptools-38.2.4.zip
total 724
-rw-r--r-- 1 root root 736683 Dec 17 22:37 setuptools-38.2.4.zip
[root@localhost temp]# unzip setuptools-38.2.4.zip #解压setuptools-38.2.4.zip
[root@localhost temp]# cd setuptools-38.2.4 #进入setuptools-38.2.4目录
[root@localhost temp]# cd setuptools-38.2.4 #安装
1.2下载pip-9.0.1.tar.gz,上传至zabbix服务器解压安装,下载地址:https://pypi.python.org/packages/11/b6/abcb525026a4be042b486df43905d6893fb04f05aac21c32c638e939e447/pip-9.0.1.tar.gz#md5=35f01da33009719497f01a4ba69d63c9
[root@localhost temp]# ll
total 1896
-rw-r--r-- 1 root root 1197370 Dec 17 22:53 pip-9.0.1.tar.gz #上传文件pip-9.0.1.tar.gz
drwxr-xr-x 9 root root 4096 Dec 17 22:55 setuptools-38.2.4
-rw-r--r-- 1 root root 736683 Dec 17 22:37 setuptools-38.2.4.zip
[root@localhost temp]# tar -xzvf pip-9.0.1.tar.gz #解压pip-9.0.1.tar.gz
[root@localhost pip-9.0.1]# python setup.py build #编译
[root@localhost pip-9.0.1]# python setup.py install #安装
[root@localhost pip-9.0.1]# pip install xlrd #安装xlrd
3.编辑Excel模板
3.1主机名、显示名、IP、主机组、模板
3.2将Execle表zabbix_host_add.xlsx上传至zabbix服务器
[root@localhost temp]# ll
total 1964
-rw-r--r-- 1 root root 46079 Dec 17 23:28 zabbix_host_add.xlsx
4.编写Python脚本,参考http://www.361way.com/zabbix-api-2/3681.html
#!/usr/bin/python
#coding:utf-8
import json
import urllib2
from urllib2 ``import URLError
import sys,argparse
import xlrd
defaultencoding ``= 'utf-8'
if sys.getdefaultencoding() !``= defaultencoding:
reload``(sys)
sys.setdefaultencoding(defaultencoding)
class zabbix_api:
def __init__(``self``):
self``.url ``= 'http://zabbix服务器IP地址/zabbix/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"``, ``#web页面登录用户名
"password"``: ``"zabbix" #web页面登录密码
},
"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_andy(``self``,hostName,visibleName, 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"``: hostName,
"name"``: visibleName,
"interfaces"``: [
{
"type"``: ``2``, ``#1:表示IP;2表示SNMP
"main"``: ``1``,
"useip"``: ``1``,
"ip"``: hostip,
"dns"``: "",
"port"``: ``"161" #IP端口10051;SNMP端口161
}
],
"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_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"``: ``2``,
"main"``: ``1``,
"useip"``: ``1``,
"ip"``: hostip,
"dns"``: "",
"port"``: ``"161"
}
],
"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(``'-L'``,``'--allin'``,dest``=``'allin'``,nargs``=``'?'``,default``=``'allin'``,``help``=``'从Excel批量导入主机'``)
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``])
if args.allin !``= 'allin'``:
workbook ``= xlrd.open_workbook(``'zabbix_host_add.xlsx'``) ``#Excel名
for row ``in xrange``(workbook.sheets()[``0``].nrows):
hostname ``= workbook.sheets()[``0``].cell(row, ``0``).value
visible ``= workbook.sheets()[``0``].cell(row, ``1``).value
hostip ``= workbook.sheets()[``0``].cell(row, ``2``).value
hostgroup ``= workbook.sheets()[``0``].cell(row, ``3``).value
hosttemp ``= workbook.sheets()[``0``].cell(row, ``4``).value
zabbix.host_create_andy(hostname,visible,hostip,hostgroup, hosttemp)
|
上面红色标注,请根据实际情况修改。
上传Python脚本zabbix_host.py至zabbix服务器
[root@localhost temp]# ll
total 1964
-rwxr-xr-x 1 root root 14644 Dec 17 23:28 zabbix_host.py
[root@localhost temp]# chown zabbix:zabbix zabbix_host.py #修改属组属主
[root@localhost temp]# chmod +x zabbix_host.py #添加执行权限
[root@localhost temp]# python zabbix_host.py -L #执行zabbix_host.py进行主机添加