1. 程式人生 > 程式設計 >解決paramiko執行命令超時的問題

解決paramiko執行命令超時的問題

問題:paramiko遠端執行命令,需要等到命令返回資訊,如果命令執行時間比較長,返回資訊就需要等很久

方案:

1、使用nohup + 待執行命令 + & ,使用後臺執行的方式,應該可以快速返回

2、設定paramiko的執行命令等待超時時間

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

其實上面的兩種方案都不可行:方案1,需要優化,下面這種直接呼叫的方式會導致test.sh啟動不起來

stdin,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,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',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,stderr = self.client.exec_command(‘bash ~/test.sh 1>&2 &',get_pty=True)

專案實踐,還有下面的方案:去掉引數get_pty,這樣就不會回傳標準輸出資訊和標準錯誤資訊

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

以上這篇解決paramiko執行命令超時的問題就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。