使用paramiko執行遠程linux主機命令


  paramiko是python的SSH庫,可用來連接遠程linux主機,然后執行linux命令或者通過SFTP傳輸文件。

  關於使用paramiko執行遠程主機命令可以找到很多參考資料了,本文在此基礎上做一些封裝,便於擴展與編寫腳本。

  下面直接給出代碼:

  

 1 # coding: utf-8
 2 
 3 import paramiko
 4 import re
 5 from time import sleep
 6 
 7 # 定義一個類,表示一台遠端linux主機
 8 class Linux(object):
 9     # 通過IP, 用戶名,密碼,超時時間初始化一個遠程Linux主機
10     def __init__(self, ip, username, password, timeout=30):
11         self.ip = ip
12         self.username = username
13         self.password = password
14         self.timeout = timeout
15         # transport和chanel
16         self.t = ''
17         self.chan = ''
18         # 鏈接失敗的重試次數
19         self.try_times = 3
20 
21     # 調用該方法連接遠程主機
22     def connect(self):
23         while True:
24             # 連接過程中可能會拋出異常,比如網絡不通、鏈接超時
25             try:
26                 self.t = paramiko.Transport(sock=(self.ip, 22))
27                 self.t.connect(username=self.username, password=self.password)
28                 self.chan = self.t.open_session()
29                 self.chan.settimeout(self.timeout)
30                 self.chan.get_pty()
31                 self.chan.invoke_shell()
32                 # 如果沒有拋出異常說明連接成功,直接返回
33                 print u'連接%s成功' % self.ip
34                 # 接收到的網絡數據解碼為str
35                 print self.chan.recv(65535).decode('utf-8')
36                 return
37             # 這里不對可能的異常如socket.error, socket.timeout細化,直接一網打盡
38             except Exception, e1:
39                 if self.try_times != 0:
40                     print u'連接%s失敗,進行重試' %self.ip
41                     self.try_times -= 1
42                 else:
43                     print u'重試3次失敗,結束程序'
44                     exit(1)
45 
46     # 斷開連接
47     def close(self):
48         self.chan.close()
49         self.t.close()
50 
51     # 發送要執行的命令
52     def send(self, cmd):
53         cmd += '\r'
54         # 通過命令執行提示符來判斷命令是否執行完成
55         p = re.compile(r':~ #')
56 
57         result = ''
58         # 發送要執行的命令
59         self.chan.send(cmd)
60         # 回顯很長的命令可能執行較久,通過循環分批次取回回顯
61         while True:
62             sleep(0.5)
63             ret = self.chan.recv(65535)
64             ret = ret.decode('utf-8')
65             result += ret
66             if p.search(ret):
67                 print result
68                 return result

  下面進行測試:

 1 # 主機IP錯誤,無法連接的情況
 2 if __name__ == '__main__':
 3     host = Linux('192.168.180.12', 'root', 'xxxx')
 4     host.connect()
 5      6     host.send('ls -l')
 7     host.close()
 8
 1 # 鏈接正常的情況
 2 if __name__ == '__main__':
 3     host = Linux('192.168.180.128', 'root', 'love')
 4     host.connect()
 5     host.send('ls -l')
 6     host.close()
 7 
 8 運行結果:
 9 連接192.168.180.128成功
10 Last login: Sat May 21 07:25:39 2016 from 192.168.180.1
11 Have a lot of fun...
12 
13 ls -l
14 192:~ # ls -l
15 total 28
16 -rw------- 1 root root 18 May 21 07:17 .bash_history
17 drwxr-xr-x 1 root root 28 May 21 06:02 .config
18 drwx------ 1 root root 22 May 21 05:57 .dbus
19 drwx------ 1 root root  0 Sep 25  2014 .gnupg
20 drwxr-xr-x 1 root root 10 May 21 06:15 .local
21 -rw------- 1 root root 55 May 21 06:03 .xauth5mesuo
22 -rw------- 1 root root 55 May 21 07:22 .xauthEYqDmK
23 -rw------- 1 root root 55 May 21 07:25 .xauthGTrohO
24 -rw------- 1 root root 55 May 21 07:09 .xauthP90TnG
25 -rw------- 1 root root 48 May 21 07:40 .xauthjW8pI9
26 -rw------- 1 root root 48 May 21 07:40 .xauthx8T4ED
27 drwxr-xr-x 1 root root  0 Sep 25  2014 bin
28 drwxr-xr-x 1 root root 38 May 21 05:43 inst-sys
29 192:~ # 
30 
31 Process finished with exit code 0

 

 9 連接192.168.180.12失敗,進行重試
10 連接192.168.180.12失敗,進行重試
11 連接192.168.180.12失敗,進行重試
12 重試3次失敗,結束程序
13 
14 Process finished with exit code 1

  從上面代碼可以看出,通過簡單封裝之后,執行遠端linux主機就非常簡單了,只需簡單幾行代碼即可。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM