python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python调用执行shell踩坑

Python代码调用执行shell踩坑解决

作者:大话性能

这篇文章主要为大家介绍了Python代码调用执行shell,踩过的坑解决方法,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪

背景

在老家 2 天,花了点时间,折腾了下 python 代码,之前一直遇到个偶现点问题,这两天总算解决了。

问题的核心是,你知道怎么正确用 python 代码调用 linux 上的 shell 命令。

这里回涉及 2 个问题。

问题一:

可以执行本机 shell 命令的相关 python 模块和函数有好几个,不过有一些已经被废弃或移除。

subprocess 模块用于创建子进程, 这个模块用于替换旧版本中的一些模块, 如:os.system, 

    os.spawn*, os.popen*, os.popen*, popen2., commands., subprocess 允许你能创建很多子进程, 创建的时候能能指定子进程和子进程的输入、输出、错误输出管道, 执行后能获取输出结果和执行状态。

在 python3.5 之后的版本中, 官方文档中提倡通过 subprocess.run() 函数替代其他函数来使用 subprocess 模块的功能。

测试:自己写了个简单的 demo,验证了 subprocess 调用 run 函数执行 linux 命令的时候是阻塞的,一直会等到命令执行完,再往下走。
另外,可以根据结果的 returncode 码,进行判断,linux 命令有没有执行成功。

问题二:

关于 python 远程执行 Linux,用的最多的还是 paramiko 模块,我之前也是用了这个模块,但是踩坑了。

核心主要代码

sshclient = paramiko.SSHClient()
        sshclient.set_missing_host_key_policy(paramiko.AutoAddPolicy())
sshclient.connect(ip, int(port), user, pwd, timeout=60)

check_in, check_out, check_err = sshclient.exec_command(self.check_slave_cmd)

坑就是,exec_command 函数是非阻塞的,不管命令有没有执行成功,python 代码就往下走了。所以,有时候回出现偶像的 bug,因为取决于命令执行的快慢和网速,譬如,假设你执行的 shell 命令耗时比较久,而代码已经执行到下面,发现没有你 shell 命令的结果就报错了。

解决方法

检测一下该 shell 命令的执行状态。调用 recv_exit_status(),该函数回一直阻塞中,直到 shell 命令结束,一般正常接受的 status 是 0.。
所以,当你要用到这个 paramiko 模块执行远程机器的 shell 命令的话,要多个心眼,保证 shell 命令有没有执行完成,会不会影响你的代码。

以上就是Python 代码调用执行 shell,踩过的坑的详细内容,更多关于Python调用执行shell踩坑的资料请关注脚本之家其它相关文章!

您可能感兴趣的文章:
阅读全文