Python子进程无法打印vnstat进程

时间:2018-07-25 19:12:07

标签: python

感谢您抽出宝贵的时间阅读这篇文章。

我正在尝试使用vnstat在python中进行实时带宽监视。不幸的是,它没有打印我想要的输出,而且我似乎无法弄清楚原因。这是我的下面的代码。

from subprocess import Popen, PIPE
import time

def run(command):
 process = Popen(command, stdout=PIPE, bufsize=1, shell=True ,universal_newlines=True)
 while True:
    line = process.stdout.readline().rstrip()
    print(line)



if __name__ == "__main__":
 run("sudo vnstat -l -i wlan1") 

当我在终端中运行此代码时,这是我得到的输出:

sudo python testingLog.py
Monitoring wlan1...    (press CTRL-C to stop)

在终端中运行“ vnstat -l -i wlan1”时,没有显示所需的输出。

所需的输出:

Monitoring wlan1...    (press CTRL-C to stop)

rx:        0 kbit/s     0 p/s          tx:        0 kbit/s     0 p/s

当我运行vnstat -l -i wlan1时会发生什么,它将更新它并实时运行,所以我怀疑我的打印错误,因为它无法打印所需的输出,但我似乎无法弄清楚原因。

1 个答案:

答案 0 :(得分:0)

这不是您的打印错误,而是vnstat不断更新同一行而没有发出新行的事实,因此process.stdout.readline()挂在一个点,等待一条永远不会出现的新行。

如果您只想将vnstat STDOUT重定向到Python的STDOUT,则可以通过管道将其传输,即:

import subprocess
import sys
import time

proc = subprocess.Popen(["vnstat", "-l", "-i", "wlan1"], stdout=sys.stdout)
while proc.poll() is None:  # loop until the process ends (kill vnstat to see the effect)
    time.sleep(1)  # wait a second...
print("\nProcess finished.")

但是,如果您想捕获输出并自己处理,则必须同时流STDOUT个字符(或安全缓冲区)以捕获vnstat发布的所有内容,然后决定如何处理。例如,要模拟上述管路,但要坐在驾驶员座位上,您可以执行以下操作:

import subprocess
import sys

proc = subprocess.Popen(["vnstat", "-l", "-i", "wlan1"], stdout=subprocess.PIPE)
while True:  # a STDOUT read loop
    output = proc.stdout.read(1)  # grab one character from vnstat's STDOUT
    if output == "" and proc.poll() is not None:  # process finished, exit the loop
        break
    sys.stdout.write(output)  # write the output to Python's own STDOUT
    sys.stdout.flush()  # flush it...
    # of course, you can collect the output instead of printing it to the screen...
print("\nProcess finished.")