使用python通過SSH登陸linux並操作

用的昨天剛接觸到的庫，在windows下通過paramiko來登錄linux系統並執行了幾個命令，基本算是初試成功，后面會接着學習的。

代碼：

>>> import paramiko

>>> ssh = paramiko.SSHClient()
>>> ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
>>> ssh.connect('10.104.6.8',username = 'xiaopeng',password='******')
>>> cmd = 'cd'    #進入用戶目錄home
>>> stdin,stdout,stderr = ssh.exec_command(cmd)
>>> cmd = 'ls >test'  #管道，ls命名的輸出到文件test里面
>>> stdin,stdout,stderr = ssh.exec_command(cmd)
>>> cmd = 'cat test'  # 顯示test的內容，即ls命名的結果
>>> stdin,stdout,stderr = ssh.exec_command(cmd)
>>> print stdout.readlines()   #結果 漢字用的字符顯示
['code\n', 'Desktop\n', 'order.cpp\n', 'python\n', 'test\n', '\xe5\x85\xac\xe5\x85\xb1\xe7\x9a\x84\n', '\xe6\xa8\xa1\xe6\x9d\xbf\n', '\xe8\xa7\x86\xe9\xa2\x91\n', '\xe5\x9b\xbe\xe7\x89\x87\n', '\xe6\x96\x87\xe6\xa1\xa3\n', '\xe4\xb8\x8b\xe8\xbd\xbd\n', '\xe9\x9f\xb3\xe4\xb9\x90\n']

這里我們都會發現，使用exec_command('cd dirname')時並不會切換目錄，execute_command() 是a single session，每次執行完后都要回到缺省目錄。所以可以 .execute_command('cd  /var; pwd')。

---

 

 

python里面的paramiko模塊，用這個實現ssh登錄更加簡單。看下面的代碼：

#-*- coding: utf-8 -*-
#!/usr/bin/python
import paramiko
import threading
def ssh2(ip,username,passwd,cmd):
    try:
        ssh = paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        ssh.connect(ip,22,username,passwd,timeout=5)
        for m in cmd:
            stdin, stdout, stderr = ssh.exec_command(m)
#           stdin.write("Y")   #簡單交互，輸入 ‘Y’
            out = stdout.readlines()
            #屏幕輸出
            for o in out:
                print o,
        print '%s\tOK\n'%(ip)
        ssh.close()
    except :
        print '%s\tError\n'%(ip)
if __name__=='__main__':
    cmd = ['cal','echo hello!']#你要執行的命令列表
    username = ""  #用戶名
    passwd = ""    #密碼
    threads = []   #多線程
    print "Begin......"
    for i in range(1,254):
        ip = '192.168.1.'+str(i)
        a=threading.Thread(target=ssh2,args=(ip,username,passwd,cmd))
        a.start()

上面的程序還是有些技巧的：

1.利用多線程，同時發出登錄請求，同時去連接電腦，這樣速度快很多，我試了一下，如果不用多線程，直接一個一個挨着執行的話，大約5～10秒鍾才能對一台電腦操作完，具體時間要根據命令的來決定，如果是軟件安裝或者卸載時間要更長一些。這樣下來怎么也要一二十分鍾，用多線程后就快多了，所有的命令執行完用了不到2分鍾！

2.最好用root用戶登錄，因為安裝或者卸載軟件的時候如果用普通用戶又會提示輸入密碼，這樣又多了一次交互，處理起來就比較麻煩！安裝軟件時apt-get install xxx 最好加上“-y”參數，因為有時安裝或刪除軟件時提示是否繼續安裝或卸載，這又是一次自動交互！加上那個參數后就沒有人機交互了。

3. 循環時循環所有ip，因為計算機的ip是路由器自動分配的，保險起見，最好全部都執行，保證沒有遺漏的主機

4.遠端執行命令時如果有交互，可以這樣用 stdin.write("Y")來完成交互，“Y”就是輸入“Y”。

5.把所有的命令放到一個列表里面，遍歷列表可以依次執行列表里面的命令

6.為了更好的進行控制，最好在電腦上提前把root用戶打開，裝好ssh服務器並讓其開機自動執行。

 

---

 

ssh = paramiko.SSHClient()

ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

 

ssh.connect('10.111.43.18',22,'root','12345',timeout=5)

 

stdin, stdout, stderr = ssh.exec_command('cd /root/epoll/')

