调用zabbix 分组api

Posted 2021-12-31 箐惜

调用zabbix 分组api，获取分组中主机host信息，并分类保存，

#!/usr/bin/env python
#coding:utf8
import requests
import json
import os
#定义url头部信息
headers = Content-Type: application/json-rpc
#zabbix地址
server_ip = 10.37.149.109
#zabbix url
url = http://%s/zabbix/api_jsonrpc.php %server_ip
#获取token
def getToken(username,passwd):

    #url = http://%s/zabbix/api_jsonrpc.php%server_ip

    #headers = Content-Type: application/json-rpc

    username = admin
    passwd = zabbix
    data = 
           "jsonrpc": "2.0",
           "method": "user.login",
           "params": 
           "user": username,
           "password": passwd
           ,
           "id": 0
       
          

    request = requests.post(url=url,headers=headers,data=json.dumps(data))
    dict = json.loads(request.text)
    return dict[result]
# 获取groupID
def getGroupid(token_num):
    group = 
    data = "jsonrpc": "2.0",
             "method":"hostgroup.get",
             "params":
             "output":["groupid","name"]
             ,
             "auth":token_num,
             "id": 0
               

    request = requests.post(url=url,headers=headers,data=json.dumps(data)) 
    dict = json.loads(request.content)
    for i in  dict[result]:
        groupid = i[groupid]
        name = i[name]
        group[name] = groupid
    return group 
# 获取group中的hosts信息
def getHost(token_num,group):
    dict_group =   
    for k in group.keys():
        groupid = group[k] 
        data = 
               "jsonrpc":"2.0",
               "method":"host.get",
               "params":
               "output":["hostid","name"],
           "selectInterfaces": [
               "interfaceid",
                "ip"
                 ],
                "groupids":groupid
                 ,
               "auth":token_num,
               "id":1,
               
        request = requests.post(url=url,headers=headers,data=json.dumps(data)) 
        hosts = json.loads(request.content)
        hosts1 = hosts[result]
        ip_lists = getProc(hosts1)
        dict_group[k]=ip_lists
   
    return dict_group
      
# host ip排序
def getProc(hosts):
    list = hosts
    ip_lists = [ ]
    if list:
        for i in list:
            host = i[name]
            inter = i[interfaces]
            for j in inter:
            
                ip = j[ip]
                ip_lists.append(ip)
                ip_lists.sort(lambda x,y: cmp(.join( [ i.rjust(3, 0) for i in x.split(.)] ), .join( [ i.rjust(3, 0) for i in y.split(.)] ) ))
    return ip_lists

# 进行ip组分类（同网段为一组） 
def splitIp(ip):
    ip_split = ip.split(.)
    ip_class = ip_split[0:-1]
        
    return ip_class

# 进行ip组分类，    
def getIp(dict_group):
    dicGroupIp = 
    dicIp = 
    ip_l = []
    for group,ip_list in dict_group.items():
        for ip in ip_list:
            
            ip_class = splitIp(ip)
            ip_group =..join(ip_class)
            if ip_group not in dicIp:
                ip_l.append(ip)
                dicIp[ip_group] = ip_l
                
            else:
                dicIp[ip_group].append(ip)
            ip_l = [ ]

        dicGroupIp[group] = dicIp
        dicIp =  
    return dicGroupIp        
    
# 分别创建group组文件将ip地址写入。
def writrIpfile(dicGroupIp):
     
    for group_class,group_dic in dicGroupIp.items():
        ip_path = os.path.join(/python/zabbix2,group_class )
        
        with open(ip_path,w)as fd:
            for ip_class,ip_list in group_dic.items():
                ip_class = [ + ip_class + ]
            #with open(ip_path,a)as fd:
                fd.write(ip_class + \\n)
                for i in ip_list:
                    
                    fd.write(i + \\n)
        
 

if __name__ == "__main__":
    #server_ip = zmonitor.enn.cn
    username = admin
    passwd = zabbix
    token_num = getToken(username,passwd)
    group = getGroupid(token_num)
    dict_group = getHost(token_num,group)
    dicIp = getIp(dict_group)
    writrIpfile(dicIp)