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进行主机添加