zabbix Python3管理

import requests
import json
import os

# user config here
ip = '192.168.52.130'
user = "root"
password = "123"

# user config end
class ZabbixApi:
    def __init__(self, ip, user, password):
        self.url = 'http://' + ip + '/api_jsonrpc.php'
        self.headers = {"Content-Type": "application/json",
                        }
        self.user = user
        self.password = password
        self.auth = self.__login()

    def __login(self):
        '''
        zabbix登陆获取auth
        :return: auth  #  样例bdc64373690ab9660982e0bafe1967dd
        '''
        data = json.dumps({
            "jsonrpc": "2.0",
            "method": "user.login",
            "params": {
                "user": self.user,
                "password": self.password
            },
            "id": 1,
        })
        try:
            response = requests.post(url=self.url, headers=self.headers, data=data, timeout=2)
            # {"jsonrpc":"2.0","result":"bdc64373690ab9660982e0bafe1967dd","id":10}
            auth = response.json().get('result', '')
            return auth
        except Exception as e:
            print("\033[0;31;0m Login error check: IP,USER,PASSWORD\033[0m")
            raise Exception

    def host_get(self, hostname=''):
        '''
        获取主机。不输入hostname 则获取所有
        :param hostName: zabbix主机名不允许重复。（IP允许重复）。#host_get(hostname='gateway1')
        :return: {'jsonrpc': '2.0', 'result': [], 'id': 21}
        :return: {'jsonrpc': '2.0', 'result': [{'hostid': '10277', ... , 'host': 'gateway', ...}], 'id': 21}
        '''
        if hostname == '':
            print("no hostname and find all host")
            data = json.dumps({
                "jsonrpc": "2.0",
                "method": "host.get",
                "params": {
                    "output": [
                        "hostid",
                        "host"
                    ],
                    "selectInterfaces": [
                        "interfaceid",
                        "ip"
                    ]
                },
                "id": 1,
                "auth": self.auth
            })
        else:
            data = json.dumps({
                "jsonrpc": "2.0",
                "method": "host.get",
                "params": {"output": "extend",
                           "filter": {"host": hostname},
                           },
                "id": 1,
                "auth": self.auth
            })
        try:
            response = requests.get(url=self.url, headers=self.headers, data=data, timeout=2)
            return response.json()  # len(ret.get('result'))为1时获取到，否则未获取到。
        except Exception as e:
            print("\033[0;31;0m HOST GET ERROR\033[0m")
            raise Exception

    def get_items(self, hostid='', host=''):
        data = json.dumps({
            "jsonrpc": "2.0",
            "method": "item.get",
            "params": {
                "output": ["itemids", host],
                "hostids": hostid,
            },
            "auth": self.auth,  # theauth id is what auth script returns, remeber it is string
            "id": 1,
        })
        try:
            response = requests.get(url=self.url, headers=self.headers, data=data, timeout=2)
            # hosts_count=len(response.json().get('result',''))
            return response.json()  # len(ret.get('result'))为1时获取到，否则未获取到。

        except Exception as e:
            print("\033[0;31;0m Template GET ERROR\033[0m")
            raise Exception

    def get_history(self,itemid=''):
        data = json.dumps({
            "jsonrpc": "2.0",
            "method": "history.get",
            "params": {
                "output": "extend",
                "history": 3,
                "itemids": itemid,
                "sortfield": "clock",
                "sortorder": "DESC",
                "limit": 10

            },
            "auth": self.auth,
            "id": 1
        })
        try:
            response = requests.get(url=self.url, headers=self.headers, data=data, timeout=2)
            # hosts_count=len(response.json().get('result',''))
            return response.json()  # len(ret.get('result'))为1时获取到，否则未获取到。

        except Exception as e:
            print("\033[0;31;0m Template GET ERROR\033[0m")
            raise Exception

    def hostgroup_create(self, hostGroupName=''):
        if len(self.hostgroup_get(hostGroupName).get('result')) == 1:
            print("无需创建，hostgroup存在")
            return 100
        data = json.dumps({
            "jsonrpc": "2.0",
            "method": "hostgroup.create",
            "params": {
                "name": hostGroupName
            },
            "auth": self.auth,
            "id": 1
        })
        try:
            response = requests.get(url=self.url, headers=self.headers, data=data, timeout=2)
            # hosts_count=len(response.json().get('result',''))
            # print('l',hosts)
            return response.json()  # len(ret.get('result'))为1时获取到，否则未获取到。

        except Exception as e:
            print("\033[0;31;0m HOSTGROUP CREATE ERROR\033[0m")
            raise Exception

    def template_get(self, templateName=''):
        data = json.dumps({
            "jsonrpc": "2.0",
            "method": "template.get",
            "params": {
                "output": "extend",
                "filter": {
                    "name": templateName
                }
            },
            "auth": self.auth,
            "id": 50,
        })
        try:
            response = requests.get(url=self.url, headers=self.headers, data=data, timeout=2)
            # hosts_count=len(response.json().get('result',''))
            # print('l',hosts)
            return response.json()  # len(ret.get('result'))为1时获取到，否则未获取到。

        except Exception as e:
            print("\033[0;31;0m Template GET ERROR\033[0m")
            raise Exception

    def host_create(self, hostname, hostip, hostGroupName, templateName):
        '''
        创建host，并分配分组，关联模版
        host_create('host3','1.1.1.1','gp1 test,gp2 test','Template App FTP Service,Template App HTTP Service')
        :param hostname: 'host3'
        :param hostip: '1.1.1.1'
        :param hostGroupName: 多个组逗号分割'gp1 test,gp2 test'
        :param templateName: 多个模版都逗号分割'Template App FTP Service,Template App HTTP Service'
        :return:
        '''
        # 判断主机名是否重复。 zabbix不允许主机名重复
        find_hostname = self.host_get(hostname)
        if len(find_hostname.get('result')):
            print("###recheck### hostname[%s] exists and return" % hostname)
            return 1

        # 判断template是否存才，不存在退出。 否则初始化备用
        template_list = []
        for t in templateName.split(','):
            find_template = self.template_get(t)
            if not len(find_template.get('result')):
                # template不存在退出 # 一般因为输错关系
                print("###recheck### template[%s] not find and return " % t)
                return 1

            tid = self.template_get(t).get('result')[0]['templateid']
            template_list.append({'templateid': tid})

        # 判断hostgroup是否存在。 不存在则创建。 并初始化hostgroup_list备用
        hostgroup_list = []
        for g in hostGroupName.split(','):
            find_hostgroupname = self.hostgroup_get(g)
            if not len(find_hostgroupname.get('result')):
                # hostgroupname 不存在，创建
                self.hostgroup_create(g)
            # {'jsonrpc': '2.0', 'result': [{'groupid': '15', 'name': 'linux group 1', 'internal': '0', 'flags': '0'}],'id': 1}

            # for g in hostGroupName.split(','):
            #     print(self.hostgroup_get(g).get('result'))
            gid = self.hostgroup_get(g).get('result')[0]['groupid']
            hostgroup_list.append({'groupid': gid})

        # print("## 测试组信息 ##")
        # print(hostgroup_list)
        # print(template_list)

        data = json.dumps({
            "jsonrpc": "2.0",
            "method": "host.create",
            "params": {
                "host": hostname,
                "interfaces": [
                    {
                        "type": 1,
                        "main": 1,
                        "useip": 1,
                        "ip": hostip,
                        "dns": "",
                        "port": "10050"
                    }
                ],
                "groups": hostgroup_list,
                "templates": template_list,
            },
            "auth": self.auth,
            "id": 1
        })

        try:
            response = requests.get(url=self.url, headers=self.headers, data=data, timeout=2)
            # hosts_count=len(response.json().get('result',''))
            # print('l',hosts)
            print("执行返回信息： 添加HOST[%s,%s]完成" % (hostname, hostip))
            return response.json()  # len(ret.get('result'))为1时获取到，否则未获取到。

        except Exception as e:
            print("\033[0;31;0m HOST CREATE ERROR\033[0m")
            raise Exception

    def hostgroup_get(self, hostGroupName=''):
        '''
        获取主机组
        :param hostGroupName:
        :return: {'jsonrpc': '2.0', 'result': [{'groupid': '15', 'name': 'linux group 1', 'internal': '0', 'flags': '0'}], 'id': 1}
        '''
        data = json.dumps({
            "jsonrpc": "2.0",
            "method": "hostgroup.get",
            "params": {
                "output": "extend",
                "filter": {
                    "name": hostGroupName
                }
            },
            "auth": self.auth,
            "id": 1,
        })
        try:
            response = requests.get(url=self.url, headers=self.headers, data=data, timeout=2)
            # hosts_count=len(response.json().get('result',''))
            # print('l',hosts)
            return response.json()  # len(ret.get('result'))为1时获取到，否则未获取到。

        except Exception as e:
            print("\033[0;31;0m HOSTGROUP GET ERROR\033[0m")
            raise Exception

