python与命令行(shell,cmd)或者其他命令行程序持续交互的方法

法一：

from subprocess import Popen, TimeoutExpired, PIPE

# 这里必须使用信号 PIPE 而不是自己创建的文本流，这样才能使用 communicate 函数
proc = Popen(args='python ./test.py', shell=True, encoding='utf-8', stdin=PIPE, stdout=PIPE, stderr=PIPE)
try:
    outs, errs = proc.communicate(input='4',timeout=15)
    print(outs)
except TimeoutExpired:
    proc.kill()

这种方式只能交流一次，之后就不能写入stdin了

 

法二：

from subprocess import Popen, TimeoutExpired, PIPE

with Popen(args='python ./test.py', shell=True, encoding='utf-8', stdin=PIPE, stdout=PIPE, stderr=PIPE) as proc:
    try:
        stdin = proc.stdin
        # 以\n结束，程序才能判断这是一次完整的读取 效果和`'1\n2\n'`相同
        stdin.write('1\n')
        stdin.write('2\n')
        # 将输入从缓存刷新到子进程中
        stdin.flush()
        s = proc.stdout.readlines()
        print(s)

    except TimeoutExpired:
        proc.kill()

注意：

1.这种方式可以持续、多次写入和读取，如果程序是持续输出类型（比如象棋ucci引擎，打印完一段信息后等待你的输入），那就不要用readlines，否则会一直block在那里（等待读取完，但不可能），这种情况建议使用readline，通过内容判断是否结束，如果没有输出了你还readline，也会block，最好的解决办法就是使用异步read，这个问题自己搜。

2.对于一些在命令行执行的应用，可以

1）将其加入环境变量，直接执行；

2）有些程序会读取相对目录的配置文件，但是当前的workdir在python项目而不是程序目录，因此程序会运行出错，这个时候只要先os.chdir到程序目录，执行程序，再切换到原python项目目录即可，例：

from subprocess import Popen, TimeoutExpired, PIPE
import os 
wd = os.getcwd()
os.chdir("程序工作目录")

with Popen(args='python ./test.py', shell=True, encoding='utf-8', stdin=PIPE, stdout=PIPE, stderr=PIPE) as proc:
    os.chdir(wd)
    try:
        stdin = proc.stdin
        # 以\n结束，程序才能判断这是一次完整的读取 效果和`'1\n2\n'`相同
        stdin.write('1\n')
        stdin.write('2\n')
        # 将输入从缓存刷新到子进程中
        stdin.flush()
        s = proc.stdout.readlines()
        print(s)

    except TimeoutExpired:
        proc.kill()

 参考文章：

1.https://www.jianshu.com/p/10a7553489dc

2.https://segmentfault.com/a/1190000016880835

3.https://www.qingheluo.com/pythondexitongjiaohusubprocessmokuai/