import json import subprocess with open('/var/log/suricata/fast.log', 'r') as f: events = json.load(f) for event in events: if event['alert']['severity'] <= 2: ip = event['src_ip'] # 调用iptables命令封禁IP result = subprocess.call(['iptables', '-A', 'INPUT', '-s', ip, '-j', 'DROP']) if result == 0: print(f'封禁IP {ip} 成功！') else: print(f'封禁IP {ip} 失败！')修改优化一下这段代码

可以考虑以下几点进行优化：

1. 将代码封装成函数，方便复用。

2. 使用 try-except 语句处理 subprocess.call() 函数返回的异常，可以更好地处理命令执行失败的情况。

3. 使用 f-string 格式化字符串，使代码更加简洁明了。

4. 将常量提取出来，避免硬编码。

下面是优化后的代码：

import json
import subprocess

LOG_FILE = '/var/log/suricata/fast.log'
IPTABLES_CMD = ['iptables', '-A', 'INPUT', '-s']

def ban_ip(ip):
    try:
        result = subprocess.call(IPTABLES_CMD + [ip, '-j', 'DROP'])
        if result == 0:
            print(f'封禁IP {ip} 成功！')
        else:
            print(f'封禁IP {ip} 失败！')
    except subprocess.CalledProcessError as e:
        print(f'命令执行失败：{e}')

with open(LOG_FILE, 'r') as f:
    events = json.load(f)
    for event in events:
        if event['alert']['severity'] <= 2:
            ip = event['src_ip']
            ban_ip(ip)

这样优化后的代码更具可读性和可维护性，也更加健壮。