stdin, stdout, stderr = ssh.exec_command('pwd')

上邊的代碼輸出應該是 /root/epoll/，但結果卻是 /root ，即使用root登陸的缺省目錄

 

原因是exec_command為單個會話，執行完成之后會回到登錄時的缺省目錄

 

修改為這樣執行結果則為預期的 /root/epoll 目錄

stdin, stdout, stderr = ssh.exec_command('cd /root/epoll/;pwd')

---

 

 

 

最近閑着學習python，看到有個paramiko模塊，貌似很強大，學着寫了個一個小程序，可以直接在window下登錄到linux，執行並返回執行結果。

 

  下面直接貼代碼：

#!/usr/bin/env python    
import paramiko    
        
#hostname='192.168.0.102'    
hostname='172.28.102.250'
username='root'
password='abc'
        
#port=22    
if __name__=='__main__':    
        paramiko.util.log_to_file('paramiko.log')    
        s=paramiko.SSHClient()    
        #s.load_system_host_keys()    
        s.set_missing_host_key_policy(paramiko.AutoAddPolicy())    
        s.connect(hostname = hostname,username=username, password=password)    
        stdin,stdout,stderr=s.exec_command('ifconfig;free;df -h')    
        print stdout.read()    
        s.close()

下面看下效果，呵呵：

s501914252.HCA-FW9CX2XDesktop學習資料python paramiko-ssh.py"    
eth0      Link encap:Ethernet  HWaddr 00:24:01:01:EA:5C
          inet addr:172.32.34.240  Bcast:172.32.255.255  Mask:255.255.255.0
          UP BROADCAST RUNNING MULTICAST  MTU:1500  Metric:1
          RX packets:1428 errors:0 dropped:0 overruns:0 frame:0
          TX packets:2 errors:0 dropped:0 overruns:0 carrier:0
          collisions:0 txqueuelen:1000
          RX bytes:190928 (186.4 KiB)  TX bytes:128 (128.0 b)    
          Interrupt:20
        
eth1      Link encap:Ethernet  HWaddr 00:1F:29:03:54:3D
          inet addr:172.28.102.250  Bcast:172.28.255.255  Mask:255.255.0.0
          UP BROADCAST RUNNING MULTICAST  MTU:1500  Metric:1
          RX packets:99698 errors:0 dropped:0 overruns:0 frame:0
          TX packets:12305 errors:0 dropped:0 overruns:0 carrier:0
          collisions:0 txqueuelen:0
          RX bytes:36420324 (34.7 MiB)  TX bytes:923630 (901.9 KiB)    
        
lo        Link encap:Local Loopback    
          inet addr:127.0.0.1  Mask:255.0.0.0
          UP LOOPBACK RUNNING  MTU:16436  Metric:1
          RX packets:0 errors:0 dropped:0 overruns:0 frame:0
          TX packets:0 errors:0 dropped:0 overruns:0 carrier:0
          collisions:0 txqueuelen:0
          RX bytes:0 (0.0 b)  TX bytes:0 (0.0 b)    
        
peth1     Link encap:Ethernet  HWaddr FE:FF:FF:FF:FF:FF    
          UP BROADCAST RUNNING NOARP  MTU:1500  Metric:1
          RX packets:99686 errors:0 dropped:0 overruns:0 frame:0
          TX packets:12333 errors:0 dropped:0 overruns:0 carrier:0
          collisions:0 txqueuelen:1000
          RX bytes:36818715 (35.1 MiB)  TX bytes:1006230 (982.6 KiB)    
          Interrupt:20 Memory:f3000000-f3010000    
        
vif0.1    Link encap:Ethernet  HWaddr FE:FF:FF:FF:FF:FF    
          UP BROADCAST RUNNING NOARP  MTU:1500  Metric:1
          RX packets:12305 errors:0 dropped:0 overruns:0 frame:0
          TX packets:99702 errors:0 dropped:0 overruns:0 carrier:0
          collisions:0 txqueuelen:0
          RX bytes:923630 (901.9 KiB)  TX bytes:36420811 (34.7 MiB)    
        
vif1.0    Link encap:Ethernet  HWaddr FE:FF:FF:FF:FF:FF    
          UP BROADCAST RUNNING NOARP  MTU:1500  Metric:1
          RX packets:5 errors:0 dropped:0 overruns:0 frame:0
          TX packets:85175 errors:0 dropped:542 overruns:0 carrier:0
          collisions:0 txqueuelen:500
          RX bytes:140 (140.0 b)  TX bytes:20377326 (19.4 MiB)    
        
