Python调包侠之阿里云堡垒机添加主机

#!/usr/bin/env python3
# coding=utf-8
# Alliot  
# blog: www.iots.vip 
import json
import base64
import sys
import subprocess
from libs.tools import async_run_command, sync_run_command


def async_run_command(cmd, cwd=None):
    """
    运行命令, 异步，不阻塞
    :param: cmd: 命令
    :param: cwd: 目录
    """
    # 非阻塞，不获取返回的内容
    subprocess.Popen(
        cmd,
        stdin=subprocess.PIPE,
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE,
        shell=True,
        cwd=cwd)


class HacManagement():
    def __init__(self):
        from aliyunsdkcore.client import AcsClient
        from aliyunsdkcore.request import CommonRequest

        # 阿里云密钥
        ID = ''
        Secret = ''
        # 堡垒机区域
        region_id = 'cn-beijing' 
        # 堡垒机实例ID
        hac_instance_id = "bastionhost-cn-xxx"
        self.client = AcsClient(ID, Secret, region_id)
        self.request = CommonRequest()
        self.request.set_accept_format('json')
        self.request.set_domain('yundun-bastionhost.aliyuncs.com')
        self.request.set_method('POST')
        self.request.set_protocol_type('https')  # https | http
        self.request.set_version('2019-12-09')
        self.request.add_query_param('RegionID', region_id)
        self.request.add_query_param('InstanceId', hac_instance_id)

    def create_host(self, **kwargs):
        """
        调用 CreateHost 接口在堡垒机中创建需要运维的主机。
        :param: kwargs: HostNmae: 主机名、Source: 主机来源(Local、Ecs、 Rds)、OSType: 操作系统类型(
        Linux、Windows)、ActiveAddressType: 地址类型(Private、Public)、HostPrivateAddress: 私网 IP 地址(当 address_type 为 Private
        时必选)、host_public_address: 公网 IP 地址(当 address_type 为 Public 时必选)、Comment: 备注(可选)、SourceInstanceId:
        指定新创建的ECS实例ID或专属集群主机ID（当 source 选择 Ecs 或 Rds 时，该参数为必填项）、InstanceRegionId: 指定新创建的ECS实例或专属集群主机所属区域ID当 source 选择
        Ecs 或 Rds 时，该参数为必填项） :return: HostId: 创建的主机的 ID
        """
        for i in kwargs.keys():
            self.request.add_query_param(i, kwargs[i])
        self.request.set_action_name('CreateHost')
        response = self.client.do_action_with_exception(self.request)

        # 调用 salt 添加用户到目标服务器系统
        async_run_command(cmd=f'salt {kwargs["HostName"]} state.sls add')
        return json.loads(response)['HostId']

    def create_host_account(self, **kwargs):
        """
        调用 CreateHostAccount 创建堡垒机中的主机账号
        :param kwargs: HostId: 主机 ID、ProtocolName: 协议名(SSH、RDP)、HostAccountName: 用户名、Password: 密码、PrivateKey: 私钥、PassPhrase: 私钥口令
        :return: HostAccountId: 主机账号ID
        """
        self.request.set_action_name('CreateHostAccount')
        for i in kwargs.keys():
            self.request.add_query_param(i, kwargs[i])
        response = self.client.do_action_with_exception(self.request)
        return json.loads(response)['HostAccountId']

    def modify_hosts_port(self, *args, protocol_name, port):
        """
        调用 ModifyHostsPort 修改主机指定协议的端口
        :param args: HostIds 主机ID（要求为列表）
        :param protocol_name: 协议名(SSH、RDP)
        :param port: 端口
        :return:
        """
        self.request.set_action_name('ModifyHostsPort')
        self.request.add_query_param('HostIds', *args)
        self.request.add_query_param('ProtocolName', protocol_name)
        self.request.add_query_param('Port', port)
        response = self.client.do_action_with_exception(self.request)
        return str(response, encoding='utf-8')

    def add_hosts_to_group(self, *args, host_group_id):
        """
        调用 AddHostsToGroup 添加指定主机到指定主机组
        :param args: HostIds 主机ID（要求为列表）
        :param host_group_id: 主机组ID
        :return:
        """
        self.request.set_action_name('AddHostsToGroup')
        self.request.add_query_param('HostIds', *args)
        self.request.add_query_param('HostGroupId', host_group_id)
        response = self.client.do_action_with_exception(self.request)
        return str(response, encoding='utf-8')



def add_host_and_users(hostname, ip, comment):
    aliyun_key = base64.b64encode('''-----BEGIN RSA PRIVATE KEY-----
# 这里填 系统用户 'aliyun' 的私钥
-----END RSA PRIVATE KEY-----'''.encode('utf-8'))

    print('添加主机到堡垒机\n')
    host_id = HacManagement().create_host(
        HostName=hostname,
        Source="Local",
        OSType="Linux",
        ActiveAddressType="Private",
        HostPrivateAddress=ip,
        Comment=comment)

    print('添加主机账户aliyun\n')
    HacManagement().create_host_account(
        HostId=host_id,
        ProtocolName='SSH',
        HostAccountName='aliyun',
        PrivateKey=aliyun_key
    )

    print('修改主机ssh端口为9999\n') # 根据服务器的 SSH 服务的端口修改
    HacManagement().modify_hosts_port(
        f'[{host_id}]',
        protocol_name='SSH',
        port=9999
    )

    print('添加主机到泛组\n')  # 这里需要自己手动去“云盾堡垒机”控制台创建一个组（
    HacManagement().add_hosts_to_group(
        f'[{host_id}]',
        host_group_id='1'  # 创建组的时候，F12打开浏览器控制台抓一下请求，看看GroupId即可
    )
    print(f'主机已添加到堡垒机，主机ID为{host_id}\n')


if __name__ == '__main__':
    add_host_and_users(sys.argv[1], sys.argv[2], sys.argv[3])