Jumpserver API调用 python

àì夳堔傛蜴生んèń 2023-02-20 02:57 23阅读 0赞

### Jumpserver API调用 ###

使用pandas 作为数据筛选工具

import requests
    import datetime
    import uuid
    from httpsig.requests_auth import HTTPSignatureAuth
    from pypinyin import lazy_pinyin
    import json
    import pandas as pd
    import os
    
    
    
    class JumpServer():
        def __init__(self):
            self.KEY_ID = ''    # KEY_ID
            self.SECRET = ''    # SECRET
            self.groups_id = { "测试": ["f3aa9f1f89a04d578f0f6f8b1de196f0"],    # group id 在数据库中查询，这块暂时懒得写。。id从数据库中查出来。
                              "开发": ["def367be15354f5b8ac124a7dc790fca", "71b050839ade464395329667cd972dea"]}
            self.signature_headers = ['(request-target)', 'accept', 'date', 'host']
            self.GMT_FORMAT = '%a, %d %b %Y %H:%M:%S GMT'
            self.headers = { 
                'Accept': 'application/json',
                'Date': datetime.datetime.utcnow().strftime(self.GMT_FORMAT)
            }
            self.auth = HTTPSignatureAuth(key_id=self.KEY_ID, secret=self.SECRET,
                                          algorithm='hmac-sha256',
                                          headers=self.signature_headers)
            self.host = ""  # 设置地址
            self.node_list = self.get_node_list()
    
        def update_node_list(self):
            self.node_list = self.get_node_list()
    
        def create_users(self, users: [], groups: []):
            groupid_list = []
            for i in groups:
                groupid_list += self.groups_id[i]
            for name in users:
                username = self.getStrAllAplha(name)
                data = { 
                    "id": str(uuid.uuid5(uuid.NAMESPACE_DNS, str(uuid.uuid1()))).replace("-", ""),
                    "name": name,
                    "username": username,
                    "password": "设置密码",
                    "email": "%s@邮箱地址后缀" % username,
                    "public_key": "",
                    "groups": groupid_list,
                    "role": "User",
                    "wechat": "",
                    "phone": "",
                    "mfa_level": 2,
                    "comment": "",
                    "source": "local",
                    "is_active": "true",
                    "date_expired": "2099-05-28T02:29:45.567Z"
                }
                req = requests.post('http://%s/api/v1/users/users/' % self.host,
                                    auth=self.auth, headers=self.headers, data=data)
                if req.status_code == 201:
                    print(name, username, "Create OK")
                else:
                    print(name, "Create FALIED")
                    print(req.content)
    
        # 获取节点
        def get_node_list(self):
            req = requests.get('http://%s/api/v1/assets/nodes/' % self.host,
                               auth=self.auth, headers=self.headers)
            node_list_json = json.loads(req.content.decode())
            node_list_df = pd.DataFrame.from_records(
                node_list_json, columns=["id", "name", "full_value", "key"])
            node_list_df["full_value"] = node_list_df["full_value"].str.replace(
                " ", "")
            return node_list_df
    
            # 获取节点
        def get_node_list_1(self):
            req = requests.get('http://%s/api/v1/assets/nodes/' % self.host,
                               auth=self.auth, headers=self.headers)
            print(req.content.decode())
    
        # 创建节点
        def create_node(self, fullpath):
            if self.get_nodeid_by_fullpath(fullpath):
                print("%s exsists" % fullpath)
                return
    
            name = os.path.basename(fullpath)
            F_path = os.path.dirname(fullpath)
            F_nodeid = self.get_nodeid_by_fullpath(F_path)
    
            if not F_nodeid:
                print("父节点 %s 不存在" % F_path)
                exit(9)
    
            # F_key = self.node_list[self.node_list["id"]
            # == F_nodeid]["key"].str.cat()
            # node_list = self.node_list[self.node_list["key"].str.startswith(
            # F_key + ":")]["key"].to_list()
            # for _ in range(0, 100):
            # node_key = F_key + ":" + str(_)
            # if node_key in node_list:
            # continue
            # break
            data = { 
                "name": name,
                "value": name
            }
            print(data)
            req = requests.post('http://%s/api/v1/assets/nodes/%s/children/' % (self.host, F_nodeid),
                                auth=self.auth, headers=self.headers, data=data)
            if req.status_code == 201:
                print("%s Create SUCCESSED" % fullpath)
                self.update_node_list()
            else:
                print("%s Create FALIED" % fullpath, req.content.decode())
    
        # 创建资产
        def create_asset(self, fullpath, ip_str, prefix=""):
            if prefix:
                prefix += "_"
    
            node_id = self.get_nodeid_by_fullpath(fullpath)
            if not node_id:
                print("没有找到对应节点 %s" % fullpath)
                exit(10)
            ip_list = [line.strip() for line in ip_str.strip().split("\n")]
    
            for ip in ip_list:
                data = { 
                    "ip": ip,
                    "hostname": prefix + ip,
                    "platform": "Linux",
                    "admin_user": "54eb2ab5-d4c1-4de3-b20f-e4000c60934b",
                    "nodes": node_id,
                    "is_active": True
                }
                req = requests.post('http://%s/api/v1/assets/assets/' % self.host,
                                    auth=self.auth, headers=self.headers, data=data)
                if req.status_code == 201:
                    print("%s add into %s SUCCESSED" % (ip, fullpath))
                else:
                    print("%s add into %s FALIED" % (ip, fullpath))
    
        # 获取节点id
        def get_nodeid_by_fullpath(self, fullpath=None):
            node_id = self.node_list["full_value"] == fullpath
            if node_id.any():
                return self.node_list[node_id]["id"].str.cat()
            else:
                return None
    
        def getStrAllAplha(self, name):
            return ''.join(lazy_pinyin(name))
    
    
    if __name__ == "__main__":
        Jumpserver = JumpServer()
    
        fullpath = "Default/ops-tool/test1"
    
        # 创建节点
        Jumpserver.create_node(fullpath)
    
        # 添加资产
        ip_str = '''118.178.89.242 123.233.233.233 '''
        
        Jumpserver.create_asset(fullpath, ip_str, prefix="test")
    
        # 创建用户
        # Jumpserver.create_users(["张三", "李四"], ["开发"])