Python实现cmd命令连续执行

Posted 2020-11-26 烟草的香味

之前是想写一个微信控制程序，通过登录网页微信，可以直接执行命令行代码。也不用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>