解決paramiko執(zhí)行命令超時(shí)的問(wèn)題

更新時(shí)間：2020年04月16日 11:42:28 作者：腳本小娃子

這篇文章主要介紹了解決paramiko執(zhí)行命令超時(shí)的問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

問(wèn)題：paramiko遠(yuǎn)程執(zhí)行命令，需要等到命令返回信息，如果命令執(zhí)行時(shí)間比較長(zhǎng)，返回信息就需要等很久

方案：

1、使用nohup + 待執(zhí)行命令 + & ，使用后臺(tái)執(zhí)行的方式，應(yīng)該可以快速返回

2、設(shè)置paramiko的執(zhí)行命令等待超時(shí)時(shí)間

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

其實(shí)上面的兩種方案都不可行：方案1，需要優(yōu)化，下面這種直接調(diào)用的方式會(huì)導(dǎo)致test.sh啟動(dòng)不起來(lái)

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

方案2，對(duì)于需要等待很久的命令，如果timeout小于程序的執(zhí)行時(shí)間，還是會(huì)失敗，命令執(zhí)行失敗

最終的解決方案來(lái)自參考1

把執(zhí)行很久的sh文件A放入另一個(gè)sh文件B中，然后paramiko執(zhí)行文件B。摘錄原文：

paramiko遠(yuǎn)程執(zhí)行后臺(tái)腳本“阻塞”問(wèn)題

我寫(xiě)的遠(yuǎn)程命令通道上線之后，發(fā)現(xiàn)在遠(yuǎn)程腳本中后臺(tái)再執(zhí)行另一個(gè)腳本，通道會(huì)一直等待后臺(tái)腳本執(zhí)行完成才會(huì)返回，有時(shí)甚至?xí)┧馈?/p>

1、復(fù)現(xiàn)過(guò)程如下：

①、編寫(xiě)測(cè)試腳本

腳本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傳到遠(yuǎn)程服務(wù)器上，比如放到192.168.1.10:/tmp/下。

②、發(fā)起遠(yuǎn)程執(zhí)行

在本地執(zhí)行 python test.py，會(huì)發(fā)現(xiàn)整個(gè)腳本不會(huì)立即打印run ok，而是等30s之后才打印包括test.sh的所有輸出信息。

2、解決辦法

將遠(yuǎn)程腳本的標(biāo)準(zhǔn)輸出stdout重定向到錯(cuò)誤輸出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

現(xiàn)在執(zhí)行，就能立即得到結(jié)果了。其實(shí)原因很簡(jiǎn)單，因?yàn)閎ash /tmp/test.sh & 雖然是后臺(tái)執(zhí)行，但是依然會(huì)產(chǎn)生標(biāo)準(zhǔn)輸出，一旦產(chǎn)生標(biāo)準(zhǔn)輸出，paramiko就會(huì)認(rèn)為命令還未執(zhí)行完成，且stdout的buffer大于stderr，因此產(chǎn)生等待問(wèn)題。

這里只要將腳本執(zhí)行的標(biāo)準(zhǔn)輸出重定向到錯(cuò)誤輸出（1>&2），然后paramiko就可以使用stderr快速讀取遠(yuǎn)程打屏信息了。

基于上面paramiko的原理：有stdout輸出，就認(rèn)為命令沒(méi)有執(zhí)行完成。得出下面的解決方案，對(duì)于需要執(zhí)行很久test.sh，我們首先把標(biāo)準(zhǔn)輸出都發(fā)給標(biāo)準(zhǔn)錯(cuò)誤輸出(1>&2)，然后后臺(tái)啟動(dòng)(&)