后台进程的日志输出到文件

Question

我有耗时的 SNMP walk 任务来执行，我使用 Popen 命令作为后台进程运行。如何在日志文件中捕获此后台任务的输出。在下面的代码中，我尝试对 ip_list 中的每个 IP 进行 snampwalk，并将所有结果记录到 abc.txt。但是，我看到生成的文件 abc.txt 是空的。

下面是我的示例代码 -

import subprocess
import sys

f = open('abc.txt', 'a+')

ip_list = ["192.163.1.104", "192.163.1.103", "192.163.1.101"]

for ip in ip_list:
    cmd = "snmpwalk.exe -t 1 -v2c -c public "
    cmd = cmd + ip
    print(cmd)
    p = subprocess.Popen(cmd, shell=True, stdout=f)
    p.wait()
f.close()

print("File output - " + open('abc.txt', 'r').read())

对于每个 IP，命令的示例输出可以是这样的 -

sysDescr.0 = STRING: Software: Whistler Version 5.1 Service Pack 2 (Build 2600)
sysObjectID.0 = OID: win32
sysUpTimeInstance = Timeticks: (15535) 0:02:35.35
sysContact.0 = STRING: unknown
sysName.0 = STRING: UDLDEV
sysLocation.0 = STRING: unknown
sysServices.0 = INTEGER: 72
sysORID.4 = OID: snmpMPDCompliance 

我已经尝试过 Popen。但如果这是一个耗时的后台进程，它不会将输出记录到文件中。但是，当我尝试运行 ls/dir 之类的后台进程时，它可以工作。任何帮助表示赞赏。

Answer 1

这里的主要问题是对 Popen 所做的事情的期望以及我假设的工作方式。 p.wait() 这里将等待进程完成，然后再继续，这就是为什么 ls 例如可以工作，但更耗时的任务却不能。在您调用 p.stdout.flush() 之前，不会自动刷新输出。

您的设置方式更适用于：

1. 执行命令
2. 等待退出
3. 捕捉输出

然后使用它。对于您的用例，您最好使用替代库或使用stdout=subprocess.PIPE 并自己捕捉它。这将意味着以下内容：

import subprocess
import sys

ip_list = ["192.163.1.104", "192.163.1.103", "192.163.1.101"]
with open('abc.txt', 'a+') as output:
    for ip in ip_list:
        print(cmd := f"snmpwalk.exe -t 1 -v2c -c public {ip}")
        process = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE) # Be wary of shell=True
        while process.poll() is None:
            for c in iter(lambda: process.stdout.read(1), ''):
                if c != '':
                    output.write(c)

with open('abc.txt', 'r') as log:
    print("File output: " + log.read())

这里要带走的关键内容是process.poll()，它检查进程是否完成，如果没有，我们将尝试使用process.stdout.read(1) 捕获输出以一次读取一个字节。如果您知道有新的线路即将推出，您可以将这三行切换为 output.write(process.stdout.readline())，一切就绪。