zabbix api

#!/usr/bin/env python
# -*- coding:utf8 -*-
# __author__ = '北方姆Q'

from pyzabbix import ZabbixAPI
from plugins.duia.singleton import Singleton


class ZabbixFactory(Singleton):
    def __init__(self, url, username='Admin', password='zabbix'):
        self.obj = ZabbixAPI(url)
        self.obj.login(username, password)

    def get_hosts(self):
        """
        获取所有的主机信息列表
        :return: dict in list
        """
        return self.obj.host.get(output='extend')

    def get_host(self, name):
        """
        获取单台的主机信息
        :param name:  需要获取的显示名
        :return: dict
        """
        return self.obj.host.get(output='extend', filter={'name': name})[0]

    def get_host_groups(self):
        """
        获取所有的主机组信息列表
        :return: dict in list
        """
        return self.obj.hostgroup.get(output='extend')

    def get_hosts_by_group(self, group_name):
        """
        获取某个主机组内所有主机信息列表
        :param group_name: 主机组名
        :return: dict in list
        """
        return self.obj.host.get(groupids=self.__get_host_group_id(group_name), output="extend")

    def get_templates(self):
        """
        获取所有模版信息
        :return: dict in list
        """
        return self.obj.template.get(output="extend")

    def get_template(self, template_name):
        """
        根据名称获取单个模版信息
        :return: dict
        """
        return self.obj.template.get(output='extend', filter={"host": template_name})[0]

    def create_group(self, group_name):
        """
        创建一个新主机组
        :param group_name: 新主机组名
        :return: 新组id，str
        """
        ret = self.obj.hostgroup.create(name=group_name)
        return ret['groupids'][0]

    def create_host(self, name, ip, groups_name, templates_name, proxy_name=None, status=0):
        """
        创建一台新主机
        :param name: 主机名
        :param ip: 主机受监控ip
        :param groups_name: 主机所在组,str、list、tuple、set
        :param templates_name: 主机关联模版,str、list、tuple、set
        :param proxy_name: 使用的代理名称，默认不使用
        :param status: 是否启用，默认开启
        :return: 新主机id，str
        """
        groups = [{'groupid': self.__get_host_group_id(groups_name)}] if isinstance(groups_name, str) 
            else [{'groupid': self.__get_host_group_id(group_name)} for group_name in groups_name]
        templates = [{'templateid': self.__get_template_id(templates_name)}] if isinstance(templates_name, str) 
            else [{'templateid': self.__get_template_id(template_name)} for template_name in templates_name]

        if proxy_name is None:
            ret = self.obj.host.create(
                host=name,
                status=status,
                groups=groups,
                templates=templates,
                interfaces=[{
                    'type': 1,              # 监控类型
                    'main': 1,              # 监控类型下第几行
                    'useip': 1,             # 使用ip还是dns
                    'ip': ip,               # ip框
                    'dns': '',              # dns框
                    'port': '10050'         # port框
                }]
            )
        else:
            ret = self.obj.host.create(
                host=name,
                status=status,
                groups=groups,
                templates=templates,
                interfaces=[{
                    'type': 1,
                    'main': 1,
                    'useip': 1,
                    'ip': ip,
                    'dns': '',
                    'port': '10050'
                }],
                proxy_hostid=self.__get_proxy_id(proxy_name)
            )
        return ret['hostids'][0]

    def __get_host_group_id(self, group_name):
        """
        获取某主机组id
        :param group_name: 主机组名
        :return: int
        """
        return int(self.obj.hostgroup.get(output='groupid', filter={'name': group_name.split(',')})[0].get('groupid'))

    def __get_template_id(self, template_name):
        """
        获取某模版id
        :param template_name: 模版名
        :return: int
        """
        return int(self.obj.template.get(output='templateid', filter={'host': template_name.split(',')})[0].get('templateid'))

    def __get_proxy_id(self, proxy_name):
        """
        获取某代理id
        :param proxy_name: 代理名
        :return: int
        """
        return int(self.obj.proxy.get(output='proxyid', filter={'host': proxy_name.split(',')})[0].get('proxyid'))