在python中，如何从c ++共享库捕获标准输出到变量

Question

由于某些其他原因，我使用的c ++共享库将一些文本输出到标准输出。在python中，我想捕获输出并 保存到一个变量中
。关于重定向标准输出，还有许多类似的问题，但在我的代码中不起作用。

1 import sys
2 import cStringIO
3 save_stdout = sys.stdout
4 sys.stdout = cStringIO.StringIO()
5 func()
6 sys.stdout = save_stdout

在第5行中， func（） 将调用共享库，共享库生成的文本仍输出到控制台！如果将 func（） 更改 为print“ hello”
，则可以使用！

我的问题是：

1. 如何将c ++共享库的stdout捕获 到变量中 ？
2. 为什么使用StringIO无法捕获共享库的输出？

Answer 1

Python的sys.stdout对象只是位于普通stdout文件描述符之上的Python包装器-
对其进行更改只会影响Python进程，而不会影响基础文件描述符。任何非Python代码，无论是execed的另一个可执行文件还是已加载的C共享库，都将无法理解，并将继续使用普通的文件描述符进行I
/ O。

因此，为了使共享库输出到其他位置，您需要通过打开一个新的文件描述符，然后使用来替换stdout来更改基础文件描述符os.dup2()。您可以使用临时文件作为输出，但是最好使用使用创建的管道os.pipe()。但是，如果没有任何内容读取管道，则存在死锁的危险，因此为了防止我们可以使用其他线程排空管道。

以下是一个完整的工作示例，该示例不使用临时文件，并且不容易死锁（在Mac OS X上进行了测试）。

C共享库代码：

// test.c
#include <stdio.h>

void hello(void)
{
  printf("Hello, world!\n");
}

编译为：

$ clang test.c -shared -fPIC -o libtest.dylib

Python驱动程序：

import ctypes
import os
import sys
import threading

print 'Start'

liba = ctypes.cdll.LoadLibrary('libtest.dylib')

# Create pipe and dup2() the write end of it on top of stdout, saving a copy
# of the old stdout
stdout_fileno = sys.stdout.fileno()
stdout_save = os.dup(stdout_fileno)
stdout_pipe = os.pipe()
os.dup2(stdout_pipe[1], stdout_fileno)
os.close(stdout_pipe[1])

captured_stdout = ''
def drain_pipe():
    global captured_stdout
    while True:
        data = os.read(stdout_pipe[0], 1024)
        if not data:
            break
        captured_stdout += data

t = threading.Thread(target=drain_pipe)
t.start()

liba.hello()  # Call into the shared library

# Close the write end of the pipe to unblock the reader thread and trigger it
# to exit
os.close(stdout_fileno)
t.join()

# Clean up the pipe and restore the original stdout
os.close(stdout_pipe[0])
os.dup2(stdout_save, stdout_fileno)
os.close(stdout_save)

print 'Captured stdout:\n%s' % captured_stdout