vif2.0    Link encap:Ethernet  HWaddr FE:FF:FF:FF:FF:FF    
          UP BROADCAST RUNNING NOARP  MTU:1500  Metric:1
          RX packets:5 errors:0 dropped:0 overruns:0 frame:0
          TX packets:84859 errors:0 dropped:411 overruns:0 carrier:0
          collisions:0 txqueuelen:500
          RX bytes:140 (140.0 b)  TX bytes:20301953 (19.3 MiB)    
        
vif3.0    Link encap:Ethernet  HWaddr FE:FF:FF:FF:FF:FF    
          UP BROADCAST RUNNING NOARP  MTU:1500  Metric:1
          RX packets:5 errors:0 dropped:0 overruns:0 frame:0
          TX packets:84337 errors:0 dropped:546 overruns:0 carrier:0
          collisions:0 txqueuelen:500
          RX bytes:140 (140.0 b)  TX bytes:20174994 (19.2 MiB)    
        
virbr0    Link encap:Ethernet  HWaddr 00:00:00:00:00:00
          inet addr:192.168.122.1  Bcast:192.168.122.255  Mask:255.255.255.0
          UP BROADCAST RUNNING MULTICAST  MTU:1500  Metric:1
          RX packets:0 errors:0 dropped:0 overruns:0 frame:0
          TX packets:0 errors:0 dropped:0 overruns:0 carrier:0
          collisions:0 txqueuelen:0
          RX bytes:0 (0.0 b)  TX bytes:0 (0.0 b)    
        
xenbr1    Link encap:Ethernet  HWaddr FE:FF:FF:FF:FF:FF    
          UP BROADCAST RUNNING NOARP  MTU:1500  Metric:1
          RX packets:86405 errors:0 dropped:0 overruns:0 frame:0
          TX packets:0 errors:0 dropped:0 overruns:0 carrier:0
          collisions:0 txqueuelen:0
          RX bytes:19498121 (18.5 MiB)  TX bytes:0 (0.0 b)    
        
             total       used       free     shared    buffers     cached    
Mem:       3359744     464724    2895020          0      31096     246296
-/+ buffers/cache:     187332    3172412
Swap:      5406712          0    5406712
Filesystem            Size  Used Avail Use% Mounted on    
/dev/mapper/VolGroup00-LogVol00    
                      221G   25G  185G  12% /
/dev/sda1              99M   22M   73M  23% /boot    
tmpfs                 1.8G     0  1.8G   0% /dev/shm    
none                  1.8G  152K  1.8G   1% /var/lib/xenstored

---

－、執行遠程命令:
#!/usr/bin/python
#coding:utf-8
import paramiko
port =22
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect("*.*.*.*",port,"username", "password")
stdin, stdout, stderr = ssh.exec_command("你的命令")
print stdout.readlines()
ssh.close()

二、上傳文件到遠程
#!/usr/bin/python
#coding:utf-8
import paramiko

port =22
t = paramiko.Transport(("IP",port))
t.connect(username = "username", password = "password")
sftp = paramiko.SFTPClient.from_transport(t)
remotepath='/tmp/test.txt'
localpath='/tmp/test.txt'
sftp.put(localpath,remotepath)
t.close()

三、從遠程下載文件
#!/usr/bin/python
#coding:utf-8
import paramiko

port =22
t = paramiko.Transport(("IP",port))
t.connect(username = "username", password = "password")
sftp = paramiko.SFTPClient.from_transport(t)
remotepath='/tmp/test.txt'
localpath='/tmp/test.txt'
sftp.get(remotepath, localpath)
t.close()

四、執行多個命令
#!/usr/bin/python
#coding:utf-8

import sys
sys.stderr = open('/dev/null')       # Silence silly warnings from paramiko
import paramiko as pm
sys.stderr = sys.__stderr__
import os

class AllowAllKeys(pm.MissingHostKeyPolicy):
    def missing_host_key(self, client, hostname, key):
        return

HOST = '127.0.0.1'
USER = ''
PASSWORD = ''

