标签:stdin stdout Python cmd 阻塞 subprocess PIPE 命令 True
之前是想写一个微信控制程序,通过登录网页微信,可以直接执行命令行代码。也不用ssh登录了,想法很方便。
但是现实很残酷,微信登录这块基本没有问题,已经有大佬写好了,但是命令行执行遇到问题了。
运行cmd
开始时,使用os.popen()执行命令,但是该命令需要手动修改运行目录。此方案被我直接丢弃了。
单开进程
那么自然想到通过启动进程的方式来实现,Python有对进程的封装subprocess
,可以通过创建Popen对象来实现。我只要单开一个bash,与它进行交互就好啦。
简单实现如下:
p = subprocess.Popen('/bin/bash', shell=True, stdin=subprocess.PIPE, stderr=subprocess.PIPE, stdout=subprocess.PIPE) while True: c = input() c += os.linesep p.stdin.write(c.encode('utf8')) print(out_s.decode('utf8'), end='')
然后,马上就有遇到问题了,输出流一直拿不到内容,被阻塞了。
刷新缓冲区
被阻塞有两种情况,一输入流阻塞,所以没有输出,二输出流阻塞。看到网上有的将输入流关闭就可以了:
p.stdin.close()
但是关闭后就不能再次运行命令了,通过查看其对象方法,发现可以直接刷新缓冲区,很好
p.stdin.flush()
但是发现读取到的文件只有一行,很明显,没有读完
循环读取
需要循环读取输出缓冲区的内容。
while True: out_s = p.stdout.readline() print(out_s.decode('utf8'), end='')
新的问题出现了,循环怎么结束啊?当缓冲区没有内容时,readline
方法会阻塞等待。
读取阻塞
很好,找了半天也没找到解决阻塞的办法。那就只能靠自己了,既然它要阻塞,那就随他阻塞好了,我单开一个线程去读取,让它一直阻塞去吧。
解决后的完整测试代码:
import subprocess import os import threading p = subprocess.Popen('/bin/bash', shell=True, stdin=subprocess.PIPE, stderr=subprocess.PIPE, stdout=subprocess.PIPE) def test(): global p while True: print(p.stdout.readline().decode('utf8'), end='') threading.Thread(target=test).start() while True: c = input() c += os.linesep p.stdin.write(c.encode('utf8')) p.stdin.flush()
很好,问题解决了,简单封装一个工具类吧。
注意:如果输入一个不存在的命令,输出内容不在stdout流中,要到stderr中获取。此方案暂时还不支持sudo命令,回头在研究研究
至此,其实还有一个小问题,我怎么能知道哪些返回是同一条命令所返回的呢?就这个微信工具来说,自然可以直接通过时间判断,若超过1s没有,则认为是一组,统一返回。感觉有些牵强,暂时没有想到更好的解决办法。
最后奉上工具链接:
<https://gitee.com/hujingnb/python_demo>
标签:stdin,stdout,Python,cmd,阻塞,subprocess,PIPE,命令,True 来源: https://www.cnblogs.com/hujingnb/p/12112670.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。