zabbix调用api添加主机

Posted 2021-06-25 linyukk

# /usr/bin/env python3
# -*- coding:utf-8 -*-
import requests
import json
import os

ip = \'172.22.25.50\'
user = \'Admin\'
password = \'N1i2H3a,o.\'

 

class ZabbixApi:
    def __init__(self, ip, user, password):
        self.url = \'http://\' + ip + \'/api_jsonrpc.php\'  # \'http://ip/zabbix/api_jsonrpc.php\'
        self.headers = {"Content-Type": "application/json",
                        }
        self.user = user
        self.password = password
        self.auth = self.__login()

    def __login(self):
        \'\'\'
        zabbix登陆获取auth
        \'\'\'
        data = json.dumps({
            "jsonrpc": "2.0",
            "method": "user.login",
            "params": {
                "user": self.user,
                "password": self.password
            },
            "id": 10,
        })
        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("\\033[0;31;0m Login error check: IP,USER,PASSWORD\\033[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("\\033[0;31;0m HOST GET ERROR\\033[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("\\033[0;31;0m HOSTGROUP GET ERROR\\033[0m")
            raise Exception

    def hostgroup_create(self,hostGroupName=\'\'):
        if len(self.hostgroup_get(hostGroupName).get(\'result\'))==1:
            print("群组%s已存在" % hostGroupName)
            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)
            print("群组%s创建成功" % hostGroupName)
            return response.json()   # len(ret.get(\'result\'))为1时获取到，否则未获取到。

        except Exception as e:
            print("\\033[0;31;0m HOSTGROUP CREATE ERROR\\033[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("\\033[0;31;0m Template GET ERROR\\033[0m")
            raise Exception

    def host_create(self,hostname,ip,groupname,templatename):
        #校验主机名是否存在
        if len(self.host_get(hostname).get(\'result\')) == 1:
            print("主机%s已存在" % hostname)
            return 100
        #校验主机群组是否存在，不存在添加,多个群组已\',\'分割
        hostgroup_list = []
        for g in groupname.split(\',\'):
            find_hostgroupname = self.hostgroup_get(g)
            if not len(find_hostgroupname.get(\'result\')):
                print(find_hostgroupname.get(\'result\'))
                # hostgroupname 不存在，创建
                self.hostgroup_create(g)
            gid = self.hostgroup_get(g).get(\'result\')[0][\'groupid\']
            hostgroup_list.append({\'groupid\': gid})
            #print(hostgroup_list)
        #校验模板是否存在，多个模板已\',\'分割，不存在请检查
        template_list = []
        for t in templatename.split(\',\'):
            find_template = self.template_get(t)
            if not len(find_template.get(\'result\')):
                # template不存在退出 # 一般因为输错关系
                print(find_template.get(\'result\'))
                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})
            #print(template_list)
        #添加zabbix-agent
        data_agent = json.dumps({
            "jsonrpc": "2.0",
            "method": "host.create",
            "params": {
                "host": hostname,
                "interfaces": [
                    {
                        "type": 1,
                        "main": 1,
                        "useip": 1,
                        "ip": ip,
                        "dns": "",
                        "port": "10050"
                    }
                ],
                "groups": hostgroup_list,
                "templates": template_list
            },
            "auth": self.auth,
            "id": 1
        })
        #添加snmp监控
        data_snmp = json.dumps({
            "jsonrpc": "2.0",
            "method": "host.create",
            "params": {
                "host": hostname,
                "interfaces": [
                    {
                        "type": 2,
                        "main": 1,
                        "useip": 1,
                        "ip": ip,
                        "dns": "",
                        "port": "161",
                        "details": {
                            "version": 2,
                            "bulk": 1,
                            "securityname": "mysecurityname",
                            "contextname": "",
                            "securitylevel": 1,
                            \'community\': \'{$SNMP_COMMUNITY}\'
                            }
                    }
                ],
                "groups": hostgroup_list,
                "templates": template_list
            },
            "auth": self.auth,
            "id": 1
        })
        try:
            data=data_snmp
            response = requests.get(url=self.url, headers=self.headers, data=data, timeout=2)
            hosts=len(response.json().get(\'result\',\'\'))
            hostid1=response.json().get(\'result\',\'\')
            hostid=hostid1[\'hostids\']
            #print(type(hostid1),type(hostid))
            print("执行返回信息： 添加HOSTNAME:%s,IP:%s,HOSTID:%s完成"%(hostname,ip,hostid))
            return response.json()  # len(ret.get(\'result\'))为1时获取到，否则未获取到。

        except Exception as e:
            print("\\033[0;31;0m HOST CREATE ERROR\\033[0m")
            raise Exception

if __name__ == \'__main__\':
    zabbix=ZabbixApi(ip,user,password)
    #查询单台
    #host=zabbix.host_get(\'192.168.56.153\')
    #添加单台
    #zabbix.host_create(\'host1\',\'192.168.56.152\',\'dell\',\'Template Dell idrac\')
    #批量添加主机，从同目录下add_hosts文件,文件格式为hostname # ip # 主机群组 #模板
    BASE_DIR = os.path.dirname(os.path.abspath(__file__))
    ADD_LIST = os.path.join(BASE_DIR, \'add_hosts\')
    print(ADD_LIST)
    with open(ADD_LIST, \'r\', encoding=\'utf-8\') as f:
        for line in f:
            if len(line.strip()):  # 跳过空行
                serverinfo = line.strip().split(\'#\')
                #print(serverinfo[0].strip("\'"),serverinfo[1].strip("\'"),serverinfo[2].strip("\'"),serverinfo[3].strip("\'"))
                zabbix.host_create(serverinfo[0].strip(), serverinfo[1].strip(), serverinfo[2].strip(), serverinfo[3].strip())