zoukankan      html  css  js  c++  java
  • zabbix通过api 批量自动添加主机

    自动添加主机

    官网api: zabbix_api

    import requests
    import json
    import os
    
    # user config here
    ip = '10.0.0.1'
    user = "user"
    password = "pass."
    
    
    # user config end
    
    class ZabbixApi:
        def __init__(self, ip, user, password):
            self.url = 'https://' + ip + '/api_jsonrpc.php'
            self.headers = {"Content-Type": "application/json",
                            }
            self.user = user
            self.password = password
            self.auth = self.__login()
    
        def __login(self):
            '''
            zabbix登陆获取auth
            :return: auth  #  样例bdc64373690ab9660982e0bafe1967dd
            '''
            data = json.dumps({
                "jsonrpc": "2.0",
                "method": "user.login",
                "params": {
                    "user": self.user,
                    "password": self.password
                },
                "id": 10,
                # "auth": 'none'
            })
            try:
                response = requests.post(url=self.url, headers=self.headers, data=data, timeout=2)
                # {"jsonrpc":"2.0","result":"bdc64373690ab9660982e0bafe1967dd","id":10}
                auth = response.json().get('result', '')
                return auth
            except Exception as e:
                print("33[0;31;0m Login error check: IP,USER,PASSWORD33[0m")
                raise Exception
    
        def host_get(self, hostname=''):
            '''
            获取主机。不输入hostname 则获取所有
            :param hostName: zabbix主机名不允许重复。(IP允许重复)。#host_get(hostname='gateway1')
            :return: {'jsonrpc': '2.0', 'result': [], 'id': 21}
            :return: {'jsonrpc': '2.0', 'result': [{'hostid': '10277', ... , 'host': 'gateway', ...}], 'id': 21}
            '''
            if hostname == '':
                # print("no hostname and find all host")
                data = json.dumps({
                    "jsonrpc": "2.0",
                    "method": "host.get",
                    "params": {
                        "output": [
                            "hostid",
                            "host"
                        ],
                        "selectInterfaces": [
                            "interfaceid",
                            "ip"
                        ]
                    },
                    "id": 20,
                    "auth": self.auth
                })
            else:
                data = json.dumps({
                    "jsonrpc": "2.0",
                    "method": "host.get",
                    "params": {"output": "extend",
                               "filter": {"host": hostname},
                               },
                    "id": 21,
                    "auth": self.auth
                })
            try:
                response = requests.get(url=self.url, headers=self.headers, data=data, timeout=2)
                # hosts_count=len(response.json().get('result',''))
                # print('l',hosts)
                return response.json()  # len(ret.get('result'))为1时获取到,否则未获取到。
            except Exception as e:
                print("33[0;31;0m HOST GET ERROR33[0m")
                raise Exception
    
        def hostgroup_get(self, hostGroupName=''):
            '''
            获取主机组
            :param hostGroupName:
            :return: {'jsonrpc': '2.0', 'result': [{'groupid': '15', 'name': 'linux group 1', 'internal': '0', 'flags': '0'}], 'id': 1}
            '''
            data = json.dumps({
                "jsonrpc": "2.0",
                "method": "hostgroup.get",
                "params": {
                    "output": "extend",
                    "filter": {
                        "name": hostGroupName
                    }
                },
                "auth": self.auth,
                "id": 1,
            })
            try:
                response = requests.get(url=self.url, headers=self.headers, data=data, timeout=2)
                # hosts_count=len(response.json().get('result',''))
                # print('l',hosts)
                return response.json()  # len(ret.get('result'))为1时获取到,否则未获取到。
    
            except Exception as e:
                print("33[0;31;0m HOSTGROUP GET ERROR33[0m")
                raise Exception
    
        def hostgroup_create(self, hostGroupName=''):
            if len(self.hostgroup_get(hostGroupName).get('result')) == 1:
                print("无需创建,hostgroup存在")
                return 100
            data = json.dumps({
                "jsonrpc": "2.0",
                "method": "hostgroup.create",
                "params": {
                    "name": hostGroupName
                },
                "auth": self.auth,
                "id": 1
            })
            try:
                response = requests.get(url=self.url, headers=self.headers, data=data, timeout=2)
                # hosts_count=len(response.json().get('result',''))
                # print('l',hosts)
                return response.json()  # len(ret.get('result'))为1时获取到,否则未获取到。
    
            except Exception as e:
                print("33[0;31;0m HOSTGROUP CREATE ERROR33[0m")
                raise Exception
    
        def template_get(self, templateName=''):
            data = json.dumps({
                "jsonrpc": "2.0",
                "method": "template.get",
                "params": {
                    "output": "extend",
                    "filter": {
                        "name": templateName
                    }
                },
                "auth": self.auth,
                "id": 50,
            })
            try:
                response = requests.get(url=self.url, headers=self.headers, data=data, timeout=2)
                # hosts_count=len(response.json().get('result',''))
                # print('l',hosts)
                return response.json()  # len(ret.get('result'))为1时获取到,否则未获取到。
    
            except Exception as e:
                print("33[0;31;0m Template GET ERROR33[0m")
                raise Exception
    
        def host_create(self, hostname, hostip, hostGroupName, templateName):
            # 判断主机名是否重复
            find_hostname = self.host_get(hostname)
            if len(find_hostname.get('result')):
                print("###recheck### hostname[%s] exists and return" % hostname)
                return 1
    
            # 判断template是否存才,不存在退出。 否则初始化备用
            template_list = []
            for t in templateName.split(','):
                find_template = self.template_get(t)
                if not len(find_template.get('result')):
                    # template不存在退出 # 一般因为输错关系
                    print("###recheck### template[%s] not find and return " % t)
                    return 1
    
                tid = self.template_get(t).get('result')[0]['templateid']
                template_list.append({'templateid': tid})
    
            # 判断hostgroup是否存在。 不存在则创建。 并
            hostgroup_list = []
            for g in hostGroupName.split(','):
                find_hostgroupname = self.hostgroup_get(g)
                if not len(find_hostgroupname.get('result')):
                    # hostgroupname 不存在,创建
                    self.hostgroup_create(g)
                gid = self.hostgroup_get(g).get('result')[0]['groupid']
                hostgroup_list.append({'groupid': gid})
    
    
            data = json.dumps({
                "jsonrpc": "2.0",
                "method": "host.create",
                "params": {
                    "host": hostname,
                    "interfaces": [
                        {
                            "type": 1,
                            "main": 1,
                            "useip": 1,
                            "ip": hostip,
                            "dns": "",
                            "port": "10050",
    
                        }
                    ],
                    "groups": hostgroup_list,
                    "templates": template_list,
                    "proxy_hostid": 13404,  # 我这使用了代理,不需要代理 去掉函数就可以
                },
                "auth": self.auth,
                "id": 1
            })
    
            try:
                response = requests.get(url=self.url, headers=self.headers, data=data, timeout=2)
                # hosts_count=len(response.json().get('result',''))
                # print('l',hosts)
                print("执行返回信息: 添加HOST[%s,%s]完成" % (hostname, hostip))
                return response.json()  # len(ret.get('result'))为1时获取到,否则未获取到。
    
            except Exception as e:
                print("33[0;31;0m HOST CREATE ERROR33[0m")
                raise Exception
    
    
    if __name__ == '__main__':
        try:
            zabbix = ZabbixApi(ip, user, password)
            print("...login success...")
    
            # 批量添加主机,从文本add_host.txt
            BASE_DIR = os.path.dirname(os.path.abspath(__file__))
            ADD_LIST = os.path.join(BASE_DIR, 'add_host.txt')
            with open(ADD_LIST, 'r', encoding='utf-8') as f:
                for line in f:
                    if len(line.strip()):  # 跳过空行
                        serverinfo = line.strip().split('#')
                        # 根据自己需要进行修改
                        zabbix.host_create(serverinfo[0], serverinfo[1], serverinfo[2], serverinfo[3])
    
        except Exception as e:
            print(e)
    
    
    
    更改为txt格式
    主机 ip 群组 代理
    host1 #10.1.1.1# #test# #proxy1
    host2 #10.1.1.1# #test2# #proxy2

    host#ip#gropup#proxy

    转:https://www.jianshu.com/p/e087cace8ddf

  • 相关阅读:
    Mina Core 10-执行器过滤器
    Mina Core 09-编解码过滤器
    Mina Core 08-IoBuffer
    Mina Basics 07-处理程序Handler
    Mina Basics 06-传输
    Mina Basics 05-过滤器
    Mina Basics 04- 会话
    Mina Basics 03-IoService
    Mina Basics 02-基础
    Mina Basics 01- 入门
  • 原文地址:https://www.cnblogs.com/liufukui/p/13158891.html
Copyright © 2011-2022 走看看