10、利用Zabbix API创建主机宏

xuke123

　　

　　需求：对所有云主机添加公网带宽监控，但不是每一台主机都有公网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()