client = pm.SSHClient()
client.load_system_host_keys()
client.load_host_keys(os.path.expanduser('~/.ssh/known_hosts'))
client.set_missing_host_key_policy(AllowAllKeys())
client.connect(HOST, username=USER, password=PASSWORD)

channel = client.invoke_shell()
stdin = channel.makefile('wb')
stdout = channel.makefile('rb')

stdin.write('''
cd tmp
ls
exit
''')
print stdout.read()

stdout.close()
stdin.close()
client.close()

五、獲取多個文件
#!/usr/bin/python
#coding:utf-8

import paramiko
import os

ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect('localhost',username='****')  

apath = '/var/log'
apattern = '"*.log"'
rawcommand = 'find {path} -name {pattern}'
command = rawcommand.format(path=apath, pattern=apattern)
stdin, stdout, stderr = ssh.exec_command(command)
filelist = stdout.read().splitlines()

ftp = ssh.open_sftp()
for afile in filelist:
    (head, filename) = os.path.split(afile)
    print(filename)
    ftp.get(afile, './'+filename)
ftp.close()
ssh.close()

---

paramiko模塊是基於Python實現的ssh遠程安全鏈接，實現命令遠程、文件傳輸、SSH代理等功能。本文詳細介紹了paramiko模塊常用的方法，且實用價值非常高，稍作修改就能做出一個實用的運維工具。

 

安裝

letong@me:~$ sudo pip install paramiko

 

簡單的遠程執行命令實例

#!/usr/bin/env python
import paramiko
hostname='127.0.0.1'
username='root'
password='123456'
port=2222

ssh=paramiko.SSHClient()
ssh.load_system_host_keys()
ssh.connect(hostname=hostname,port=port,username=username,password=password)
stdin,stdout,stderr=ssh.exec_command('ls /')
print stdout.read()
ssh.close()

 

SSHClient類

SSHClient類是SSH服務會話的高級表示，封裝了傳輸、通道以及SFTPClient的校驗、建立方法，通常用於執行命令。

connect方法

connect(self, hostname, port=22, username=None, password=None, pkey=None, Key_filename=None, timeout=None, allow_agent=True, look_for_keys=True, compress=False)

參數說明：

hostname，連接目標的主機地址 port，連接目標的端口，默認22 username，用戶名 password，用戶密碼 pkey，私鑰方式用戶驗證 key_filename，私鑰文件名 timeout，超時時間 allow_agent，是否允許ssh代理 look_for_keys，是否允許搜索私鑰文件 compress，打開時是否壓縮

 

exec_command方法

exec_command(self, command, bufzise=-1)

參數說明：

command，執行的命令 bufsize，文件緩沖區大小，-1不限制

 

load_system_host_keys方法

load_system_host_keys(self, filename=None)

參數說明：

filename，指定遠程主機公鑰文件

 

set_missing_host_key_policy方法

ssh=paramiko.SSHClient()

ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

參數說明：

AutoAddPolicy，自動添加主機名及密鑰到本地並保存，不依賴load_system_host_keys()配置 RejectPolicy，自動拒絕未知主機名和密鑰，依賴load_system_host_keys()配置 WarnningPolicy，功能與AutoAddPolicy相同，但是未知主機會報警

 

SFTPClient類

根據SSH傳輸協議的sftp會話，實現遠程文件上傳、下載等操作。

from_transport方法

from_transport(cls, t)

參數說明：

t，一個已通過驗證的傳輸對象

例：

>>> import paramiko >>> a = paramiko.Transport((“127.0.0.1″,2222)) >>> a.connect(username=”root”, password=’123456′) >>> sftp = paramiko.SFTPClient.from_transport(a)

 

put方法

put(self, localpath, remotepath, callback=None, confirm=True)

參數說明：

localpath，上傳源文件的本地路徑 remotepath，目標路徑 callback，獲取接收與總傳輸字節數 confirm，上傳完畢后是否調用stat()方法，以便確認文件大小

例：

>>> localpath=’ftp-test.log’ >>> remotepath=’/data/ftp-test.log’ >>> sftp.put(localpath,remotepath)

 

get方法

get(self, remotepath, localpath, callback=None)

參數說明：

remotepath，需要下載的遠程文件 localpath，本地存儲路徑 callback，同put方法

 

其他方法

mkdir，用於創建目錄 remove，刪除目錄 rename，重命名 stat，獲取文件信息 listdir，獲取目錄列表

 

 

---