从zabbix添加主机到jumpserver！

# Python 示例
# pip install requests drf-httpsig
import requests, datetime, json
from httpsig.requests_auth import HTTPSignatureAuth
# from zabbix import Zabbix
DEBUG = False


class Zabbix(object):


    def __init__(self):
        self.base_url = 'http://192.168.3.243/zabbix/api_jsonrpc.php'
        self.token = 'ZABBIXTOKEN'
        self.headers = {
            'Authorization': f'Bearer {self.token}',
            }


    # 获取主机操作系统平台
    def get_platform_zabbix(self, items_list):
        platform = 'Other'
        for item in items_list:
            if item['name'] == 'Operating system':
                if 'Linux' in item['lastvalue'] or 'linux' in item['lastvalue']:
                    platform = 'Linux'
                elif 'Windows' in item['lastvalue'] or 'windows' in item['lastvalue']:
                    platform = 'Windows'
                else:
                    platform = 'Other'
        return platform


    # 格式化主机信息
    def get_format_host_info(self, host_list):
        host_list_new = []
        for host in host_list:
            data = {}
            data['platform'] = self.get_platform_zabbix(host['items'])
            data['name'] = host['host']
            data['host_groups'] = [group['name'] for group in host['hostgroups']]
            data['ip'] = host['interfaces'][0]['ip']
            host_list_new.append(data)
        print(host_list_new)
        return host_list_new


    # 获取主机列表
    def get_host_list(self):
        payload = {
            'jsonrpc': '2.0',
            'method': 'host.get',
            'params': {
                'output': ['hostid', 'host'],
                'selectHostGroups': ['name'],
                'selectInterfaces': ['ip'],
                'selectItems': ['name', 'lastvalue'],
                'filter': {
                    'type': 1
                }
            },
            'id': 1
        }
        response = requests.post(self.base_url, json=payload, headers=self.headers)
        host_list = self.get_format_host_info(response.json()['result'])
        return host_list




class Jumpserver(Zabbix):
    def __init__(self):
        super().__init__()
        self.jms_url = 'http://192.168.3.29'
        self.key_id = 'AccessKeyID'
        self.secret_id = 'AccessKeySecret'
        self.auth = self.get_auth()
        self.host_list_zabbix = self.get_host_list()
        self.host_dict_jumpserver = {}
        self.asset_nodes_dict = {}
        self.asset_nodes_list = []
        self.platform_dict = {}
        self.platform_list = []
        self.gmt_form = '%a, %d %b %Y %H:%M:%S GMT'
        self.headers = {
            'Accept': 'application/json',
            'X-JMS-ORG': '00000000-0000-0000-0000-000000000002',
            'Date': datetime.datetime.utcnow().strftime(self.gmt_form)
        }


    # 获取跳板机系统认证信息
    def get_auth(self):
        signature_headers = ['(request-target)', 'accept', 'date']
        auth = HTTPSignatureAuth(key_id=self.key_id, secret=self.secret_id, algorithm='hmac-sha256',
                                 headers=signature_headers)
        return auth


    # 获取跳板机系统用户信息
    def get_user_info(self):
        url = self.jms_url + '/api/v1/users/users/'
        response = requests.get(url, auth=self.auth, headers=self.headers)
        if DEBUG:
            print(response.status_code)
            print(response.text)


    # 获取跳板机系统平台和协议列表
    def get_platform_and_protocol(self):
        url = self.jms_url + '/api/v1/assets/platforms/'
        response = requests.get(url, auth=self.auth, headers=self.headers)
        self.platform_list = response.json()
        for platform in self.platform_list:
            self.platform_dict[platform['name']] = {
                'id': platform['id'],
                'pk': platform['id']
            }
            protocol_list = []
            for protocol in platform['protocols']:
                protocol_list.append({
                    'name': protocol['name'],
                    'port': protocol['port']
                })
            self.platform_dict[platform['name']]['protocols'] = protocol_list
        if DEBUG:
            print(response.status_code)
            print(response.text)


    # 获取跳板机系统主机列表
    def get_host_list_jumpserver(self):
        url = self.jms_url + '/api/v1/assets/hosts/'
        response = requests.get(url, auth=self.auth, headers=self.headers)
        host_info_list = response.json()
        for host_info in host_info_list:
            self.host_dict_jumpserver[host_info['name']] = host_info


    # 获取跳板机系统节点信息
    def get_asset_nodes(self):
        url = self.jms_url + '/api/v1/assets/nodes/'
        response = requests.get(url, auth=self.auth, headers=self.headers)
        for node_info in response.json():
            self.asset_nodes_dict[node_info['name']] = node_info['id']
        self.asset_nodes_list = response.json()
        if DEBUG:
            print(response.status_code)
            print(response.text)


    # 创建节点到跳板机系统
    def create_node(self, name):
        url = self.jms_url + '/api/v1/assets/nodes/'
        data = {
            'value': name,
            'full_value': name
        }
        response = requests.post(url, auth=self.auth, headers=self.headers, json=data)
        if DEBUG:
            print(response.status_code)
            print(response.text)
        return response.json()['id']


    # 创建主机到跳板机系统
    def create_host(self, host_info):
        url = self.jms_url + '/api/v1/assets/hosts/'
        response = requests.post(url, auth=self.auth, headers=self.headers, json=host_info)
        if DEBUG:
            print(response.status_code)
            print(response.text)


    # 更新主机信息到跳板机系统
    def update_host(self, host_info):
        host_id = self.host_dict_jumpserver[host_info['name']]['id']
        url = self.jms_url + f'/api/v1/assets/hosts/{host_id}/'
        response = requests.patch(url, auth=self.auth, headers=self.headers, json=host_info)
        if DEBUG:
            print(response.status_code)
            print(response.text)


    # 创建或更新主机到跳板机系统
    def create_or_update_host(self):
        for host_info in self.host_list_zabbix:
            data = {
                'labels': [],
                'is_active': True,
                'name': host_info['name'],
                'address': host_info['ip'],
                'accounts': []
            }
            if host_info['platform'] in self.platform_dict:
                platform_name = host_info['platform']
                data['platform'] = {
                    'pk': self.platform_dict[platform_name]['pk']
                }
                data['protocols'] = self.platform_dict[platform_name]['protocols']
            nodes_list = []
            for group in host_info['host_groups']:
                if group in self.asset_nodes_dict:
                    nodes_list.append({
                        'pk': self.asset_nodes_dict[group]
                    })
                else:
                    nodes_list.append({
                        'pk': self.create_node(group)
                    })
            data['nodes'] = nodes_list
            if host_info['name'] not in self.host_dict_jumpserver:
                print(f"正在创建主机 {host_info['name']} 到跳板机系统中")
                self.create_host(host_info=data)
            else:
                print(f"正在更新主机 {host_info['name']} 到跳板机系统中")
                self.update_host(host_info=data)




J = Jumpserver()
J.get_platform_and_protocol()
J.get_host_list_jumpserver()
J.get_asset_nodes()
J.create_or_update_host()