需求:对所有云主机添加公网带宽监控,但不是每一台主机都有公网IP,有公网IP的主机的带宽也不完全相同,比如有的是1M,有的是5M,带宽监控的阈值为带宽值*0.8*1000K。例如1M的带宽,监控的阈值为800K。(注意:经过测试,在zabbix的trigger中这是0.8M,不会发出告警,但是阈值为大于1M的小数时又可以告警,怀疑是小于1M时,zabbix自动将单位换成K导致的。后来索性在定义带宽监控阈值时将单位统一为K,并且一定要为整数。如1300K可以,但是1300.5K则不会告警。)


脚本思路1:监控阈值在每台主机上去添加主机宏,不是手动添加,而是通过zabbix API为主机添加宏。

具体的过程如下:

通过CMDB接口拿到所有主机的信息,包括zabbix主机可见名和带宽信息,做判断,公网带宽为0M时,不添加主机宏。当不为0M时,先判断主机上是否存着这个宏,如果不存在则创建这个宏,如果存在,先比较该宏的值是否和由的带宽值计算出的告警阈值相等,如果相等,表示带宽没有更新过,不做处理。如果不相等,则表示带宽是调整过的,需要更新主机宏上的带宽阈值。


脚本思路2:当主机上存在带宽告警阈值宏的时候删除宏,如果公网带宽获取到的信息不是0M,则创建宏,如果是0M,则不创建宏。

思路2明显比思路1要简单。下面是思路2对应的脚本

先分别定义获取主机宏、删除主机宏、创建主机宏的函数出来。然后定义一个main函数,在main()函数中先删除宏,再创建宏即可。


Python脚本如下:

#!/usr/bin/python
#coding:utf-8
import requests,json
from zabbix_api import ZabbixAPI
server="http://zabbix.cheyaoshicorp.com"
username="zhouhai"
password="zhouhai"
zapi = ZabbixAPI(server=server, path="", log_level=0)
zapi.login(username, password)
def get_hostmacroid(host_name):
    '''
        通过host_name获取当前主机的$TRAFFIC_THRESHOLD宏ID
    '''
    hosts = zapi.host.get({"output":"extend","filter":{"name":host_name}})
    hostid = hosts[0].get('hostid')
    host_macros = zapi.usermacro.get({"output":"extend", "hostids":hostid})
    macros = [_ for _ in host_macros if _.get('macro') == '{$TRAFFIC_THRESHOLD}']
    hostmacroid = [macros[0].get('hostmacroid')] if len(macros) > 0 else None
    return hostid, hostmacroid
def delete_host_macro(host_name):
    hostid, hostmacroid = get_hostmacroid(host_name)
    if hostmacroid:
        zapi.usermacro.delete(hostmacroid)
def create_host_macro(host_name, bandwidthout):
    band = int(bandwidthout.replace('M', ''))
    if band == 0:
        return
    hostid, hostmacroid = get_hostmacroid(host_name)
    if hostid:
        zapi.usermacro.create({
            "hostid": hostid,
            "macro": "{$TRAFFIC_THRESHOLD}",
            "value": '{0}K'.format(band * 800)
        })
def main():
    '''
        1.删除主机的带宽宏
        2.创建主机宏
    '''
    url = 'http://cmdb.cheyaoshicorp.com/api/ecs'
    r = requests.get(url)
    ecss = json.loads(r.text)
    for _ in ecss:
        host_name = _.get('name')
        bandwidthout = _.get('bandwidthout')
        delete_host_macro(host_name)
        create_host_macro(host_name, bandwidthout)
main()