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

2023-10-14 00:38| 来源: 网络整理| 查看: 265

法一：

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/

【本文地址】

公司简介

联系我们