使用paramiko庫ssh連接到遠端云主機上時，非常偶現(xiàn)卡死現(xiàn)象，連接無法退出（可以是執(zhí)行命令時云主機重啟等造成）。需要給定一段時間，不管命令執(zhí)行是否卡住，都退出連接，顯示命令執(zhí)行超時錯誤。

線程+事件，在線程中執(zhí)行ssh命令，給事件配置超時時間。

1 from threading import Thread, Event

2 import paramiko

class SshClient(object): def __init__(self, ip, port, username, password): self.ip = ip self.host = host self.username = username self.password = password def exec_command(cmd, timeout): log.info(u'在ip:%s上執(zhí)行命令%s' % (self.ip, cmd)) sc = paramiko.SSHClient() sc.set_missing_host_key_policy(paramiko.AutoAddPolicy()) # 用來接收stdout stderror status信息 res = [None, None, None] def get_return(start_event, res_list): _, cmd_stdout, cmd_stderr = sc.exec_command(command=cmd, timeout=timeout) channel = cmd_stdout.channel cmd_status = channel.recv_exit_status() res_list[0] = cmd_stdout res_list[1] = cmd_stderr res_list[2] = cmd_status start_event.set() # 表示線程已經(jīng)執(zhí)行完畢 try: sc.connect(hostname=self.ip, port=self.port, username=self.username, password=self.password, timeout=30) # 這里的timeout是連接使用的，與我們要的不同 start_evt = Event() t = Thread(target=get_return, args=(start_evt, res)) t.start() start_evt.wait(timeout=timeout) # 執(zhí)行到這里說明線程已經(jīng)退出 if None in res:raise Exception(u'命令超時退出') stdout, stderr, status = res if status != 0:raise Exception(u'命令執(zhí)行返回非0！返回值為%s,錯誤信息為%s' % (status, stdout.read() + stderr.read())) return stdout.read() + stderr.read() finally: sc.close()}

知識點補充：

python paramiko的使用介紹

#設置ssh連接的遠程主機地址和端口t=paramiko.Transport((ip,port))#設置登錄名和密碼t.connect(username=username,password=password)#連接成功后打開一個channelchan=t.open_session()#設置會話超時時間chan.settimeout(session_timeout)#打開遠程的terminalchan.get_pty()#激活terminalchan.invoke_shell()

然后就可以通過chan.send(’command’)和chan.recv(recv_buffer)來遠程執(zhí)行命令以及本地獲取反饋。

paramiko有兩個模塊SSHClient()和SFTPClient()

SSHClient()的使用代碼:

import paramikossh = paramiko.SSHClient() # 創(chuàng)建SSH對象# 允許連接不在know_hosts文件中的主機ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())# 連接服務器ssh.connect(hostname=’192.168.2.103’, port=22, username=’root’, password=’123456’)stdin, stdout, stderr = ssh.exec_command(’ls’) # 執(zhí)行命令result = stdout.read() # 獲取命令結果print (str(result,encoding=’utf-8’))ssh.close() # 關閉連接

SSHClient()里有個transport變量,是用于獲取連接，我們也可單獨的獲取到transport變量，然后執(zhí)行連接操作

import paramikotransport = paramiko.Transport((’192.168.2.103’, 22))transport.connect(username=’root’, password=’123456’)ssh = paramiko.SSHClient()ssh._transport = transportstdin, stdout, stderr = ssh.exec_command(’df’)print (str(stdout.read(),encoding=’utf-8’))transport.close()

用transport實現(xiàn)上傳下載以及命令的執(zhí)行：

#coding:utf-8import paramikoimport uuidclass SSHConnection(object):def __init__(self, host=’192.168.2.103’, port=22, username=’root’,pwd=’123456’):self.host = hostself.port = portself.username = usernameself.pwd = pwdself.__k = Nonedef connect(self):transport = paramiko.Transport((self.host,self.port))transport.connect(username=self.username,password=self.pwd)self.__transport = transportdef close(self):self.__transport.close()def upload(self,local_path,target_path):# 連接，上傳# file_name = self.create_file()sftp = paramiko.SFTPClient.from_transport(self.__transport)# 將location.py 上傳至服務器 /tmp/test.pysftp.put(local_path, target_path)def download(self,remote_path,local_path):sftp = paramiko.SFTPClient.from_transport(self.__transport)sftp.get(remote_path,local_path)def cmd(self, command):ssh = paramiko.SSHClient()ssh._transport = self.__transport# 執(zhí)行命令stdin, stdout, stderr = ssh.exec_command(command)# 獲取命令結果result = stdout.read()print (str(result,encoding=’utf-8’))return resultssh = SSHConnection()ssh.connect()ssh.cmd('ls')ssh.upload(’s1.py’,’/tmp/ks77.py’)ssh.download(’/tmp/test.py’,’kkkk’,)ssh.cmd('df')ssh.close()

到此這篇關于使paramiko庫執(zhí)行命令時，在給定的時間強制退出的文章就介紹到這了,更多相關paramiko庫執(zhí)行命令內容請搜索好吧啦網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持好吧啦網(wǎng)！