博客
关于我
Python实现cmd命令连续执行
阅读量:420 次
发布时间:2019-03-06

本文共 1487 字,大约阅读时间需要 4 分钟。

之前是想写一个微信控制程序,通过登录网页微信,可以直接执行命令行代码。也不用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 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没有,则认为是一组,统一返回。感觉有些牵强,暂时没有想到更好的解决办法。

最后奉上工具链接:

转载地址:http://kohuz.baihongyu.com/

你可能感兴趣的文章
记一次讲故事机器人的开发-我有故事,让机器人来读
查看>>
seo 回忆录百度基本概念(一)
查看>>
kettle 执行 kjb 临时文件夹 /tmp permission denied 问题
查看>>
netcore中使用session
查看>>
Android 开发学习进程0.25 自定义控件
查看>>
多媒体文件格式全解说(下)--图片
查看>>
淘宝WAP版小BUG分析
查看>>
asp.net打印网页后自动关闭网页【无需插件】
查看>>
【Maven】POM基本概念
查看>>
【Java思考】Java 中的实参与形参之间的传递到底是值传递还是引用传递呢?
查看>>
【设计模式】单例模式
查看>>
远程触发Jenkins的Pipeline任务的并发问题处理
查看>>
entity framework core在独立类库下执行迁移操作
查看>>
Asp.Net Core 2.1+的视图缓存(响应缓存)
查看>>
【wp】HWS计划2021硬件安全冬令营线上选拔赛
查看>>
Ef+T4模板实现代码快速生成器
查看>>
JQuery选择器
查看>>
2.2.2原码补码移码的作用
查看>>
Java面试题:Servlet是线程安全的吗?
查看>>
Java集合总结系列2:Collection接口
查看>>