首页 ›编程› Python › 查看内容

Python实现cmd命令连续执行

2019-12-29 11:24 |来自: 互联网 2276 0

摘要: 之前是想写一个微信控制程序，通过登录网页微信，可以直接执行命令行代码。也不用ssh登录了，想法很方便。但是现实很残酷，微信登录这块基本没有问题，已经有大佬写好了，但是命令行执行遇到问题了。运行cmd开始时，使用os.popen()执行命令，但是该命令需要手动修改运行目录。此方案被我直接丢弃了。单 ...

关键词： 阻塞 subprocess 命令 39 缓冲区 stdin 问题输出但是

之前是想写一个微信控制程序，通过登录网页微信，可以直接执行命令行代码。也不用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 subprocessimport osimport threadingp = 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://www.toutiao.com/a6775441238855254532/

声明：文章版权归原作者所有部分文章转自互联网如有侵权请联系 [邮箱地址] 删除