Python-在Python中对子进程进行非阻塞读取
我使用subprocess
模块启动一个子流程并连接到它的输出流(stdout
)。我希望能够在它的标准输出上执行非阻塞读取。在调用.readline
之前,是否有方法使.readline
不阻塞或检查流中是否有数据?我希望它是可移植的,或者至少在Windows
和Linux
下工作。
这是我目前的操作方式(.readline
如果没有可用数据,则会阻塞):
p = subprocess.Popen('myprogram.exe', stdout = subprocess.PIPE)
output_str = p.stdout.readline()
-
fcntl
,select
,asyncproc
不会在这种情况下帮助。不管使用什么操作系统,一种可靠的读取流而不阻塞的方法是使用
Queue.get_nowait()
:import sys from subprocess import PIPE, Popen from threading import Thread try: from queue import Queue, Empty except ImportError: from Queue import Queue, Empty # python 2.x ON_POSIX = 'posix' in sys.builtin_module_names def enqueue_output(out, queue): for line in iter(out.readline, b''): queue.put(line) out.close() p = Popen(['myprogram.exe'], stdout=PIPE, bufsize=1, close_fds=ON_POSIX) q = Queue() t = Thread(target=enqueue_output, args=(p.stdout, q)) t.daemon = True # thread dies with the program t.start() # ... do other things here # read line without blocking try: line = q.get_nowait() # or q.get(timeout=.1) except Empty: print('no output yet') else: # got line # ... do something with line