如何在另一个程序中使用控制台的运行输出

Question

我正在从命令行运行一个程序。这个程序打印出我的一些 CPU 的统计数据（如果有人感兴趣，可以链接到程序：https://github.com/opcm/pcm）。程序不退出。它会在同一个终端定期打印更新的统计信息。

我正在尝试在我自己编写的另一个程序中使用程序正在检索的这些统计信息。我正在编写的程序只是应该在 python 中实时绘制这些 CPU 统计数据与时间的关系图。

我不确定如何在我的程序中实时检索和使用打印到控制台的统计信息。我知道我可以通过键入./program.x > output.txt 将输出直接输出到文本文件而不是控制台。

我的第一个想法是将输出的统计信息推送到文本文件，同时在我的程序中从该文本文件中读取。但这感觉像是一种蛮力方法，我不确定它是否会起作用。

有谁知道我如何在另一个程序中使用一个程序的实时输出？

Answer 1

您可以通过多种方式实现此结果。

在 unix 系统上，我认为最简单的方法是使用管道 假设您有 program1 (p1) 输出一些数据和 program2(p2) 读取和使用这些数据。

您可以按照@Charles Landau 的建议使用| 将p1 的输出通过管道传输到p2，这会在两个程序的stdout 和stdin 之间创建一个管道

./p1 | p2

如果您还想可视化 p1 的输出，您可以使用 tee，它获取标准输入并将其分叉到标准输出和程序

./p1 | tee p2

现在这只有在您打算同时启动 p1 和 p2 时才有效（此时相信最好将其设为单个程序）。 管道的更通用用法是让 p1 创建一个命名管道 (FIFO) 并让 p2 从这个管道中读取：

p1.py（服务器）

import os
from random import randint
import time

FIFO="/tmp/test_fifo"

if not os.path.exists(FIFO):
    os.mkfifo(FIFO)

pipe = os.open( FIFO, os.O_RDWR) #I use this to avoid the program hanging on this step.
                                 #When you open a FIFO, the behavior is blocking, meaning that the program will hang until both end (r/w) of the pipe are being used
f = os.fdopen( pipe, "w")        #Then i want a file object so i can flush it

while 1:
    # Do operation
    output = randint(0,999999)
    # Print output
    print(output)

    # Send output into pipe
    f.flush()
    f.write( str(output) + '\n')
    time.sleep(0.5)

f.close()
pipe.close()

os.unlink(FIFO)

p2.py（客户端）

import os
import sys

FIFO="/tmp/test_fifo"

if not os.path.exists(FIFO):
    exit()

f = open( FIFO, "r")

while 1:
    read_fifo = f.readline()
    print( read_fifo, end='')
    sys.stdout.flush()

您也可以使用 shell 对 linux 中的 FIFO 进行一些简单的测试

mkfifo 1
shell1: tail -f 1
shell2: echo "Hello!" > 1

以类似的方式，您可以使用 TCP 协议让 2 程序进行通信 有关此方法的更多信息，请参阅 python https://docs.python.org/3/library/socket.html

Answer 2

你可以在 python 中制作类似 tail 的东西。这是代码示例，但如果文件被删除，则无法处理。

def tail_F(some_file):
    while True:
        try:
            for line in sh.tail("-f", some_file, _iter=True):
                yield line
                # do whatever with new line
        except sh.ErrorReturnCode_1:
            yield None

Answer 3

听起来你想要管道。管道获取程序的标准输出并将其直接传递给下一个程序。我能想到的最简单的例子：

ls | grep ".bin"

很容易记住，因为管道运算符有点像管道|

关于管道的更多信息：http://web.cse.ohio-state.edu/~mamrak.1/CIS762/pipes_lab_notes.html

编辑：使用sys.stdin 永久读取管道已得到很好的解释here

从中汲取的重要模式就在那里：

import sys
while True:
    line = sys.stdin.readline()
    # do something with line