[zabbix/使用API进行批量创建screen(聚合图形)]

####因公司需求，需要根据主机名称(hostname)模糊匹配查询出来,进行批量添加聚合图形(网卡流量监控)

• 试着在网上找案例,发现均是代码 潦潦草草 复制一大片,无任何重点
• 官方文档案例提笔带过
• 官方文档 网上 翻了个遍 找到了重点

思路点

• 需要从zabbix 数据库里面匹配传递进来的hostname模糊参数 进行匹配
• zabbix api hosts.get 返回所有从数据库里面返回出来的hostsid
• 用返回的hostsid 请求 zabbix api graph.get(图形模版) 来匹配graphid 返回出来的name为模版名称(这个很重要)
• 通过返回出来的graphid 再去请求 zabbix api screen.add创建聚合图形
• 至此，思路已经通了,和需求正是一样 下面开始 详解代码

当前目录下创建三个.py文件

QuerySet.py ### 查询数据库返回host主机
parame.py   ### 请求的参数以及 mysql用户,密码 zabbixy用户,密码
screen_api.py      ### 请求内容,返回结果

1: parame.py:

# -*- coding:utf-8 -*-

mysql_user  = 'root'
mysql_host  = 'localhost'
mysql_pass  = 'your MySQL Password'
zabbix_user = 'your Zabbix user'
zabbix_pass = 'your zabbix Password'

host_get = {
    "jsonrpc": "2.0",
    "method": "host.get", ###此处为请求主机返回hostsid 的请求参数,当前为空,后期会append 进去
    "params": {
        "filter": {
            "host": [
            ]
        }
    },
    "id": 1
}

graph_get = {
    "jsonrpc": "2.0",
    "method": "graph.get", ###获取主机图形的方法
    "params": {
        "output": "extend",
        "hostids":[],
        "sortfield": "name"
    },
    "id": 1
}

screen_add = {
    "jsonrpc": "2.0",
    "method": "screen.create",###创建图形方法
    "params": {
        "hsize": 2,
        "vsize": 60,
        "screenitems": [
        ]
    },
    "id": 1
}

2: QuerySet.py:

#!/usr/bin/env python
# -*- coding:utf-8 -*-
import pymysql as mysql
import sys
import re
import parame
def kwargs(args):   ###连接数据库
    conn = mysql.connect(
        host = parame.mysql_host,    ###parame文件里面定义的host
        port = 3306,
        user = parame.mysql_user,    ###parame文件里面定义的user
        passwd = parame.mysql_pass,  ###parame文件里面定义的passwd
        db = 'zabbix',
        )
    cur = conn.cursor()
    ###模糊查询传递进来的参数例如 cmb-he-sjz1-123-34-5-
    response = cur.execute('select host from hosts where host like %s',args)
    result1 = [ i for i in cur.fetchmany(response)]
    if len(result1) == 0:
        return False
    cur.close()
    conn.commit()
    conn.close()
    result = []
    ###此处可以省略，因公司需求,名称前面带VIP的主机名不需要监控
    for x in result1:
        if re.match('VIP',x[0]):
            pass
        else:
            result.append(x)
     ###排序,这块可以无视,将数据库查询出来的主机排序从小到大,一条命令就可以搞定
     ###sorted(hosts, key=lambda x: int(x.split("-")[6]))
     ### 因为每个公司的主机名定义不一致,这块大家 如果没有排序需求 删掉就可以,直接return result       
    after = [re.search('(.*?)-(.*?)-(.*?)-(.*?)-(.*?)-(.*?)-(.*?)',x[0]).group() for x in result]
    before = [int(y[0].split('-')[6]) for y in result]
    before.sort()
    response = []
    z = 0
    for result in after:
        response.append(result + str(before[z]))
        z += 1
    return response

if __name__ == '__main__':
    try:
        params = sys.argv[1]
        print(kwargs(params + '%'))
    except Exception as e:
        raise TypeError

2: screen_api.py:

#!/usr/bin/env python
# -*- coding:utf-8 -*-

import sys
import json
import time
import requests
import parame
import sys
import re
from QuerySet import kwargs

class Api:
    def __init__(self):
        self.url = 'http://your addres /api_jsonrpc.php'
        self.headers = {'Content-Type': 'application/json'}
        auth = {
            "jsonrpc": "2.0",
            "method": "user.login",
            "params": {
                "user": parame.zabbix_user, ###验证
                "password":parame.zabbix_pass
            },
            "id": 1,
            "auth":None,
        }
        response = requests.post(self.url,data=json.dumps(auth),headers=self.headers)
        self.authid = json.loads(response.text)['result'] ### auth的id

    def get_host(self,hosts): #数据库传递进来的主机参数 一大堆～
        data = parame.host_get
        data['auth'] = self.authid ###将返回id 写进去 接口会验证这个唯一
        if hosts: ### 判断是否为空 为空说明没有找到主机 直接退出 
            pass
        else:
            print("result not found")
            sys.exit(10)
        hosts = sorted(hosts, key=lambda x: int(x.split("-")[6])) ###这块是排序 QuerySet 已经做过了 没必要再写 
        for host in hosts:
            data['params']['filter']['host'].append(host) 将主机加入参数里面
        response = requests.post(self.url,data=json.dumps(data),headers=self.headers)
        lists =  json.loads(response.text)['result']
        result = [ lists[i]['hostid'] for i in range(len(lists)) ]
        return result ###返回主机hostsid 

    def graph_get(self,res): ###请求主机图形结果
        hosts_id = self.get_host(res) ### 讲hostsid 赋值给hosts_id 
        data = parame.graph_get
        data['auth'] = self.authid
        for hosts in hosts_id:
            data['params']['hostids'].append(hosts)
        response = requests.post(self.url,data=json.dumps(data),headers=self.headers)
        graph_id  = json.loads(response.text)['result']
        result = []
        for index in range(len(graph_id)): ###循环遍布判断是否是网卡图形,需求不同意此处可以不用re 你们直接 append 就好
            if re.search('Network traffic on em.',graph_id[index]['name']):
                result.append(graph_id[index]['graphid'])
            elif re.search('eth',graph_id[index]['name']):
                result.append(graph_id[index]['graphid'])
            else:
                pass
        return result

    def screen_add(self,graph_ids,screen_name): ###创建图形
        response = self.graph_get(graph_ids)
        x =  0 ### x为行 此处的定义 取决于parame.screen_add 里面的hsize 和vsize 最多为 hsize - 1 or vsize -1 
        y =  0 ### y为列 
        parame.screen_add['params']['name'] = screen_name
        parame.screen_add['auth'] = self.authid
        for value in response:
            parame.screen_add['params']['screenitems'].append({
                "resourcetype": 0,
                "resourceid": value,
                "width": "500", ###宽高度
                "height": "100",
                "rowspan": 1,
                "colspan": 1,
                "x": x,
                "y": y
            })
            if x == 1: ###此处判断  我们需求只需要两行一列 等于1 x为0 
                x  = 0
                y += 1
            else:
                x += 1
            time.sleep(0.01)
        response = requests.post(self.url,data=json.dumps(parame.screen_add),headers=self.headers)
        print(response.text)

if __name__ == '__main__':
    try:
        func = Api()
        res = kwargs("%" + str(sys.argv[4]) + "%") ###数据库传递的参数
        func.screen_add(res,str(sys.argv[2]))  ### 聚合图形群组名

    except Exception as e:
        print(e)
        print('--name 聚合图形组名 --hostname 匹配加入的主机')
        sys.exit(1)

##展示结果

此刻正是我们需要的结果，有问题 欢迎在下方留言 QQ:1301927919