import re import subprocess import time log_file = '/var/log/suricata/fast.log' ip_set = set() while True: with open(log_file) as file: file.seek(0, 2) # 将文件定位到文件末尾 for line in file: match = re.search(r'\d+\.\d+\.\d+\.\d+', line) if not match: continue ip = match.group() priority = re.search(r'Priority\.+', line).group() if ip in ip_set: continue if 'Priority: 1' in priority or 'Priority: 2' in priority: continue result = subprocess.call(['iptables', '-A', 'INPUT', '-s', ip, '-j', 'DROP']) if result == 0: print(f'封禁IP {ip} 成功!') ip_set.add(ip) else: print(f'封禁IP {ip} 失败!') time.sleep(5) 这段代码中我想用file.readlines()来读取文件,应该如何优化修改一下
时间: 2023-12-12 18:02:01 浏览: 28
可以将文件指针移到文件末尾,然后使用 file.readline() 来逐行读取新添加的内容,可以避免每次都读取整个文件。以下是修改后的代码:
```python
import re
import subprocess
import time
log_file = '/var/log/suricata/fast.log'
ip_set = set()
with open(log_file) as file:
file.seek(0, 2) # 将文件定位到文件末尾
while True:
line = file.readline()
if not line:
time.sleep(0.1) # 没有新内容就休眠一段时间
continue
match = re.search(r'\d+\.\d+\.\d+\.\d+', line)
if not match:
continue
ip = match.group()
priority = re.search(r'Priority\.+', line).group()
if ip in ip_set:
continue
if 'Priority: 1' in priority or 'Priority: 2' in priority:
continue
result = subprocess.call(['iptables', '-A', 'INPUT', '-s', ip, '-j', 'DROP'])
if result == 0:
print(f'封禁IP {ip} 成功!')
ip_set.add(ip)
else:
print(f'封禁IP {ip} 失败!')
```