zabbix = ZabbixApi(ip, user, password)
hostinfo = zabbix.host_get("XuYiJi")
hostid = hostinfo.get("result")[0].get("hostid")
host = hostinfo.get("result")[0].get("host")
items = zabbix.get_items(hostid, host)
itemid = items.get("result")[0].get("itemid")
history = zabbix.get_history(itemid)
stat = history.get("result")[0].get("value")
clock = history.get("result")[0].get("clock")
print("hostid:","\033[0;37;40m{}\033[0m".format(hostid))
print("host:","\033[0;37;40m{}\033[0m".format(host))
print("itemid:","\033[0;37;40m{}\033[0m".format(itemid))
print("stat:","\033[0;37;40m{}\033[0m".format(stat))
import time, datetime
timeStamp = int(clock)
timeArray = time.localtime(timeStamp)
otherStyleTime = time.strftime("%Y-%m-%d %H:%M:%S", timeArray)

print("clock:","\033[0;37;40m{}\033[0m".format(otherStyleTime))

# 获取/创建hostgroup测试
# groupinfo=zabbix.hostgroup_get('Linux servers')
# print(groupinfo)
# r=zabbix.hostgroup_create(hostGroupName='gt3')
# print(r)
# 获取template测试
# r=zabbix.template_get('Template OS Linux by Prom')
# print(r)
# https://www.cnblogs.com/infaaf/p/9950004.html 参考本文档
# if __name__ == '__main__':
#     try:
#         zabbix = ZabbixApi(ip, user, password)
#         print("...login success...")
#         # unittest()
#
#         # 添加主机单台
#         # zabbix.host_create('host7','1.1.1.1','gp1 test,gp2 test','Template OS Linux by Prom')
#
#         # 批量添加主机，从文本serverlist.txt
#         BASE_DIR = os.path.dirname(os.path.abspath(__file__))
#
#         ADD_LIST= os.path.join(BASE_DIR,'serverlist.txt')
#         with open(ADD_LIST,'r',encoding='utf-8') as f:
#             for line in f:
#                 if len(line.strip()): #跳过空行
#                     serverinfo=line.strip().split('#')
#                     print(serverinfo)
#                     print("-"*60)
#                     print(serverinfo[0].strip("'"),serverinfo[1].strip("'"),serverinfo[2].strip("'"),serverinfo[3].strip("'"))
#                     zabbix.host_create(serverinfo[0],serverinfo[1],serverinfo[2],serverinfo[3])
#
#     except Exception as e:
#         print(e)