首页 | 编程 | python

解决paramiko执行命令超时的问题

2020年04月16日 作者：脚本小娃子原文

问题：paramiko远程执行命令，需要等到命令返回信息，如果命令执行时间比较长，返回信息就需要等很久

方案：

1、使用nohup + 待执行命令 + & ，使用后台执行的方式，应该可以快速返回

2、设置paramiko的执行命令等待超时时间

stdin, stdout, stderr = self.client.exec_command(cmd,timeout=10,get_pty=True)

其实上面的两种方案都不可行：方案1，需要优化，下面这种直接调用的方式会导致test.sh启动不起来

stdin, stdout, stderr = self.client.exec_command(‘sh ~/test.sh &',get_pty=True)

方案2，对于需要等待很久的命令，如果timeout小于程序的执行时间，还是会失败，命令执行失败

最终的解决方案来自参考1

把执行很久的sh文件A放入另一个sh文件B中，然后paramiko执行文件B。摘录原文：

paramiko远程执行后台脚本“阻塞”问题

我写的远程命令通道上线之后，发现在远程脚本中后台再执行另一个脚本，通道会一直等待后台脚本执行完成才会返回，有时甚至会僵死。

1、复现过程如下：

①、编写测试脚本

脚本1：test.sh

#!/bin/bash
sleep 30
echo test end
exit 0

脚本2：run.sh

#!/bin/bash
bash /tmp/test.sh &
echo run ok!
exit 0

脚本3：test.py

import paramiko
client = paramiko.SSHClient()
client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
client.connect(hostname='192.168.1.10', port=22, username='root', password='123456', timeout=300,allow_agent=False,look_for_keys=False)       
stdin,stdout,stderr=client.exec_command("bash /tmp/run.sh") 
            
result_info = ""

for line in stdout.readlines():
 result_info += line

print result_info

将test.sh和run.sh传到远程服务器上，比如放到192.168.1.10:/tmp/下。

②、发起远程执行

在本地执行 python test.py，会发现整个脚本不会立即打印run ok，而是等30s之后才打印包括test.sh的所有输出信息。

2、解决办法

将远程脚本的标准输出stdout重定向到错误输出stderr即可，test.py 修改如下：

import paramiko
client = paramiko.SSHClient()
client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
client.connect(hostname='192.168.1.10', port=22, username='root', password='123456', timeout=300,allow_agent=False,look_for_keys=False)       
stdin,stdout,stderr=client.exec_command("bash /tmp/run.sh 1>&2") 
            
result_info = ""

for line in stderr.readlines():
 result_info += line

print result_info

现在执行，就能立即得到结果了。其实原因很简单，因为bash /tmp/test.sh & 虽然是后台执行，但是依然会产生标准输出，一旦产生标准输出，paramiko就会认为命令还未执行完成，且stdout的buffer大于stderr，因此产生等待问题。

这里只要将脚本执行的标准输出重定向到错误输出（1>&2），然后paramiko就可以使用stderr快速读取远程打屏信息了。

基于上面paramiko的原理：有stdout输出，就认为命令没有执行完成。得出下面的解决方案，对于需要执行很久test.sh，我们首先把标准输出都发给标准错误输出(1>&2)，然后后台启动(&)

stdin, stdout, stderr = self.client.exec_command(‘bash ~/test.sh 1>&2 &',get_pty=True)

项目实践，还有下面的方案：去掉参数get_pty,这样就不会回传标准输出信息和标准错误信息

self.client.exec_command(‘bash ~/test.sh')

以上这篇解决paramiko执行命令超时的问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持来客网。

相关链接