调用zabbix 分组api

调用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, '') for i in x.split('.')] ), ''.join( [ i.rjust(3, '') 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)