有什么办法可以提高效率吗？ Tenable API 报告调用

Question

我有一个执行大量数据操作的脚本，但它受到了这个函数的限制。

Tenable 生成器数组ips 的长度始终在 1000 左右，无论取舍。 ips[row]的长度为5。

我可以在此处进行任何改进以提高效率吗？我觉得这需要的时间比它应该的要长得多。

def get_ten(sc):
    now = time.time()
    ips = [sc.analysis.vulns(('ip', '=', ip), tool='sumseverity', sortDirection='desc') for ip in [x[15] for x in csv.reader(open('full.csv', 'r'))
     if x[15] != 'PrivateIpAddress']]
    row = 0
    while row < len(ips):
        scan_data = []
        scan_count = 0
        for scan in ips[row]:
            count = scan['count']
            scan_data.append(count)
            scan_count += int(count)
        row += 1
    print(time.time() - now)

输出：2702.747463464737

谢谢！

Answer 1

我建议你尝试颠倒你的逻辑。

通过查看您的代码，您现在看起来像：

1. 读取 CSV 以获取 IP。
2. 为每个 IP 调用 sc.analysis API
3. 处理结果

我的最佳猜测是大部分时间都用于发送 API 调用然后等待结果。相反，我建议您尝试：

1. 调用 sc.analysis API（不过滤 ip）并读取结果以获取所有 IP（作为一组？）
2. 读取 CSV 并获取另一组 IP。
3. 设置联合操作以查找易受攻击的 IP 地址。