使用ARL灯塔API编写Py脚本，实现快速便捷信息搜集+资产管理！

from re import S
from cryptography.hazmat.primitives.asymmetric.ec import BrainpoolP256R1
import requests
import readline
import json
from prettytable import PrettyTable
import urllib3
from os import system, name
import urllib.request

apikey = "ff44256c-fa2b-483f-9956-b0a84e153ade"

def task():
    table = PrettyTable(['任务名称', '目标地址', '当前状态', '开始时间', '结束时间', '任务ID'])
    headers = {
    'accept': 'application/json',
    'Token': apikey
    }
    ceshi = requests.get("https://IP:5003/api/task/", headers=headers, verify=False)
    good = ceshi.text
    res = json.loads(good.encode('latin-1').decode('unicode_escape'))
    for i in res['items']:
        table.add_row([i['name'],i['target'],i['status'],i['start_time'],i['end_time'],i['_id']])
    else:
        print(table)
    table.clear_rows()

def add(name, target):
    headers = {
    'Content-type': 'application/json',
    'Token': apikey,
    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/72.0.3626.121 Safari/537.36"
    }
    data = {
        "name": name,
        "task_tag": "task",
        "target": target,
        "policy_id": "策略ID",
        "result_set_id": "策略ID"}
    jsondata = json.dumps(data, separators=(',', ':'), ensure_ascii=False)
    ceshi = requests.post("https://IP:5003/api/task/policy/", headers=headers, data=jsondata, verify=False)
    good = ceshi.text
    res = json.loads(good.encode('latin-1').decode('unicode_escape'))
    if res['message'] == 'success':
        print(res['message'] + '  \033[1;31m下发任务成功！\033[0m')
    else:
        print('\033[1;32m下发任务失败！\033[0m')

xunhuan = 1
while xunhuan == 1:
    a = input("\033[1;31m>> \033[0m")
    if a == 'ls':
        task()
    if a == 'addfile':
        name = input('\033[1;34m[+] >> 任务名称：\033[0m')
        file = input('\033[1;34m[+] >> 目标文件：\033[0m')
        with open(file, 'r', encoding='utf-8') as file1:
            for readFile in file1: #逐行读取
                add(name.encode("utf-8","ignore").decode("latin-1"), readFile)
    if a == 'add':
        print('\033[1;31m[+] 任务目标英文逗号分割！例：1.2.3.4,4.3.2.1\033[0m')
        print('\033[1;32m--------------------------------------------------\033[0m')
        name = input('\033[1;34m[+] >> 任务名称：\033[0m')
        target = input('\033[1;34m[+] >> 任务目标：\033[0m')
        add(name.encode("utf-8","ignore").decode("latin-1"), target)
    if a == 'c':
        clear()
    if a == 'e':
        break