码迷,mamicode.com
首页 > 其他好文 > 详细

zabbix批量添加主机

时间:2019-02-15 09:14:25      阅读:174      评论:0      收藏:0      [点我收藏+]

标签:lte   xlsx   rpc   size   listt   [1]   update   必须   could not   

zabbix通过excel批量添加主机
本次批量添加针对监控**网络设备**,监控介质通过snmp方式。

不需要再一次一次添加每个网络设备。
只需要编写excel表格,即可一键添加监控。

安装excel python模块xlrd

yum -y install epel-release
yum -y install python-pip
pip install --upgrade pip
pip install xlrd

准备excel表格

表格列的信息依次为:
主机名,主机显示名,IP,主机所属主,主机模板组,snmp团体名

技术图片

上传excel表格到服务器

表格名字为:add-host-2.xlsx

技术图片

脚本内容如下

#!/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‘ #修改zabbix服务器ip地址,或者域名
            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, temp_count):
            count_add_host = temp_count     # 用于记录添加excel主机的数量
            if self.host_get(hostip):
                print "\033[041m该主机已经添加! host_create_andy\033[0m"
            else:
                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,
                                            "macros": [
                                                    {
                                                        "macro": "{$SNMP_COMMUNITY}",
                                                        "value": ‘hc360tc‘
                                                    }
                                                ]
                                             },
                                   "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()
                    count_add_host +=1
                    print "添加主机%s : \033[42m%s\033[0m \tid :\033[31m%s\033[0m" % (count_add_host, hostip, response[‘result‘][‘hostids‘])
                    return count_add_host

        def host_create(self, hostip, hostgroupName, templateName):
            if self.host_get(hostip):
                print "\033[041m该主机已经添加! host_create\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(‘add-host-2.xlsx‘)     #Excel表名
                    count = 0   # 用于记录添加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

                        count = zabbix.host_create_andy(hostname,visible,hostip,hostgroup, hosttemp, count)

                    print "通过excel一共添加主机 : \033[42m%s\033[0m台!" % (count)

注意:
必须修改脚本三个部分内容:

1. ‘http://zabbix服务器的IP/zabbix/api_jsonrpc.php‘ #修改zabbix服务器ip地址,或者域名
2. "user": "admin", # 修改web页面登录用户名
3. "password": "zabbix" # 修改web页面登录密码

运行脚本批量添加主机

hong6.py文件为脚本文件名。
运行脚本:python hong6.py -L 

技术图片

同时脚本也支持:
python hong6.py -H 查看所有主机
python hong6.py -H --host 查看单个主机,如:python hong6.py -H --host 172.16.8.10
python hong6.py -G 查看所有主机组
python hong6.py -T 查看所有临时模板
python hong6.py -A 添加主机组
python hong6.py -C 添加多个主机或者多个模板
python hong6.py -d 禁用主机
python hong6.py -L 批量添加excel主机
python hong6.py -D 批量删除主机

参考:https://www.cnblogs.com/momoshouhu/p/8053907.html
zabbix api:https://www.zabbix.com/documentation/4.0/manual/api

zabbix批量添加主机

标签:lte   xlsx   rpc   size   listt   [1]   update   必须   could not   

原文地址:http://blog.51cto.com/506554897/2350269

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!