如何使用 subprocess.POPEN 获取异步输入和输出

Question

我已经为此工作了几个小时，但未能提出一个好的解决方案。一点背景知识，我正在运行一个密码破解程序，该程序从命令行关闭源代码，但当我的 gpu 温度太高时必须不断暂停它。

我使用这个程序在 python 中进行其他操作，所以这是我更喜欢的语言。无论如何，密码程序会定期更新它的运行情况、gpu 温度等，并允许我随时暂停它。

我的温度正常，但由于阻塞问题，我猜我无法发送暂停命令。至少它什么也没做。我见过几个线程化输出的例子，但还没有看到使用线程化输入和输出而不会导致任何问题的东西。

我的意思是，据我所知，在当前的 POPEN 限制下这可能是不可能的，但希望有一些方向。

popen = Popen(command, stdout=PIPE, stdin=PIPE, shell=True)
lines_iterator = iter(popen.stdout.readline, b"")
while 1:
    for line in lines_iterator:
        cleanLine = line.replace("\n", "")
        p = re.compile('[0-9][0-9]c Temp')
        m = p.search(cleanLine)
        print cleanLine
        if m:
            temperature = m.group(0)
            if int(temperature[:2]) > 80:
                overheating = True
                print "overheating"
        if overheating:
            if "[s]tatus [p]ause [r]esume [b]ypass [q]uit" in line:

                #It's not doing anything right here, it just continues
                print popen.communicate("p")[0]

这是我的代码的要点。它仍然处于 hacky 阶段，所以我知道它可能没有遵循最佳编码实践。

Answer 1

编辑：对不起，我在最初的回答中对overheating 的范围感到困惑。我删除了答案的第一部分，因为它不再相关。

communicate 将等待进程退出，因此在这种情况下它可能不是您想要的。如果您希望流程继续进行 你可以使用popen.stdin.write("p") 之类的东西。如果您的流程需要，您可能还需要发送“\n”。

此外，如果您对额外的依赖项感到满意，您可能会对设计用于控制交互式进程的 pexpect 模块感兴趣。

Answer 2

一个简单的便携式解决方案是use threads here。没有block buffering issues就够了。

如果检测到过热则读取输出并停止输入（未测试）：

#!/usr/bin/env python
from subprocess import Popen, PIPE, CalledProcessError
from threading import Event, Thread

def detect_overheating(pipe, overheating):
    with pipe: # read output here
        for line in iter(pipe.readline, ''): 
            if detected_overheating(line.rstrip('\n')):
                overheating.set()    # overheating
            elif paused: #XXX global
                overheating.clear()  # no longer overheating

process = Popen(args, stdout=PIPE, stdin=PIPE, bufsize=1,
                universal_newlines=True) # enable text mode
overheating = Event() 
t = Thread(target=detect_overheating, args=[process.stdout, overheating])
t.daemon = True # close pipe if the process dies
t.start()
paused = False
with process.stdin: # write input here
    while process.poll() is None:
        if overheating.wait(1): # Python 2.7+
            # overheating
            if not paused:
                process.stdin.write('p\n') # pause
                process.stdin.flush()
                paused = True
        elif paused: # no longer overheating
            pass #XXX unpause here
            paused = False
if process.wait() != 0: # non-zero exit status may indicate failure
    raise CalledProcessError(process.returncode, args)